Mercurial > summit-idp

changeset 13:42093a5a19b4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
some refactorings
author Atul Varma <avarma@mozilla.com>
date Thu, 24 Jun 2010 18:49:27 -0700
parents 1fb8071162fd
children 667ebf2a5e8b
files summitidp/app.py
diffstat 1 files changed, 16 insertions(+), 7 deletions(-) [+]
line wrap: on
line diff
--- a/summitidp/app.py	Thu Jun 24 18:45:33 2010 -0700
+++ b/summitidp/app.py	Thu Jun 24 18:49:27 2010 -0700
@@ -106,14 +106,23 @@
     def __init__(self, emails, send_email, gentoken=gentoken,
                  challenge_lifetime=DEFAULT_CHALLENGE_LIFETIME,
                  utcnow=datetime.datetime.utcnow,
-                 delegate_404s=None):
+                 delegate_404s=None,
+                 challenge_tokens=None,
+                 auth_tokens=None,
+                 profiles=None):
+        if challenge_tokens is None:
+            challenge_tokens = TokenStore(lifetime=challenge_lifetime,
+                                          utcnow=utcnow,
+                                          gentoken=gentoken)
+        if auth_tokens is None:
+            auth_tokens = TokenStore(utcnow=utcnow, gentoken=gentoken)
+        if profiles is None:
+            profiles = ProfileStore()
+
         self.send_email = send_email
-        self.challenge_tokens = TokenStore(lifetime=challenge_lifetime,
-                                           utcnow=utcnow,
-                                           gentoken=gentoken)
-        self.auth_tokens = TokenStore(utcnow=utcnow, gentoken=gentoken)
-        self.oauth2_consumers = TokenStore(utcnow=utcnow, gentoken=gentoken)
-        self.profiles = ProfileStore()
+        self.challenge_tokens = challenge_tokens
+        self.auth_tokens = auth_tokens
+        self.profiles = profiles
         self.emails = emails
         self.delegate_404s = delegate_404s