fix test for py3

pull/3630/head
Neil Johnson 2018-08-01 14:02:10 +01:00
parent 303f1c851f
commit 6eed16d8a2
1 changed files with 4 additions and 0 deletions

View File

@ -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()
)