Do not use file for small test data list

Instead inline the data for improved readability

Signed-off-by: Philippe Ombredanne <[email protected]>

Author: pombredanne

vulntotal/tests/test_data/deps/purls.txt

@@ -21,9 +21,14 @@ class TestDeps(testcase.FileBasedTesting):
     test_data_dir = str(Path(__file__).resolve().parent / "test_data" / "deps")
 
     def test_generate_meta_payload(self):
-        file_purls = self.get_test_loc("purls.txt")
-        with open(file_purls) as f:
-            purls = f.readlines()
+        purls = [
+            "pkg:pypi/[email protected]",
+            "pkg:maven/org.apache.tomcat/[email protected]",
+            "pkg:npm/[email protected]",
+            "pkg:golang/github.com/cloudflare/[email protected]",
+            "pkg:cargo/[email protected]",
+        ]
+
         results = [deps.generate_meta_payload(PackageURL.from_string(purl)) for purl in purls]
         expected_file = self.get_test_loc("payloads_meta-expected.json", must_exist=False)
         util_tests.check_results_against_json(results, expected_file)

vulntotal/tests/test_data/github/purls.txt

@@ -21,9 +21,17 @@ class TestGithub(testcase.FileBasedTesting):
     test_data_dir = str(Path(__file__).resolve().parent / "test_data" / "github")
 
     def test_generate_graphql_payload(self):
-        file_purls = self.get_test_loc("purls.txt")
-        with open(file_purls) as f:
-            purls = f.readlines()
+        purls = [
+            "pkg:pypi/[email protected]",
+            "pkg:maven/org.apache.tomcat/[email protected]",
+            "pkg:nuget/[email protected]",
+            "pkg:npm/[email protected]",
+            "pkg:golang/github.com/cloudflare/[email protected]",
+            "pkg:composer/symfony/[email protected]",
+            "pkg:rust/[email protected]",
+            "pkg:erlang/[email protected]",
+            "pkg:gem/[email protected]",
+        ]
         results = [
             github.generate_graphql_payload(PackageURL.from_string(purl), "") for purl in purls
         ]

vulntotal/tests/test_data/gitlab/purls.txt

@@ -20,9 +20,14 @@ class TestGitlab(testcase.FileBasedTesting):
     test_data_dir = str(Path(__file__).resolve().parent / "test_data" / "gitlab")
 
     def test_generate_package_advisory_url(self):
-        file_purls = self.get_test_loc("purls.txt")
-        with open(file_purls) as f:
-            purls = f.readlines()
+        purls = [
+            "pkg:pypi/[email protected]",
+            "pkg:maven/org.apache.tomcat/[email protected]",
+            "pkg:npm/[email protected]",
+            "pkg:golang/github.com/mattermost/mattermost-server/v6/[email protected]",
+            "pkg:composer/bolt/[email protected]",
+            "pkg:nuget/[email protected]",
+        ]
         results = [gitlab.get_package_slug(PackageURL.from_string(purl)) for purl in purls]
         expected_file = self.get_test_loc("package_advisory_url-expected.json", must_exist=False)
         util_tests.check_results_against_json(results, expected_file)

vulntotal/tests/test_data/osv/purls.txt

@@ -21,9 +21,18 @@ class TestOSV(testcase.FileBasedTesting):
     test_data_dir = str(Path(__file__).resolve().parent / "test_data" / "osv")
 
     def test_generate_payload(self):
-        file_purls = self.get_test_loc("purls.txt")
-        with open(file_purls) as f:
-            purls = f.readlines()
+        purls = [
+            "pkg:pypi/[email protected]",
+            "pkg:android/System@10",
+            "pkg:debian:8/[email protected]",
+            "pkg:maven/org.apache.tomcat/[email protected]",
+            "pkg:linux/[email protected]",
+            "pkg:packagist/dolibarr/[email protected]",
+            "pkg:crates.io/[email protected]",
+            "pkg:nuget/[email protected]",
+            "pkg:npm/[email protected]",
+            "pkg:golang/github.com/cloudflare/[email protected]",
+        ]
         results = [osv.generate_payload(PackageURL.from_string(purl)) for purl in purls]
         expected_file = self.get_test_loc("payloads_data-expected.json", must_exist=False)
         util_tests.check_results_against_json(results, expected_file)