add some tests

tests/dummy.py

@@ -1,3 +1,4 @@
+import functools
 from cas_server import models
 
 class DummyUserManager(object):
@@ -10,6 +11,75 @@ class DummyUserManager(object):
         else:
             raise models.User.DoesNotExist()
 
+
+def dummy(*args, **kwds):
+    pass
+
+def dummy_service_pattern(**kwargs):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwds):
+            service_validate = models.ServicePattern.validate
+            models.ServicePattern.validate = classmethod(lambda x,y: models.ServicePattern(**kwargs))
+            ret = func(*args, **kwds)
+            models.ServicePattern.validate = service_validate
+            return ret
+        return wrapper
+    return decorator
+
+def dummy_user(username, session_key):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwds):
+            user_manager = models.User.objects
+            user_save = models.User.save
+            user_delete = models.User.delete
+            models.User.objects = DummyUserManager(username, session_key)
+            models.User.save = dummy
+            models.User.delete = dummy
+            ret = func(*args, **kwds)
+            models.User.objects = user_manager
+            models.User.save = user_save
+            models.User.delete = user_delete
+            return ret
+        return wrapper
+    return decorator
+
+def dummy_ticket(ticket_class, service, ticket):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwds):
+            ticket_manager = ticket_class.objects
+            ticket_save = ticket_class.save
+            ticket_delete = ticket_class.delete
+            ticket_class.objects = DummyTicketManager(ticket_class, service, ticket)
+            ticket_class.save = dummy
+            ticket_class.delete = dummy
+            ret = func(*args, **kwds)
+            ticket_class.objects = ticket_manager
+            ticket_class.save = ticket_save
+            ticket_class.delete = ticket_delete
+            return ret
+        return wrapper
+    return decorator
+
+
+def dummy_proxy(func):
+    @functools.wraps(func)
+    def wrapper(*args, **kwds):
+        proxy_manager = models.Proxy.objects
+        models.Proxy.objects = DummyProxyManager()
+        ret = func(*args, **kwds)
+        models.Proxy.objects = proxy_manager
+        return ret
+    return wrapper
+
+class DummyProxyManager(object):
+    def create(self, **kwargs):
+        for field in models.Proxy._meta.fields:
+            field.allow_unsaved_instance_assignment = True
+        return models.Proxy(**kwargs)
+
 class DummyTicketManager(object):
     def __init__(self, ticket_class, service, ticket):
         self.ticket_class = ticket_class
@@ -17,7 +87,7 @@ class DummyTicketManager(object):
         self.ticket = ticket
 
     def create(self, **kwargs):
-        for field in models.ServiceTicket._meta.fields:
+        for field in self.ticket_class._meta.fields:
             field.allow_unsaved_instance_assignment = True
         return self.ticket_class(**kwargs)
 
@@ -25,6 +95,8 @@ class DummyTicketManager(object):
         return DummyQuerySet()
 
     def get(self, **kwargs):
+        for field in self.ticket_class._meta.fields:
+            field.allow_unsaved_instance_assignment = True
         if 'value' in kwargs:
             if kwargs['value'] != self.ticket:
                 raise self.ticket_class.DoesNotExist()
@@ -41,7 +113,7 @@ class DummyTicketManager(object):
         
         for field in models.ServiceTicket._meta.fields:
             field.allow_unsaved_instance_assignment = True
-        for key in kwargs.keys():
+        for key in list(kwargs):
             if '__' in key:
                 del kwargs[key]
         kwargs['attributs'] = {'mail': 'test@example.com'}