1 files changed, 133 insertions, 0 deletions
diff --git a/websdk/werkzeug/testsuite/local.py b/websdk/werkzeug/testsuite/local.py
new file mode 100644
index 0000000..236e753
--- /dev/null
+++ b/websdk/werkzeug/testsuite/local.py
@@ -0,0 +1,133 @@
+# -*- coding: utf-8 -*-
+"""
+    werkzeug.testsuite.local
+    ~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Local and local proxy tests.
+
+    :copyright: (c) 2011 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+import time
+import unittest
+from threading import Thread
+
+from werkzeug.testsuite import WerkzeugTestCase
+
+from werkzeug import local
+
+
+class LocalTestCase(WerkzeugTestCase):
+
+    def test_basic_local(self):
+        l = local.Local()
+        l.foo = 0
+        values = []
+        def value_setter(idx):
+            time.sleep(0.01 * idx)
+            l.foo = idx
+            time.sleep(0.02)
+            values.append(l.foo)
+        threads = [Thread(target=value_setter, args=(x,))
+                   for x in [1, 2, 3]]
+        for thread in threads:
+            thread.start()
+        time.sleep(0.2)
+        assert sorted(values) == [1, 2, 3]
+
+        def delfoo():
+            del l.foo
+        delfoo()
+        self.assert_raises(AttributeError, lambda: l.foo)
+        self.assert_raises(AttributeError, delfoo)
+
+        local.release_local(l)
+
+    def test_local_release(self):
+        loc = local.Local()
+        loc.foo = 42
+        local.release_local(loc)
+        assert not hasattr(loc, 'foo')
+
+        ls = local.LocalStack()
+        ls.push(42)
+        local.release_local(ls)
+        assert ls.top is None
+
+    def test_local_proxy(self):
+        foo = []
+        ls = local.LocalProxy(lambda: foo)
+        ls.append(42)
+        ls.append(23)
+        ls[1:] = [1, 2, 3]
+        assert foo == [42, 1, 2, 3]
+        assert repr(foo) == repr(ls)
+        assert foo[0] == 42
+        foo += [1]
+        assert list(foo) == [42, 1, 2, 3, 1]
+
+    def test_local_stack(self):
+        ident = local.get_ident()
+
+        ls = local.LocalStack()
+        assert ident not in ls._local.__storage__
+        assert ls.top is None
+        ls.push(42)
+        assert ident in ls._local.__storage__
+        assert ls.top == 42
+        ls.push(23)
+        assert ls.top == 23
+        ls.pop()
+        assert ls.top == 42
+        ls.pop()
+        assert ls.top is None
+        assert ls.pop() is None
+        assert ls.pop() is None
+
+        proxy = ls()
+        ls.push([1, 2])
+        assert proxy == [1, 2]
+        ls.push((1, 2))
+        assert proxy == (1, 2)
+        ls.pop()
+        ls.pop()
+        assert repr(proxy) == '<LocalProxy unbound>'
+
+        assert ident not in ls._local.__storage__
+
+    def test_local_proxies_with_callables(self):
+        foo = 42
+        ls = local.LocalProxy(lambda: foo)
+        assert ls == 42
+        foo = [23]
+        ls.append(42)
+        assert ls == [23, 42]
+        assert foo == [23, 42]
+
+    def test_custom_idents(self):
+        ident = 0
+        loc = local.Local()
+        stack = local.LocalStack()
+        mgr = local.LocalManager([loc, stack], ident_func=lambda: ident)
+
+        loc.foo = 42
+        stack.push({'foo': 42})
+        ident = 1
+        loc.foo = 23
+        stack.push({'foo': 23})
+        ident = 0
+        assert loc.foo == 42
+        assert stack.top['foo'] == 42
+        stack.pop()
+        assert stack.top is None
+        ident = 1
+        assert loc.foo == 23
+        assert stack.top['foo'] == 23
+        stack.pop()
+        assert stack.top is None
+
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(LocalTestCase))
+    return suite