Port Swift integration tests to pytest.

criemen · criemen · commit e5261f8bfcb9 · 2024-08-08T08:47:49.000+02:00
diff --git a/swift/ql/integration-tests/BUILD.bazel b/swift/ql/integration-tests/BUILD.bazel
@@ -1,7 +1,6 @@
 py_library(
     name = "utils",
     srcs = [
-        "create_database_utils.py",
         "diagnostics_test_utils.py",
     ],
     visibility = ["//swift:__subpackages__"],
diff --git a/swift/ql/integration-tests/create_database_utils.py b/swift/ql/integration-tests/create_database_utils.py
diff --git a/swift/ql/integration-tests/legacy b/swift/ql/integration-tests/legacy
diff --git a/swift/ql/integration-tests/linux-only/RegexLiteralExpr/test.py b/swift/ql/integration-tests/linux-only/RegexLiteralExpr/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
+import runs_on
 
-run_codeql_database_create([
-    "swiftc -enable-bare-slash-regex regex.swift -o /dev/null",
-], lang="swift")
+
+@runs_on.linux
+def test(codeql, swift):
+    codeql.database.create(command="swiftc -enable-bare-slash-regex regex.swift -o /dev/null")
diff --git a/swift/ql/integration-tests/linux-only/autobuilder/unsupported-os/test.py b/swift/ql/integration-tests/linux-only/autobuilder/unsupported-os/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.linux
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/failure/test.py b/swift/ql/integration-tests/osx-only/autobuilder/failure/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/no-build-system/test.py b/swift/ql/integration-tests/osx-only/autobuilder/no-build-system/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/no-swift-with-spm/test.py b/swift/ql/integration-tests/osx-only/autobuilder/no-swift-with-spm/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/no-swift/test.py b/swift/ql/integration-tests/osx-only/autobuilder/no-swift/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/no-xcode-with-spm/test.py b/swift/ql/integration-tests/osx-only/autobuilder/no-xcode-with-spm/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/only-tests-with-spm/test.py b/swift/ql/integration-tests/osx-only/autobuilder/only-tests-with-spm/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/only-tests/test.py b/swift/ql/integration-tests/osx-only/autobuilder/only-tests/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
-from diagnostics_test_utils import *
+import runs_on
 
-run_codeql_database_create([], lang='swift', keep_trap=True, db=None, runFunction=runUnsuccessfully)
-check_diagnostics()
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(_assert_failure=True)
diff --git a/swift/ql/integration-tests/osx-only/autobuilder/xcode-fails-spm-works/test.py b/swift/ql/integration-tests/osx-only/autobuilder/xcode-fails-spm-works/test.py
@@ -1,5 +1,8 @@
-from create_database_utils import *
+import runs_on
+import pytest
 
-runSuccessfully(['xcodebuild', 'clean'])
-runSuccessfully(['swift', 'package', 'clean'])
-run_codeql_database_create([], lang='swift', keep_trap=True)
+
+@runs_on.macos
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift):
+    codeql.database.create()
diff --git a/swift/ql/integration-tests/osx-only/canonical-case/test.py b/swift/ql/integration-tests/osx-only/canonical-case/test.py
@@ -1,5 +1,6 @@
-from create_database_utils import *
+import runs_on
 
-run_codeql_database_create([
-    './build.sh',
-], lang='swift')
+
+@runs_on.macos
+def test(codeql, swift):
+    codeql.database.create(command="./build.sh")
diff --git a/swift/ql/integration-tests/osx-only/hello-xcode/test.py b/swift/ql/integration-tests/osx-only/hello-xcode/test.py
@@ -1,9 +1,13 @@
-from create_database_utils import *
+import runs_on
+import pytest
 
-run_codeql_database_create([
-    'xcodebuild clean',
-    'xcodebuild build '
-    '-project codeql-swift-autobuild-test.xcodeproj '
-    '-target codeql-swift-autobuild-test '
-    'CODE_SIGNING_REQUIRED=NO CODE_SIGNING_ALLOWED=NO',
-], lang='swift', keep_trap=True)
+
+@runs_on.macos
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift):
+    codeql.database.create(
+        command="xcodebuild build "
+        "-project codeql-swift-autobuild-test.xcodeproj "
+        "-target codeql-swift-autobuild-test "
+        "CODE_SIGNING_REQUIRED=NO CODE_SIGNING_ALLOWED=NO",
+    )
diff --git a/swift/ql/integration-tests/posix-only/cross-references/test.py b/swift/ql/integration-tests/posix-only/cross-references/test.py
@@ -1,6 +1,8 @@
-from create_database_utils import *
+import runs_on
+import pytest
 
-run_codeql_database_create([
-    'swift package clean',
-    'swift build'
-], lang='swift')
+
+@runs_on.posix
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift):
+    codeql.database.create(command="swift build")
diff --git a/swift/ql/integration-tests/posix-only/deduplication/test.py b/swift/ql/integration-tests/posix-only/deduplication/test.py
@@ -1,6 +1,6 @@
-from create_database_utils import *
+import runs_on
 
-run_codeql_database_create([
-    'swift package clean',
-    'swift build',
-], lang='swift', keep_trap=True)
+
+@runs_on.posix
+def test(codeql, swift):
+    codeql.database.create(command="swift build", keep_trap=True)
diff --git a/swift/ql/integration-tests/posix-only/frontend-invocations/test.py b/swift/ql/integration-tests/posix-only/frontend-invocations/test.py
@@ -1,22 +1,24 @@
-from create_database_utils import *
-from subprocess import check_call
 from hashlib import sha256
 from pathlib import Path
+import runs_on
+import pytest
 
-run_codeql_database_create([
-    './build.sh',
-], lang='swift',
-)
 
-with open('hashes.expected', 'w') as expected:
-    for f in sorted(Path().glob("*.swiftmodule")):
-        with open(f, 'rb') as module:
-            print(f.name, sha256(module.read()).hexdigest(), file=expected)
+@runs_on.posix
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift, expected_files):
+    codeql.database.create(command="./build.sh")
+    with open("hashes.expected", "w") as expected:
+        for f in sorted(Path().glob("*.swiftmodule")):
+            with open(f, "rb") as module:
+                print(f.name, sha256(module.read()).hexdigest(), file=expected)
 
-with open('hashes.actual', 'w') as actual:
-    hashes = [(s.name, s.resolve().name) for s in Path("test-db/working/swift-extraction-artifacts/store").iterdir()]
-    hashes.sort()
-    for module, hash in hashes:
-        print(module, hash, file=actual)
-
-check_call(['diff', '-u', 'hashes.expected', 'hashes.actual'])
+    with open("hashes.actual", "w") as actual:
+        hashes = [
+            (s.name, s.resolve().name)
+            for s in Path("test-db/working/swift-extraction-artifacts/store").iterdir()
+        ]
+        hashes.sort()
+        for module, hash in hashes:
+            print(module, hash, file=actual)
+    expected_files.add("hashes.expected")
diff --git a/swift/ql/integration-tests/posix-only/hello-world/test.py b/swift/ql/integration-tests/posix-only/hello-world/test.py
@@ -1,6 +1,6 @@
-from create_database_utils import *
+import runs_on
 
-run_codeql_database_create([
-    'swift package clean',
-    'swift build'
-], lang='swift')
+
+@runs_on.posix
+def test(codeql, swift):
+    codeql.database.create(command="swift build")
diff --git a/swift/ql/integration-tests/posix-only/linkage-awareness/test.py b/swift/ql/integration-tests/posix-only/linkage-awareness/test.py
@@ -1,5 +1,8 @@
-from create_database_utils import *
+import runs_on
+import pytest
 
-run_codeql_database_create([
-    './build.sh',
-], lang='swift', keep_trap=True)
+
+@runs_on.posix
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift):
+    codeql.database.create(command="./build.sh")
diff --git a/swift/ql/integration-tests/posix-only/partial-modules/test.py b/swift/ql/integration-tests/posix-only/partial-modules/test.py
@@ -1,6 +1,8 @@
-from create_database_utils import *
+import runs_on
+import pytest
 
-run_codeql_database_create([
-    'swift package clean',
-    'swift build'
-], lang='swift', keep_trap=True)
+
+@runs_on.posix
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift):
+    codeql.database.create(command="swift build")
diff --git a/swift/ql/integration-tests/posix-only/symlinks/test.py b/swift/ql/integration-tests/posix-only/symlinks/test.py
@@ -1,20 +1,21 @@
-from create_database_utils import *
 import os
+import runs_on
+import pytest
 
-symlinks = ['preserve/Sources/A.swift', 'resolve/Sources/A.swift']
 
-for s in symlinks:
-    try:
-        os.symlink(os.getcwd() + '/main.swift', s)
-    except:
-        pass
+@runs_on.posix
+@pytest.mark.ql_test("DB-CHECK", xfail=True)
+def test(codeql, swift):
+    symlinks = ["preserve/Sources/A.swift", "resolve/Sources/A.swift"]
 
-run_codeql_database_create([
-    'swift package clean --package-path resolve',
-    'swift build --package-path resolve',
-    'swift package clean --package-path preserve',
-    'env CODEQL_PRESERVE_SYMLINKS=true swift build --package-path preserve'
-], lang='swift', keep_trap=True)
-
-for s in symlinks:
-    os.unlink(s)
+    for s in symlinks:
+        try:
+            os.symlink(os.getcwd() + "/main.swift", s)
+        except:
+            pass
+    codeql.database.create(
+        command=[
+            "swift build --package-path resolve",
+            "env CODEQL_PRESERVE_SYMLINKS=true swift build --package-path preserve",
+        ]
+    )
diff --git a/swift/ql/integration-tests/qlpack.yml b/swift/ql/integration-tests/qlpack.yml
diff --git a/swift/ql/integration-tests/runner.py b/swift/ql/integration-tests/runner.py
