Store userid to simplepush mapping (in a memory db for now). Fix 971993

development.ini

@@ -1,5 +1,6 @@
 [app:main]
 token-secret = this is not a secret
+storage = pants.storage.memory.MemoryStorage
 
 use = egg:pants
 

pants/__init__.py

@@ -18,8 +18,12 @@ def main(global_config, **settings):
     token_manager = TokenManager(secret=settings['token-secret'])
     config.registry.token_manager = token_manager
 
-    def add_db_to_request(event):
+    backend_class = config.maybe_dotted(settings['storage'])
+    config.registry.storage = storage = backend_class(config)
+
+    def attach_services_to_request(event):
         event.request.token_manager = token_manager
-    config.add_subscriber(add_db_to_request, NewRequest)
+        event.request.storage = storage
+    config.add_subscriber(attach_services_to_request, NewRequest)
 
     return config.make_wsgi_app()

pants/views.py

@@ -33,6 +33,11 @@ def generate_callurl(request):
     Generate a callurl based on user ID.
     """
     userid = authenticated_userid(request)
+
+    # We need to try/except here in case the db fails.
+    request.storage.store_simplepush_url(
+        userid, request.validated['simple-push-url'])
+
     token = request.token_manager.make_token({"userid": userid})
     call_url = '{root}/call/{token}'.format(root=request.application_url,
                                             token=token)

tests/test_functional.py

@@ -14,6 +14,7 @@ class FunctionalTest(unittest.TestCase):
     def setUp(self):
         self.app = webtest.TestApp("config:tests.ini", relative_to=__HERE__)
         self.token_manager = self.app.app.registry.token_manager
+        self.storage = self.app.app.registry.storage
         self.simple_push_url = 'https://token.services.mozilla.org'
 
     # POST /call-url
@@ -62,8 +63,13 @@ def test_token_creation_requires_authn(self):
         auth_header = resp.headers.get('WWW-Authenticate')
         self.assertEquals(auth_header, 'Hawk', auth_header)
 
-    def test_token_creation_validates_authn(self):
-        pass
+    @sign_requests(user='n1k0')
+    def test_token_creation_stores_push_url(self):
+        self.app.post_json('/call-url', {
+            'simple-push-url': self.simple_push_url
+        }, status=200)
+        self.assertIn(self.simple_push_url,
+                      self.storage.get_simplepush_urls('n1k0'))
 
     # GET /call/<token>
     def test_invalid_callurl_token_returns_400(self):

tests/tests.ini

@@ -1,5 +1,6 @@
 [app:main]
 token-secret = this is not a secret
+storage = pants.storage.memory.MemoryStorage
 
 use = egg:pants