62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
|
from cas_server import models
|
||
|
|
|
||
|
|
class DummyUserManager(object):
|
||
|
|
def __init__(self, username, session_key):
|
||
|
|
self.username = username
|
||
|
|
self.session_key = session_key
|
||
|
|
def get(self, username=None, session_key=None):
|
||
|
|
if username == self.username and session_key == self.session_key:
|
||
|
|
return models.User(username=username, session_key=session_key)
|
||
|
|
else:
|
||
|
|
raise models.User.DoesNotExist()
|
||
|
|
|
||
|
|
class DummyTicketManager(object):
|
||
|
|
def __init__(self, ticket_class, service, ticket):
|
||
|
|
self.ticket_class = ticket_class
|
||
|
|
self.service = service
|
||
|
|
self.ticket = ticket
|
||
|
|
|
||
|
|
def create(self, **kwargs):
|
||
|
|
for field in models.ServiceTicket._meta.fields:
|
||
|
|
field.allow_unsaved_instance_assignment = True
|
||
|
|
return self.ticket_class(**kwargs)
|
||
|
|
|
||
|
|
def filter(self, *args, **kwargs):
|
||
|
|
return DummyQuerySet()
|
||
|
|
|
||
|
|
def get(self, **kwargs):
|
||
|
|
if 'value' in kwargs:
|
||
|
|
if kwargs['value'] != self.ticket:
|
||
|
|
raise self.ticket_class.DoesNotExist()
|
||
|
|
else:
|
||
|
|
kwargs['value'] = self.ticket
|
||
|
|
|
||
|
|
if 'service' in kwargs:
|
||
|
|
if kwargs['service'] != self.service:
|
||
|
|
raise self.ticket_class.DoesNotExist()
|
||
|
|
else:
|
||
|
|
kwargs['service'] = self.service
|
||
|
|
if not 'user' in kwargs:
|
||
|
|
kwargs['user'] = models.User(username="test")
|
||
|
|
|
||
|
|
for field in models.ServiceTicket._meta.fields:
|
||
|
|
field.allow_unsaved_instance_assignment = True
|
||
|
|
for key in kwargs.keys():
|
||
|
|
if '__' in key:
|
||
|
|
del kwargs[key]
|
||
|
|
kwargs['attributs'] = {'mail': 'test@example.com'}
|
||
|
|
kwargs['service_pattern'] = models.ServicePattern()
|
||
|
|
return self.ticket_class(**kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class DummySession(dict):
|
||
|
|
session_key = "test_session"
|
||
|
|
|
||
|
|
def set_expiry(self, int):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class DummyQuerySet(set):
|
||
|
|
pass
|