fixes in tests

barseghyanartur · Aug 11, 2018 · 496c272 · 496c272
1 parent fe12b45
commit 496c272
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/fobi/tests/helpers.py b/src/fobi/tests/helpers.py
@@ -187,14 +187,18 @@ def create_form_with_entries(user=None,
     return form_entry
 
 
-def db_clean_up():
+def db_clean_up(clean_form=False, clean_elements=True, clean_handlers=True):
     """Clean up the database.
 
     Clean up the database by removing all form element and form handler
     entries.
     """
-    FormElementEntry._default_manager.all().delete()
-    FormHandlerEntry._default_manager.all().delete()
+    if clean_form:
+        FormEntry._default_manager.all().delete()
+    if clean_elements:
+        FormElementEntry._default_manager.all().delete()
+    if clean_handlers:
+        FormHandlerEntry._default_manager.all().delete()
 
 
 def phantom_js_clean_up():