def test_remove_non_existent_is_silent(): team_id = "team" user_id = "user" expected_token = "foobar" gateway = StubUserTokenStore() expected_user = User(user_id=user_id, team_id=team_id, token=expected_token) unstored_user = User(user_id="foo", team_id="bar", token=None) gateway.store(expected_user) assert gateway.remove(unstored_user) is False
def test_remove_a_user_token(): team_id = "team" user_id = "user" expected_token = "foobar" gateway = StubUserTokenStore() expected_user = User(user_id=user_id, team_id=team_id, token=expected_token) gateway.store(expected_user) gateway.fetch(expected_user) assert gateway.remove(expected_user) is True with pytest.raises(KeyError): gateway.fetch(expected_user)