Fix document deletion - tests

1 files changed, 5 insertions, 2 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 03f8e3a69..9f97fd33b 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -330,6 +330,7 @@ class ManagedModelTestRunner(DiscoverRunner):
 
 class WizardTest(object):
     url_name = None
+    url_uri = None
     wizard_name = ''
     steps = None
     condition_dict = None
@@ -416,7 +417,7 @@ class WizardTest(object):
             next_form = self.steps[idx + 1][0]
             self.assertRedirects(
                 response,
-                '/{}/{}'.format(self.url_name, next_form),
+                '/{}/{}'.format(self.url_uri, next_form),
                 msg_prefix="Dataset n{} Redirection to {} has failed - "
                            "Error on previous form ({})?".format(
                     data_idx + 1, next_form, current_step)
@@ -425,13 +426,15 @@ class WizardTest(object):
             #  last form
             self.assertRedirects(
                 response,
-                '/{}/done'.format(self.url_name))
+                '/{}/done'.format(self.url_uri))
         return response
 
     def test_wizard(self):
         if self.pass_test():
             return
         url = reverse(self.url_name)
+        if not self.url_uri:
+            self.url_uri = self.url_name
         self.pre_wizard()
         for data_idx, test_form_data in enumerate(self.form_datas):
             test_form_data.inits(self)