Add test for default status

canonical · Jan 7, 2025 · 536267e · 536267e
1 parent 6df7ae2
commit 536267e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/tests/fixtures/payloads.py b/tests/fixtures/payloads.py
@@ -48,6 +48,7 @@
     },
     "priority": "critical",
     "published": "2020-08-01 12:42:54",
+    "status": "not-in-ubuntu",
 }
 
 cve2 = {

diff --git a/tests/test_routes.py b/tests/test_routes.py
@@ -48,6 +48,23 @@ def test_cves_query_no_500(self):
 
         assert response.status_code == 200
 
+    def test_cves_default_status(self):
+        # Add new CVE without status
+        cve_payload = payloads.cve1.copy()
+
+        add_cve_response = self.client.put(
+            "/security/updates/cves.json",
+            json=[cve_payload],
+        )
+
+        assert add_cve_response.status_code == 200
+        response = self.client.get("/security/cves.json")
+
+        assert response.status_code == 200
+        # Only the CVE with the default status should be returned
+        assert len(response.json["cves"]) == 1
+        assert response.json["cves"][0]["status"] == "in-progress"
+
     def test_cves_returns_200_for_non_existing_package_name(self):
         response = self.client.get("/security/cves.json?package=no-exist")