def test_record_access(self):
app_user = AppUser.for_user(users.get_current_user())
first_access_date = app_user.first_access
first_latest_date = app_user.latest_access
time.sleep(0.1)
AppUser.record_access(users.get_current_user())
app_user = AppUser.for_user(users.get_current_user())
self.assertEqual(first_access_date, app_user.first_access)
self.assertTrue(app_user.latest_access > first_latest_date)
def mock_user(monkeypatch, client, roles):
user = AppUser('someUserId', roles)
monkeypatch.setattr('auth_provider.AuthProvider.get_authenticated_user',
lambda self, auth_code: user)
monkeypatch.setattr('auth_provider.AuthProvider.get_user',
lambda self, user_id: user)
# Invoke login callback to simulate successful login
client.get('/login/callback?code=someUserId')
def get_user(self, user_id):
return AppUser(user_id, self.get_roles(user_id))
def test_create(self):
app_user = AppUser.for_user(users.get_current_user())
self.assertEqual('*****@*****.**', app_user.email)
self.assertEqual('123', app_user.user_id)
def test_for_user(self):
first_app_user = AppUser.for_user(users.get_current_user())
first_id = first_app_user.key().id()
second_app_user = AppUser.for_user(users.get_current_user())
self.assertEqual(first_id, second_app_user.key().id())
