Changeset 10398 for main/waeup.cas/trunk/waeup/cas/tests

Timestamp:

4 Jul 2013, 22:08:53 (11 years ago)

Author:

uli

Message:

Enable session cookies for single sign on.

File:

• main/waeup.cas/trunk/waeup/cas/tests/test_server.py (modified) (3 diffs)

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

@@ -12 +12 @@
     CASServer, create_service_ticket, create_login_ticket,
     create_tgc_value, check_login_ticket, set_session_cookie,
+    check_session_cookie,
     )
-
 
 RE_ALPHABET = re.compile('^[a-zA-Z0-9\-]*$')
@@ -184 +184 @@
         assert b'ticket=ST-' in resp.body
 
+    def test_login_cred_acceptor_sso_no_service(self):
+        # 2.2.4: successful login via single sign on
+        db = self.raw_app.db
+        tgc = create_tgc_value()
+        db.add(tgc)
+        value = str(tgc.value)
+        resp = self.app.get('/login', headers={'Cookie': 'cas-tgc=%s' % value})
+        assert resp.status == '200 OK'
+        assert b'already' in resp.body
+        assert 'Set-Cookie' not in resp.headers
+        return
+
     def test_login_cred_acceptor_failed(self):
         # 2.2.4: failed login yields a message
@@ -269 +281 @@
             TicketGrantingCookie.value == value)
         assert len(list(q)) == 1
+
+    def test_check_session_cookie(self):
+        db = DB('sqlite:///')
+        tgc = create_tgc_value()
+        db.add(tgc)
+        value = tgc.value
+        assert check_session_cookie(db, value) == tgc
+        assert check_session_cookie(db, 'foo') is None
+        assert check_session_cookie(db, b'foo') is None
+        value2 = value.encode('utf-8')
+        assert check_session_cookie(db, value2) == tgc