Add register -> login system test.

jslvtr · jslvtr · commit 178f6fcfea9e · 2017-04-16T12:24:47.000+01:00
diff --git a/section4/video_code/tests/system/user_test.py b/section4/video_code/tests/system/user_test.py
@@ -14,6 +14,17 @@ def test_register_user(self):
                 self.assertDictEqual(d1={'message': 'User created successfully.'},
                                      d2=json.loads(r.data))
 
+    def test_register_and_login(self):
+        with self.app() as c:
+            with self.app_context():
+                c.post('/register', data={'username': 'test', 'password': '1234'})
+                auth_request = c.post('/auth', data=json.dumps({
+                    'username': 'test',
+                    'password': '1234'
+                }), headers={'Content-Type': 'application/json'})
+
+                self.assertIn('access_token', json.loads(auth_request.data).keys())
+
     def test_register_duplicate_user(self):
         with self.app() as c:
             with self.app_context():
