Complete tests for pet

riadelimemmedov · Feb 17, 2024 · 53d92c3 · 53d92c3
1 parent 6db8b14
commit 53d92c3
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/backend/abstract/constants.py b/backend/abstract/constants.py
@@ -26,3 +26,6 @@ class AppName(Enum):
     ("Spayed", _("spayed")),
     ("Other", _("other")),
 ]
+
+# !Image Extension
+ImageExtension = ["png", "jpg", "jpeg"]
diff --git a/backend/infrastructure/test_data_clear.py b/backend/infrastructure/test_data_clear.py
@@ -0,0 +1,16 @@
+import logging
+
+logging.basicConfig(
+    level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s"
+)
+
+
+# ?TruncateTestData
+def TruncateTestData(model):
+    qs = model.objects.all()
+    if qs is not None:
+        for obj in qs:
+            obj.pet_photo_url.delete()
+            logging.info("Photo truncated successfully!")
+    else:
+        logging.error("Photo not truncated.Please try again!")
diff --git a/backend/infrastructure/test_data_init.py b/backend/infrastructure/test_data_init.py