Address review.

go/ql/integration-tests/all-platforms/go/diagnostics/build-constraints-exclude-all-go-files/test.py

@@ -2,7 +2,7 @@ import os
 import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "test"
     codeql.database.create(source_root="work")

go/ql/integration-tests/all-platforms/go/diagnostics/go-files-found-not-processed/test.py

@@ -1,8 +1,7 @@
 import os
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "test"
     codeql.database.create(source_root="work")

go/ql/integration-tests/all-platforms/go/diagnostics/newer-go-version-needed/test.py

@@ -1,8 +1,7 @@
 import os
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "test"
     codeql.database.create(source_root="work", _assert_failure=True)

go/ql/integration-tests/all-platforms/go/diagnostics/no-go-files-found/test.py

@@ -1,8 +1,7 @@
 import os
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "test"
     codeql.database.create(source_root="work", _assert_failure=True)

go/ql/integration-tests/all-platforms/go/diagnostics/package-not-found-with-go-mod/test.py

@@ -1,8 +1,7 @@
 import os
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "test"
     codeql.database.create(source_root="work")

go/ql/integration-tests/all-platforms/go/diagnostics/package-not-found-without-go-mod/test.py

@@ -1,8 +1,7 @@
 import os
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "test"
     codeql.database.create(source_root="work")

go/ql/integration-tests/all-platforms/go/diagnostics/unsupported-relative-path/test.py

@@ -1,8 +1,7 @@
 import os
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["GITHUB_REPOSITORY"] = "a/b"
     codeql.database.create(source_root="work")

go/ql/integration-tests/all-platforms/go/resolve-build-environment/newer-go-needed/test.py

@@ -1,6 +1,3 @@
-import pytest
-
-
-@pytest.mark.resolve_build_environment(env={"GOTOOLCHAIN": "go1.21.0"})
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.env = {"GOTOOLCHAIN": "go1.21.0"}
     codeql.database.create(source_root="src")

go/ql/integration-tests/linux-only/go/dep-sample/test.py

@@ -1,10 +1,9 @@
 import os
 import runs_on
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
 @runs_on.linux
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "deptest"
     codeql.database.create(source_root="work")

go/ql/integration-tests/linux-only/go/glide-sample/test.py

@@ -1,10 +1,9 @@
 import os
 import runs_on
-import pytest
 
 
-@pytest.mark.resolve_build_environment(source_root="work")
 @runs_on.linux
-def test(codeql, go):
+def test(codeql, go, check_build_environment):
+    check_build_environment.source_root = "work"
     os.environ["LGTM_INDEX_IMPORT_PATH"] = "glidetest"
     codeql.database.create(source_root="work")