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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
# Copyright 2017 Canonical Ltd.
# Licensed under the LGPLv3, see LICENCE file for details.
import unittest
import macaroonbakery.bakery as bakery
import macaroonbakery.httpbakery as httpbakery
from httmock import HTTMock, urlmatch
class TestKeyRing(unittest.TestCase):
def test_cache_fetch(self):
key = bakery.generate_key()
@urlmatch(path='.*/discharge/info')
def discharge_info(url, request):
return {
'status_code': 200,
'content': {
'Version': bakery.LATEST_VERSION,
'PublicKey': str(key.public_key),
}
}
expectInfo = bakery.ThirdPartyInfo(
public_key=key.public_key,
version=bakery.LATEST_VERSION
)
kr = httpbakery.ThirdPartyLocator(allow_insecure=True)
with HTTMock(discharge_info):
info = kr.third_party_info('http://0.1.2.3/')
self.assertEqual(info, expectInfo)
def test_cache_norefetch(self):
key = bakery.generate_key()
@urlmatch(path='.*/discharge/info')
def discharge_info(url, request):
return {
'status_code': 200,
'content': {
'Version': bakery.LATEST_VERSION,
'PublicKey': str(key.public_key),
}
}
expectInfo = bakery.ThirdPartyInfo(
public_key=key.public_key,
version=bakery.LATEST_VERSION
)
kr = httpbakery.ThirdPartyLocator(allow_insecure=True)
with HTTMock(discharge_info):
info = kr.third_party_info('http://0.1.2.3/')
self.assertEqual(info, expectInfo)
info = kr.third_party_info('http://0.1.2.3/')
self.assertEqual(info, expectInfo)
def test_cache_fetch_no_version(self):
key = bakery.generate_key()
@urlmatch(path='.*/discharge/info')
def discharge_info(url, request):
return {
'status_code': 200,
'content': {
'PublicKey': str(key.public_key),
}
}
expectInfo = bakery.ThirdPartyInfo(
public_key=key.public_key,
version=bakery.VERSION_1
)
kr = httpbakery.ThirdPartyLocator(allow_insecure=True)
with HTTMock(discharge_info):
info = kr.third_party_info('http://0.1.2.3/')
self.assertEqual(info, expectInfo)
def test_allow_insecure(self):
kr = httpbakery.ThirdPartyLocator()
with self.assertRaises(bakery.ThirdPartyInfoNotFound):
kr.third_party_info('http://0.1.2.3/')
def test_fallback(self):
key = bakery.generate_key()
@urlmatch(path='.*/discharge/info')
def discharge_info(url, request):
return {
'status_code': 404,
}
@urlmatch(path='.*/publickey')
def public_key(url, request):
return {
'status_code': 200,
'content': {
'PublicKey': str(key.public_key),
}
}
expectInfo = bakery.ThirdPartyInfo(
public_key=key.public_key,
version=bakery.VERSION_1
)
kr = httpbakery.ThirdPartyLocator(allow_insecure=True)
with HTTMock(discharge_info):
with HTTMock(public_key):
info = kr.third_party_info('http://0.1.2.3/')
self.assertEqual(info, expectInfo)
|