fix test for py3

1 files changed, 4 insertions, 0 deletions

diff --git a/tests/handlers/test_auth.py b/tests/handlers/test_auth.py

index 440a453082..55eab9e9cf 100644
--- a/tests/handlers/test_auth.py
+++ b/tests/handlers/test_auth.py
@@ -156,6 +156,10 @@ class AuthTestCase(unittest.TestCase):
         )
         # Ensure does not raise exception
         yield self.auth_handler.get_access_token_for_user_id('user_a')
+
+        self.hs.get_datastore().count_monthly_users = Mock(
+            return_value=defer.succeed(self.small_number_of_users)
+        )
         yield self.auth_handler.validate_short_term_login_token_and_get_user_id(
             self._get_macaroon().serialize()
         )