adjusted Update view

intelowlproject · May 13, 2024 · f4dd486 · f4dd486
1 parent 402a1af
commit f4dd486
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/api_app/views.py b/api_app/views.py
@@ -922,4 +922,8 @@ def pull(self, request, name=None):
                 {"detail": "Unexpected exception raised. Check the code."}
             )
         else:
+            if update_status is None:
+                raise ValidationError(
+                    {"detail": "This Plugin has no Update implemented"}
+                )
             return Response(data={"status": update_status}, status=status.HTTP_200_OK)
diff --git a/tests/api_app/analyzers_manager/test_views.py b/tests/api_app/analyzers_manager/test_views.py
@@ -40,13 +40,16 @@ def test_pull(self):
         self.assertIn("status", result)
         self.assertTrue(result["status"])
 
-        analyzer = "QuarkEngine"
+        analyzer = "Doc_Info"
         response = self.client.post(f"{self.URL}/{analyzer}/pull")
         self.assertEqual(response.status_code, 400)
         result = response.json()
+        print(result)
         self.assertIn("errors", result)
         self.assertIn("detail", result["errors"])
-        self.assertEqual(result["errors"]["detail"], "No update implemented")
+        self.assertEqual(
+            result["errors"]["detail"], "This Plugin has no Update implemented"
+        )
 
     def test_health_check(self):
         analyzer = "ClamAV"