1 files changed, 81 insertions, 0 deletions
diff --git a/websdk/werkzeug/testsuite/contrib/sessions.py b/websdk/werkzeug/testsuite/contrib/sessions.py
new file mode 100644
index 0000000..18015a6
--- /dev/null
+++ b/websdk/werkzeug/testsuite/contrib/sessions.py
@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+"""
+    werkzeug.testsuite.sessions
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Added tests for the sessions.
+
+    :copyright: (c) 2011 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+import unittest
+import shutil
+
+from werkzeug.testsuite import WerkzeugTestCase
+
+from werkzeug.contrib.sessions import FilesystemSessionStore
+
+from tempfile import mkdtemp, gettempdir
+
+
+class SessionTestCase(WerkzeugTestCase):
+
+    def setup(self):
+        self.session_folder = mkdtemp()
+
+    def teardown(self):
+        shutil.rmtree(self.session_folder)
+
+    def test_default_tempdir(self):
+        store = FilesystemSessionStore()
+        assert store.path == gettempdir()
+
+    def test_basic_fs_sessions(self):
+        store = FilesystemSessionStore(self.session_folder)
+        x = store.new()
+        assert x.new
+        assert not x.modified
+        x['foo'] = [1, 2, 3]
+        assert x.modified
+        store.save(x)
+
+        x2 = store.get(x.sid)
+        assert not x2.new
+        assert not x2.modified
+        assert x2 is not x
+        assert x2 == x
+        x2['test'] = 3
+        assert x2.modified
+        assert not x2.new
+        store.save(x2)
+
+        x = store.get(x.sid)
+        store.delete(x)
+        x2 = store.get(x.sid)
+        # the session is not new when it was used previously.
+        assert not x2.new
+
+    def test_renewing_fs_session(self):
+        store = FilesystemSessionStore(self.session_folder, renew_missing=True)
+        x = store.new()
+        store.save(x)
+        store.delete(x)
+        x2 = store.get(x.sid)
+        assert x2.new
+
+    def test_fs_session_lising(self):
+        store = FilesystemSessionStore(self.session_folder, renew_missing=True)
+        sessions = set()
+        for x in xrange(10):
+            sess = store.new()
+            store.save(sess)
+            sessions.add(sess.sid)
+
+        listed_sessions = set(store.list())
+        assert sessions == listed_sessions
+
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(SessionTestCase))
+    return suite