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
|
# Copyright 2017 Canonical Ltd.
# Licensed under the LGPLv3, see LICENCE file for details.
import json
from datetime import datetime
from unittest import TestCase
import macaroonbakery.bakery as bakery
import pymacaroons
import pytz
from macaroonbakery._utils import cookie
from pymacaroons.serializers import json_serializer
class CookieTest(TestCase):
def test_cookie_expires_naive(self):
timestamp = datetime.utcnow()
c = cookie('http://example.com', 'test', 'value', expires=timestamp)
self.assertEqual(
c.expires, int((timestamp - datetime(1970, 1, 1)).total_seconds()))
def test_cookie_expires_with_timezone(self):
timestamp = datetime.now(pytz.UTC)
self.assertRaises(
ValueError, cookie, 'http://example.com', 'test', 'value',
expires=timestamp)
class TestB64Decode(TestCase):
def test_decode(self):
test_cases = [{
'about': 'empty string',
'input': '',
'expect': '',
}, {
'about': 'standard encoding, padded',
'input': 'Z29+IQ==',
'expect': 'go~!',
}, {
'about': 'URL encoding, padded',
'input': 'Z29-IQ==',
'expect': 'go~!',
}, {
'about': 'standard encoding, not padded',
'input': 'Z29+IQ',
'expect': 'go~!',
}, {
'about': 'URL encoding, not padded',
'input': 'Z29-IQ',
'expect': 'go~!',
}, {
'about': 'standard encoding, not enough much padding',
'input': 'Z29+IQ=',
'expect_error': 'illegal base64 data at input byte 8',
}]
for test in test_cases:
if test.get('expect_error'):
with self.assertRaises(ValueError, msg=test['about']) as e:
bakery.b64decode(test['input'])
self.assertEqual(str(e.exception), 'Incorrect padding')
else:
self.assertEqual(bakery.b64decode(test['input']), test['expect'].encode('utf-8'), msg=test['about'])
class MacaroonToDictTest(TestCase):
def test_macaroon_to_dict(self):
m = pymacaroons.Macaroon(
key=b'rootkey', identifier=b'some id', location='here', version=2)
as_dict = bakery.macaroon_to_dict(m)
data = json.dumps(as_dict)
m1 = pymacaroons.Macaroon.deserialize(data, json_serializer.JsonSerializer())
self.assertEqual(m1.signature, m.signature)
pymacaroons.Verifier().verify(m1, b'rootkey')
|