252697b842
Upgrade site and modules to Django 2.2. Remove and replace obsolete functionality with current equivalents. Update requirements to latest versions where possible. Remove unused dependencies.
137 lines
4.6 KiB
Python
137 lines
4.6 KiB
Python
import pytz
|
|
|
|
from registrasion.tests.test_cart import RegistrationCartTestCase
|
|
|
|
from registrasion.controllers.batch import BatchController
|
|
|
|
UTC = pytz.timezone('UTC')
|
|
|
|
|
|
class BatchTestCase(RegistrationCartTestCase):
|
|
|
|
def test_no_caches_outside_of_batches(self):
|
|
cache_1 = BatchController.get_cache(self.USER_1)
|
|
cache_2 = BatchController.get_cache(self.USER_2)
|
|
|
|
# Identity testing is important here
|
|
self.assertIsNot(cache_1, cache_2)
|
|
|
|
def test_cache_clears_at_batch_exit(self):
|
|
with BatchController.batch(self.USER_1):
|
|
cache_1 = BatchController.get_cache(self.USER_1)
|
|
|
|
cache_2 = BatchController.get_cache(self.USER_1)
|
|
|
|
self.assertIsNot(cache_1, cache_2)
|
|
|
|
def test_caches_identical_within_nestings(self):
|
|
with BatchController.batch(self.USER_1):
|
|
cache_1 = BatchController.get_cache(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_2):
|
|
cache_2 = BatchController.get_cache(self.USER_1)
|
|
|
|
cache_3 = BatchController.get_cache(self.USER_1)
|
|
|
|
self.assertIs(cache_1, cache_2)
|
|
self.assertIs(cache_2, cache_3)
|
|
|
|
def test_caches_are_independent_for_different_users(self):
|
|
with BatchController.batch(self.USER_1):
|
|
cache_1 = BatchController.get_cache(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_2):
|
|
cache_2 = BatchController.get_cache(self.USER_2)
|
|
|
|
self.assertIsNot(cache_1, cache_2)
|
|
|
|
def test_cache_clears_are_independent_for_different_users(self):
|
|
with BatchController.batch(self.USER_1):
|
|
cache_1 = BatchController.get_cache(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_2):
|
|
cache_2 = BatchController.get_cache(self.USER_2)
|
|
|
|
with BatchController.batch(self.USER_2):
|
|
cache_3 = BatchController.get_cache(self.USER_2)
|
|
|
|
cache_4 = BatchController.get_cache(self.USER_1)
|
|
|
|
self.assertIs(cache_1, cache_4)
|
|
self.assertIsNot(cache_1, cache_2)
|
|
self.assertIsNot(cache_2, cache_3)
|
|
|
|
def test_new_caches_for_new_batches(self):
|
|
with BatchController.batch(self.USER_1):
|
|
cache_1 = BatchController.get_cache(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_1):
|
|
cache_2 = BatchController.get_cache(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_1):
|
|
cache_3 = BatchController.get_cache(self.USER_1)
|
|
|
|
self.assertIs(cache_2, cache_3)
|
|
self.assertIsNot(cache_1, cache_2)
|
|
|
|
def test_memoisation_happens_in_batch_context(self):
|
|
with BatchController.batch(self.USER_1):
|
|
output_1 = self._memoiseme(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_1):
|
|
output_2 = self._memoiseme(self.USER_1)
|
|
|
|
self.assertIs(output_1, output_2)
|
|
|
|
def test_memoisaion_does_not_happen_outside_batch_context(self):
|
|
output_1 = self._memoiseme(self.USER_1)
|
|
output_2 = self._memoiseme(self.USER_1)
|
|
|
|
self.assertIsNot(output_1, output_2)
|
|
|
|
def test_memoisation_is_user_independent(self):
|
|
with BatchController.batch(self.USER_1):
|
|
output_1 = self._memoiseme(self.USER_1)
|
|
with BatchController.batch(self.USER_2):
|
|
output_2 = self._memoiseme(self.USER_2)
|
|
output_3 = self._memoiseme(self.USER_1)
|
|
|
|
self.assertIsNot(output_1, output_2)
|
|
self.assertIs(output_1, output_3)
|
|
|
|
def test_memoisation_clears_outside_batches(self):
|
|
with BatchController.batch(self.USER_1):
|
|
output_1 = self._memoiseme(self.USER_1)
|
|
|
|
with BatchController.batch(self.USER_1):
|
|
output_2 = self._memoiseme(self.USER_1)
|
|
|
|
self.assertIsNot(output_1, output_2)
|
|
|
|
@classmethod
|
|
@BatchController.memoise
|
|
def _memoiseme(self, user):
|
|
return object()
|
|
|
|
def test_batch_end_functionality_is_called(self):
|
|
class Ender(object):
|
|
end_count = 0
|
|
|
|
def end_batch(self):
|
|
self.end_count += 1
|
|
|
|
@BatchController.memoise
|
|
def get_ender(user):
|
|
return Ender()
|
|
|
|
# end_batch should get called once on exiting the batch
|
|
with BatchController.batch(self.USER_1):
|
|
ender = get_ender(self.USER_1)
|
|
self.assertEqual(1, ender.end_count)
|
|
|
|
# end_batch should get called once on exiting the batch
|
|
# no matter how deep the object gets cached
|
|
with BatchController.batch(self.USER_1):
|
|
with BatchController.batch(self.USER_1):
|
|
ender = get_ender(self.USER_1)
|
|
self.assertEqual(1, ender.end_count)
|