From 496c2729e5683cf0acb07536594d8a85b9992352 Mon Sep 17 00:00:00 2001 From: Artur Barseghyan Date: Sat, 11 Aug 2018 23:59:46 +0200 Subject: [PATCH] fixes in tests --- src/fobi/tests/helpers.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/fobi/tests/helpers.py b/src/fobi/tests/helpers.py index 111c21bf9..4f220d59d 100644 --- 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():