88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
from datetime import datetime
|
|
|
|
from django.http import HttpRequest
|
|
from django.conf import settings
|
|
from django.utils.importlib import import_module
|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
|
|
def get_request():
|
|
request = HttpRequest()
|
|
engine = import_module(settings.SESSION_ENGINE)
|
|
request.session = engine.SessionStore(None)
|
|
return request
|
|
|
|
|
|
class TestStorage(object):
|
|
def setUp(self):
|
|
self.testuser, created = User.objects.get_or_create(username='testuser1')
|
|
|
|
def test_current_step(self):
|
|
request = get_request()
|
|
storage = self.get_storage()('wizard1', request, None)
|
|
my_step = 2
|
|
|
|
self.assertEqual(storage.current_step, None)
|
|
|
|
storage.current_step = my_step
|
|
self.assertEqual(storage.current_step, my_step)
|
|
|
|
storage.reset()
|
|
self.assertEqual(storage.current_step, None)
|
|
|
|
storage.current_step = my_step
|
|
storage2 = self.get_storage()('wizard2', request, None)
|
|
self.assertEqual(storage2.current_step, None)
|
|
|
|
def test_step_data(self):
|
|
request = get_request()
|
|
storage = self.get_storage()('wizard1', request, None)
|
|
step1 = 'start'
|
|
step_data1 = {'field1': 'data1',
|
|
'field2': 'data2',
|
|
'field3': datetime.now(),
|
|
'field4': self.testuser}
|
|
|
|
self.assertEqual(storage.get_step_data(step1), None)
|
|
|
|
storage.set_step_data(step1, step_data1)
|
|
self.assertEqual(storage.get_step_data(step1), step_data1)
|
|
|
|
storage.reset()
|
|
self.assertEqual(storage.get_step_data(step1), None)
|
|
|
|
storage.set_step_data(step1, step_data1)
|
|
storage2 = self.get_storage()('wizard2', request, None)
|
|
self.assertEqual(storage2.get_step_data(step1), None)
|
|
|
|
def test_extra_context(self):
|
|
request = get_request()
|
|
storage = self.get_storage()('wizard1', request, None)
|
|
extra_context = {'key1': 'data1',
|
|
'key2': 'data2',
|
|
'key3': datetime.now(),
|
|
'key4': self.testuser}
|
|
|
|
self.assertEqual(storage.extra_data, {})
|
|
|
|
storage.extra_data = extra_context
|
|
self.assertEqual(storage.extra_data, extra_context)
|
|
|
|
storage.reset()
|
|
self.assertEqual(storage.extra_data, {})
|
|
|
|
storage.extra_data = extra_context
|
|
storage2 = self.get_storage()('wizard2', request, None)
|
|
self.assertEqual(storage2.extra_data, {})
|
|
|
|
def test_extra_context_key_persistence(self):
|
|
request = get_request()
|
|
storage = self.get_storage()('wizard1', request, None)
|
|
|
|
self.assertFalse('test' in storage.extra_data)
|
|
|
|
storage.extra_data['test'] = True
|
|
|
|
self.assertTrue('test' in storage.extra_data)
|