diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 7 |
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) |