38 lines
840 B
Python
38 lines
840 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import cas
|
||
|
|
|
||
|
|
|
||
|
|
class MockCASClient(cas.CASClientV2):
|
||
|
|
"""
|
||
|
|
Base class to mock cas.CASClient
|
||
|
|
"""
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
kwargs.pop('version')
|
||
|
|
super(MockCASClient, self).__init__(*args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
class VerifyCASClient(MockCASClient):
|
||
|
|
"""
|
||
|
|
CAS client which verifies ticket is '123456'.
|
||
|
|
"""
|
||
|
|
def verify_ticket(self, ticket):
|
||
|
|
if ticket == '123456':
|
||
|
|
return 'username', {}, None
|
||
|
|
return None, {}, None
|
||
|
|
|
||
|
|
|
||
|
|
class AcceptCASClient(MockCASClient):
|
||
|
|
"""
|
||
|
|
CAS client which accepts all tickets.
|
||
|
|
"""
|
||
|
|
def verify_ticket(self, ticket):
|
||
|
|
return 'username', {}, None
|
||
|
|
|
||
|
|
|
||
|
|
class RejectCASClient(MockCASClient):
|
||
|
|
"""
|
||
|
|
CAS client which rejects all tickets.
|
||
|
|
"""
|
||
|
|
def verify_ticket(self, ticket):
|
||
|
|
return None, {}, None
|