def test_revalidation(self): request = get_request() testform = NamedUrlSessionFormWizard([('start', Step1), ('step2', Step2)], url_name='nwiz_session') response = testform(request, step='done') testform.render_done(None) self.assertEqual(testform.storage.get_current_step(), 'start')
def test_add_extra_context(self): request = get_request() testform = NamedUrlSessionFormWizard([('start', Step1), ('step2', Step2)], url_name='nwiz_session') response = testform(request, step='form1', extra_context={'key1': 'value1'}) self.assertEqual(testform.get_extra_context(), {'key1': 'value1'}) testform.reset_wizard() response = testform(request, extra_context={'key2': 'value2'}) self.assertEqual(testform.get_extra_context(), {'key2': 'value2'})