112 lines
3.6 KiB
Python
112 lines
3.6 KiB
Python
from datetime import datetime
|
|
from importlib import import_module
|
|
import os
|
|
import tempfile
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.core.files.storage import FileSystemStorage
|
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
|
|
|
|
|
temp_storage_location = tempfile.mkdtemp(dir=os.environ.get('DJANGO_TEST_TEMP_DIR'))
|
|
temp_storage = FileSystemStorage(location=temp_storage_location)
|
|
|
|
|
|
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)
|
|
|
|
def test_reset_deletes_tmp_files(self):
|
|
request = get_request()
|
|
storage = self.get_storage()('wizard1', request, temp_storage)
|
|
|
|
step = 'start'
|
|
file_ = SimpleUploadedFile('file.txt', b'content')
|
|
storage.set_step_files(step, {'file': file_})
|
|
|
|
with storage.get_step_files(step)['file'] as file:
|
|
tmp_name = file.name
|
|
|
|
self.assertTrue(storage.file_storage.exists(tmp_name))
|
|
|
|
storage.reset()
|
|
storage.update_response(HttpResponse())
|
|
self.assertFalse(storage.file_storage.exists(tmp_name))
|