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.assertEquals(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.assertEquals(1, ender.end_count)
|