Repository navigation
Expand file tree
/
Copy pathtest_my_service.py
More file actions
41 lines (34 loc) · 1.47 KB
/
Copy pathtest_my_service.py
File metadata and controls
41 lines (34 loc) · 1.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from unittest.mock import Mock
from my_service import MyService, Request
from single_sign_on import SSOToken, SingleSignOnRegistry
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)
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_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 confirm_token(correct_token):
def is_valid(actual_token):
if actual_token != correct_token:
raise ValueError('wrong token received')
return is_valid