def test_update_from_dict_empty(self):
dic = {}
store = TokenStore()
store.update_from_dict(dic)
self.assertIsNone(store.access_token)
self.assertIsNone(store.refresh_token)
self.assertIsNone(store._timestamp)
def test_update_from_dict_no_ts(self):
dic = {
_ACCESS_TOKEN: 'a',
_REFRESH_TOKEN: 'b',
}
store = TokenStore()
store.update_from_dict(dic)
self.assertEqual('a', store.access_token)
self.assertEqual('b', store.refresh_token)
self.assertIsNone(store._timestamp)
def test_update_from_dict(self):
ts = datetime.datetime.utcnow()
dic = {
_ACCESS_TOKEN: 'a',
_REFRESH_TOKEN: 'b',
_KEY_TIMESTAMP: ts,
}
store = TokenStore()
store.update_from_dict(dic)
self.assertEqual('a', store.access_token)
self.assertEqual('b', store.refresh_token)
self.assertEqual(ts, store._timestamp)
def test_update_from_dict_none(self):
store = TokenStore()
self.assertRaises(AttributeError, lambda: store.update_from_dict(None))