def test_success(self):
user_detail = {'email': '*****@*****.**', 'id': '4'}
sign_cmd = SignCmd('user', user_detail)
sign_cmd.execute()
request = Mock()
request.cookies.get = Mock(return_value=sign_cmd.result)
retrieve_cmd = RetrieveCookieData(request, 'user')
retrieve_cmd.execute()
self.assertDictEqual(user_detail, retrieve_cmd.result)
def test_modified_cookie(self):
dct = {'a': 'asdfsafdsdf', 'id': 4}
name = 'somevalue'
sign = SignCmd(name, dct)
sign.execute()
self.assertIsNotNone(sign.result)
retrieve = RetrieveCmd(name, sign.result[1:], 100)
retrieve.execute()
self.assertIsNone(retrieve.result)
def test_sign_and_retrive(self):
dct = {'a': 'asdfsafdsdf', 'id': 4}
name = 'somevalue'
sign = SignCmd(name, dct)
sign.execute()
self.assertIsNotNone(sign.result)
retrieve = RetrieveCmd(name, sign.result, 100)
retrieve.execute()
self.assertDictEqual(dct, retrieve.result)
def test_expired(self):
dct = {'a': 'asdfsafdsdf', 'id': 4}
name = 'somevalue'
sign = SignCmd(name, dct)
sign.execute()
self.assertIsNotNone(sign.result)
retrieve = RetrieveCmd(name, sign.result, 1)
time.sleep(2)
retrieve.execute()
self.assertIsNone(retrieve.result)
def sign(name, obj):
'''
Returns a command that encode obj as a signed json string
'''
return SignCmd(name, obj)
