Merge pull request #39 from yftacherzog/RHTAPWATCH-694-remove-spec-fr…

…om-snapshot fix(RHTAPWATCH-694): snapshot should not have spec field
redhat-appstudio · Dec 31, 2023 · 6eea7e0 · 6eea7e0
2 parents 3e4b003 + de5167f
commit 6eea7e0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 17 deletions.
diff --git a/tests/test_rpm_verifier.py b/tests/test_rpm_verifier.py
@@ -198,20 +198,18 @@ def test_get_unsigned_rpms(test_input: list[str], expected: list[str]) -> None:
             dedent(
                 """
                 {
-                    "spec": {
-                        "application": "test",
-                        "components": [
-                            {
-                                "containerImage": "quay.io/container-image@sha256:123"
-                            },
-                            {
-                                "containerImage": "quay.io/container-image@sha256:456"
-                            },      
-                            {
-                                "containerImage": "quay.io/container-image@sha256:789"
-                            }
-                        ]
-                    }
+                    "application": "test",
+                    "components": [
+                        {
+                            "containerImage": "quay.io/container-image@sha256:123"
+                        },
+                        {
+                            "containerImage": "quay.io/container-image@sha256:456"
+                        },
+                        {
+                            "containerImage": "quay.io/container-image@sha256:789"
+                        }
+                    ]
                 }
                 """
             ).strip(),

diff --git a/verify_rpms/rpm_verifier.py b/verify_rpms/rpm_verifier.py
@@ -106,9 +106,7 @@ def parse_image_input(image_input: str) -> list[str]:
         snapshot = loads(s=image_input)
     except JSONDecodeError:
         return [image_input]
-    components = snapshot["spec"]["components"]
-    container_images = [component["containerImage"] for component in components]
-    return container_images
+    return [component["containerImage"] for component in snapshot["components"]]
 
 
 @dataclass(frozen=True)