Changeset 10351

Timestamp:

22 Jun 2013, 21:47:04 (12 years ago)

Author:

uli

Message:

Add a db_path option for CAS server.

Location:

main/waeup.cas/trunk/waeup/cas

Files:

• server.py (modified) (2 diffs)
• tests/sample2.ini (copied) (copied from main/waeup.cas/trunk/waeup/cas/tests/sample1.ini)
• tests/test_server.py (modified) (1 diff)

main/waeup.cas/trunk/waeup/cas/server.py

@@ -2 +2 @@
 """
 import os
+import tempfile
 from webob import exc, Response
 from webob.dec import wsgify
@@ -10 +11 @@
 class CASServer(object):
     """A WSGI CAS server.
+
+    This CAS server stores credential data (tickets, etc.) in a local
+    sqlite3 database file.
+
+    `db_path` -
+       The filesystem path to the database to use. If none is given
+       CAS server will create a new one in some new, temporary
+       directory. Please note that credentials will be lost after a
+       CAS server restart.
+
+       If the path is given and the file exists already, it will be
+       used.
+
+       If the database file does not exist, it will be created.
     """
+    def __init__(self, db_path=None):
+        if db_path is None:
+            db_path = os.path.join(tempfile.mkdtemp(), 'cas.db')
+        self.db_path = db_path
+
     @wsgify
     def __call__(self, req):

main/waeup.cas/trunk/waeup/cas/tests/test_server.py

@@ -38 +38 @@
         assert isinstance(app, CASServer)
 
+    def test_init(self):
+        # we get a db dir set automatically
+        app = CASServer()
+        assert hasattr(app, 'db_path')
+        assert app.db_path is not None
+        assert os.path.exists(os.path.dirname(app.db_path))
+
+    def test_init_explicit_db_path(self):
+        # we can set a db_path explicitly
+        app = CASServer(db_path=self.db_path)
+        assert hasattr(app, 'db_path')
+        assert app.db_path == self.db_path
+
     def test_call_root(self):
         # the CAS protocol requires no root