Fix tests for refresh_token removal

matrix-org · Nov 1, 2017 · 02237ce · 02237ce
1 parent 356bcaf
commit 02237ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/storage/test_registration.py b/tests/storage/test_registration.py
@@ -86,7 +86,8 @@ def test_user_delete_access_tokens(self):
 
         # now delete some
         yield self.store.user_delete_access_tokens(
-            self.user_id, device_id=self.device_id, delete_refresh_tokens=True)
+            self.user_id, device_id=self.device_id,
+        )
 
         # check they were deleted
         user = yield self.store.get_user_by_access_token(self.tokens[1])
@@ -97,8 +98,7 @@ def test_user_delete_access_tokens(self):
         self.assertEqual(self.user_id, user["name"])
 
         # now delete the rest
-        yield self.store.user_delete_access_tokens(
-            self.user_id, delete_refresh_tokens=True)
+        yield self.store.user_delete_access_tokens(self.user_id)
 
         user = yield self.store.get_user_by_access_token(self.tokens[0])
         self.assertIsNone(user,