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