def test_single_sign_on_receives_correct_token():
mock_sso_registry = Mock(SingleSignOnRegistry)
correct_token = SSOToken()
mock_sso_registry.is_valid = Mock(side_effect=confirm_token(correct_token))
service = MyService(mock_sso_registry)
service.handle(Request("Emily"), correct_token)
mock_sso_registry.is_valid.assert_called()
def test_single_sign_on_with_invalid_token():
spy_sso_registry = Mock(SingleSignOnRegistry)
spy_sso_registry.is_valid.return_value = False
service = MyService(spy_sso_registry)
token = SSOToken()
response = service.handle(Request("Emily"), token)
spy_sso_registry.is_valid.assert_called_with(token)
assert response.text == "Please sign in"
def test_hello_name():
stub_sso_registry = Mock(SingleSignOnRegistry)
service = MyService(stub_sso_registry)
response = service.handle(Request("Emily"), SSOToken())
assert response.text == "Hello Emily!"
def test_single_sign_on():
spy_sso_registry = Mock(SingleSignOnRegistry)
service = MyService(spy_sso_registry)
token = SSOToken()
service.handle(Request("Emily"), token)
spy_sso_registry.is_valid.assert_called_with(token)