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'})
