fix: convert data model tests to unittest format for CI compatibility

Author: bluesentinelsec

entrypoint/tests/test_data_model.py

@@ -1,4 +1,4 @@
-import pytest
+import unittest
 from entrypoint.data_model import ArtifactType, ScanConfig
 
 
@@ -7,87 +7,83 @@ class MockArgs:
     artifact_path = './test'
 
 
-def test_artifact_type_repository_exists():
-    assert ArtifactType.REPOSITORY.value == "repository"
+class TestDataModel(unittest.TestCase):
 
-def test_artifact_type_container_exists():
-    assert ArtifactType.CONTAINER.value == "container"
+    def test_artifact_type_repository_exists(self):
+        self.assertEqual(ArtifactType.REPOSITORY.value, "repository")
 
-def test_artifact_type_binary_exists():
-    assert ArtifactType.BINARY.value == "binary"
+    def test_artifact_type_container_exists(self):
+        self.assertEqual(ArtifactType.CONTAINER.value, "container")
 
-def test_artifact_type_archive_exists():
-    assert ArtifactType.ARCHIVE.value == "archive"
+    def test_artifact_type_binary_exists(self):
+        self.assertEqual(ArtifactType.BINARY.value, "binary")
 
-def test_artifact_type_from_string():
-    assert ArtifactType("repository") == ArtifactType.REPOSITORY
+    def test_artifact_type_archive_exists(self):
+        self.assertEqual(ArtifactType.ARCHIVE.value, "archive")
 
-def test_artifact_type_invalid_string_raises_exception():
-    try:
-        ArtifactType("invalid")
-        assert False, "Expected ValueError but none was raised"
-    except ValueError:
-        assert True
+    def test_artifact_type_from_string(self):
+        self.assertEqual(ArtifactType("repository"), ArtifactType.REPOSITORY)
 
-def test_artifact_type_empty_string_raises_exception():
-    try:
-        ArtifactType("")
-        assert False, "Expected ValueError but none was raised"
-    except ValueError:
-        assert True
+    def test_artifact_type_invalid_string_raises_exception(self):
+        with self.assertRaises(ValueError):
+            ArtifactType("invalid")
 
-def test_scan_config_can_be_created():
-    config = ScanConfig()
-    assert config is not None
+    def test_artifact_type_empty_string_raises_exception(self):
+        with self.assertRaises(ValueError):
+            ArtifactType("")
 
-def test_scan_config_has_artifact_type():
-    config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
-    assert config.artifact_type == ArtifactType.REPOSITORY
+    def test_scan_config_can_be_created(self):
+        config = ScanConfig()
+        self.assertIsNotNone(config)
 
-def test_scan_config_has_artifact_path():
-    config = ScanConfig(artifact_path="./test")
-    assert config.artifact_path == "./test"
+    def test_scan_config_has_artifact_type(self):
+        config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
+        self.assertEqual(config.artifact_type, ArtifactType.REPOSITORY)
 
-def test_scan_config_from_args_exists():
-    mock_args = MockArgs()
-    config = ScanConfig.from_args(mock_args)
-    assert config is not None
+    def test_scan_config_has_artifact_path(self):
+        config = ScanConfig(artifact_path="./test")
+        self.assertEqual(config.artifact_path, "./test")
 
-def test_scan_config_from_args_converts_artifact_type():
-    mock_args = MockArgs()
-    config = ScanConfig.from_args(mock_args)
-    assert config.artifact_type == ArtifactType.REPOSITORY
+    def test_scan_config_from_args_exists(self):
+        mock_args = MockArgs()
+        config = ScanConfig.from_args(mock_args)
+        self.assertIsNotNone(config)
 
-def test_scan_config_from_args_converts_artifact_path():
-    mock_args = MockArgs()
-    config = ScanConfig.from_args(mock_args)
-    assert config.artifact_path == './test'
+    def test_scan_config_from_args_converts_artifact_type(self):
+        mock_args = MockArgs()
+        config = ScanConfig.from_args(mock_args)
+        self.assertEqual(config.artifact_type, ArtifactType.REPOSITORY)
 
-def test_scan_config_repository_type_comparison():
-    config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
-    assert config.artifact_type == ArtifactType.REPOSITORY
+    def test_scan_config_from_args_converts_artifact_path(self):
+        mock_args = MockArgs()
+        config = ScanConfig.from_args(mock_args)
+        self.assertEqual(config.artifact_path, './test')
 
-def test_scan_config_container_type_comparison():
-    config = ScanConfig(artifact_type=ArtifactType.CONTAINER)
-    assert config.artifact_type == ArtifactType.CONTAINER
+    def test_scan_config_repository_type_comparison(self):
+        config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
+        self.assertEqual(config.artifact_type, ArtifactType.REPOSITORY)
 
-def test_scan_config_binary_type_comparison():
-    config = ScanConfig(artifact_type=ArtifactType.BINARY)
-    assert config.artifact_type == ArtifactType.BINARY
+    def test_scan_config_container_type_comparison(self):
+        config = ScanConfig(artifact_type=ArtifactType.CONTAINER)
+        self.assertEqual(config.artifact_type, ArtifactType.CONTAINER)
 
-def test_scan_config_archive_type_comparison():
-    config = ScanConfig(artifact_type=ArtifactType.ARCHIVE)
-    assert config.artifact_type == ArtifactType.ARCHIVE
+    def test_scan_config_binary_type_comparison(self):
+        config = ScanConfig(artifact_type=ArtifactType.BINARY)
+        self.assertEqual(config.artifact_type, ArtifactType.BINARY)
 
-def test_scan_config_container_platform_check():
-    config = ScanConfig(artifact_type=ArtifactType.CONTAINER)
-    assert config.artifact_type == ArtifactType.CONTAINER
+    def test_scan_config_archive_type_comparison(self):
+        config = ScanConfig(artifact_type=ArtifactType.ARCHIVE)
+        self.assertEqual(config.artifact_type, ArtifactType.ARCHIVE)
 
-def test_scan_config_artifact_type_value():
-    config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
-    assert config.artifact_type.value == "repository"
+    def test_scan_config_container_platform_check(self):
+        config = ScanConfig(artifact_type=ArtifactType.CONTAINER)
+        self.assertEqual(config.artifact_type, ArtifactType.CONTAINER)
 
-def test_scan_config_display_mapping():
-    config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
-    display_type = "repository" if config.artifact_type == ArtifactType.REPOSITORY else config.artifact_type.value
-    assert display_type == "repository"
+    def test_scan_config_artifact_type_value(self):
+        config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
+        self.assertEqual(config.artifact_type.value, "repository")
+
+    def test_scan_config_display_mapping(self):
+        config = ScanConfig(artifact_type=ArtifactType.REPOSITORY)
+        display_type = "repository" if config.artifact_type == ArtifactType.REPOSITORY else config.artifact_type.value
+        self.assertEqual(display_type, "repository")