| commit | author | age | ||
| b60bdb | 1 | from pyramid import testing |
| a1a9fb | 2 | import unittest |
| CM | 3 | |
| 0c29cf | 4 | |
| a1a9fb | 5 | class TestThreadLocalManager(unittest.TestCase): |
| CM | 6 | def setUp(self): |
| cbfafb | 7 | testing.setUp() |
| a1a9fb | 8 | |
| CM | 9 | def tearDown(self): |
| cbfafb | 10 | testing.tearDown() |
| a1a9fb | 11 | |
| CM | 12 | def _getTargetClass(self): |
| b60bdb | 13 | from pyramid.threadlocal import ThreadLocalManager |
| 0c29cf | 14 | |
| a1a9fb | 15 | return ThreadLocalManager |
| CM | 16 | |
| 17 | def _makeOne(self, default=lambda *x: 1): | |
| 18 | return self._getTargetClass()(default) | |
| 19 | ||
| 20 | def test_init(self): | |
| 21 | local = self._makeOne() | |
| 22 | self.assertEqual(local.stack, []) | |
| 23 | self.assertEqual(local.get(), 1) | |
| 24 | ||
| 25 | def test_default(self): | |
| 17b8bc | 26 | def thedefault(): |
| CM | 27 | return '123' |
| 0c29cf | 28 | |
| 17b8bc | 29 | local = self._makeOne(thedefault) |
| a1a9fb | 30 | self.assertEqual(local.stack, []) |
| 17b8bc | 31 | self.assertEqual(local.get(), '123') |
| a1a9fb | 32 | |
| CM | 33 | def test_push_and_pop(self): |
| 34 | local = self._makeOne() | |
| 35 | local.push(True) | |
| 36 | self.assertEqual(local.get(), True) | |
| 37 | self.assertEqual(local.pop(), True) | |
| 38 | self.assertEqual(local.pop(), None) | |
| 39 | self.assertEqual(local.get(), 1) | |
| 40 | ||
| 41 | def test_set_get_and_clear(self): | |
| 42 | local = self._makeOne() | |
| 43 | local.set(None) | |
| 44 | self.assertEqual(local.stack, [None]) | |
| 45 | self.assertEqual(local.get(), None) | |
| 46 | local.clear() | |
| 47 | self.assertEqual(local.get(), 1) | |
| 48 | local.clear() | |
| 49 | self.assertEqual(local.get(), 1) | |
| 50 | ||
| 947b8b | 51 | |
| CM | 52 | class TestGetCurrentRequest(unittest.TestCase): |
| 53 | def _callFUT(self): | |
| b60bdb | 54 | from pyramid.threadlocal import get_current_request |
| 0c29cf | 55 | |
| 947b8b | 56 | return get_current_request() |
| CM | 57 | |
| 58 | def test_it_None(self): | |
| 59 | request = self._callFUT() | |
| 60 | self.assertEqual(request, None) | |
| 61 | ||
| 62 | def test_it(self): | |
| b60bdb | 63 | from pyramid.threadlocal import manager |
| 0c29cf | 64 | |
| 947b8b | 65 | request = object() |
| CM | 66 | try: |
| 0c29cf | 67 | manager.push({'request': request}) |
| 947b8b | 68 | self.assertEqual(self._callFUT(), request) |
| CM | 69 | finally: |
| 70 | manager.pop() | |
| 71 | self.assertEqual(self._callFUT(), None) | |
| 0c29cf | 72 | |
| 947b8b | 73 | |
| CM | 74 | class GetCurrentRegistryTests(unittest.TestCase): |
| 75 | def setUp(self): | |
| cbfafb | 76 | testing.setUp() |
| 947b8b | 77 | |
| CM | 78 | def tearDown(self): |
| cbfafb | 79 | testing.tearDown() |
| 0c29cf | 80 | |
| 947b8b | 81 | def _callFUT(self): |
| b60bdb | 82 | from pyramid.threadlocal import get_current_registry |
| 0c29cf | 83 | |
| 947b8b | 84 | return get_current_registry() |
| CM | 85 | |
| cbfafb | 86 | def test_it(self): |
| b60bdb | 87 | from pyramid.threadlocal import manager |
| 0c29cf | 88 | |
| 947b8b | 89 | try: |
| 0c29cf | 90 | manager.push({'registry': 123}) |
| 947b8b | 91 | self.assertEqual(self._callFUT(), 123) |
| CM | 92 | finally: |
| 93 | manager.pop() | |
| cbfafb | 94 | |
| 0c29cf | 95 | |
| cbfafb | 96 | class GetCurrentRegistryWithoutTestingRegistry(unittest.TestCase): |
| CM | 97 | def _callFUT(self): |
| b60bdb | 98 | from pyramid.threadlocal import get_current_registry |
| 0c29cf | 99 | |
| cbfafb | 100 | return get_current_registry() |
| CM | 101 | |
| 102 | def test_it(self): | |
| b60bdb | 103 | from pyramid.registry import global_registry |
| 0c29cf | 104 | |
| 59c5df | 105 | self.assertEqual(self._callFUT(), global_registry) |
