blob: 230bb3726e9be1290696b9f01ade4b4f8058a2bb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
from repoze.bfg.testing import cleanUp
import unittest
class TestThreadLocalManager(unittest.TestCase):
def setUp(self):
cleanUp()
def tearDown(self):
cleanUp()
def _getTargetClass(self):
from repoze.bfg.threadlocal import ThreadLocalManager
return ThreadLocalManager
def _makeOne(self, default=lambda *x: 1):
return self._getTargetClass()(default)
def test_init(self):
local = self._makeOne()
self.assertEqual(local.stack, [])
self.assertEqual(local.get(), 1)
def test_default(self):
from zope.component import getGlobalSiteManager
local = self._makeOne(getGlobalSiteManager)
self.assertEqual(local.stack, [])
self.assertEqual(local.get(), getGlobalSiteManager())
def test_push_and_pop(self):
local = self._makeOne()
local.push(True)
self.assertEqual(local.get(), True)
self.assertEqual(local.pop(), True)
self.assertEqual(local.pop(), None)
self.assertEqual(local.get(), 1)
def test_set_get_and_clear(self):
local = self._makeOne()
local.set(None)
self.assertEqual(local.stack, [None])
self.assertEqual(local.get(), None)
local.clear()
self.assertEqual(local.get(), 1)
local.clear()
self.assertEqual(local.get(), 1)
|
