Fixed a test, applied a new style of adding attributes

3mdeb · Nov 25, 2017 · 4d2f85d · 4d2f85d
1 parent 18f2a04
commit 4d2f85d
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/warrant/tests/tests.py b/warrant/tests/tests.py
@@ -90,15 +90,16 @@ def test_verify_token(self):
     #     self.assertEqual(self.user.access_token,None)
 
     @patch('warrant.Cognito', autospec=True)
-    def test_register(self,cognito_user):
+    def test_register(self, cognito_user):
         u = cognito_user(self.cognito_user_pool_id, self.app_id,
                          username=self.username)
-        res = u.register('sampleuser','sample4#Password',
-                given_name='Brian',family_name='Jones',
-                name='Brian Jones',
-                email='[email protected]',
-                phone_number='+19194894555',gender='Male',
-                preferred_username='billyocean')
+        u.add_base_attributes(
+            given_name='Brian', family_name='Jones',
+            name='Brian Jones', email='[email protected]',
+            phone_number='+19194894555', gender='Male',
+            preferred_username='billyocean')
+        res = u.register('sampleuser', 'sample4#Password')
+
         #TODO: Write assumptions