def test_get_miss(self): _called_with = [] def _get_multi(*args, **kw): _called_with.append((args, kw)) return [] creds = object() client = self._makeOne(credentials=creds) client.get_multi = _get_multi key = object() self.assertIsNone(client.get(key)) self.assertEqual(_called_with[0][0], ()) self.assertEqual(_called_with[0][1]['keys'], [key]) self.assertIsNone(_called_with[0][1]['missing']) self.assertIsNone(_called_with[0][1]['deferred']) self.assertIsNone(_called_with[0][1]['transaction'])
def test_get_hit(self): TXN_ID = '123' _called_with = [] _entity = object() def _get_multi(*args, **kw): _called_with.append((args, kw)) return [_entity] creds = object() client = self._makeOne(credentials=creds) client.get_multi = _get_multi key, missing, deferred = object(), [], [] self.assertIs(client.get(key, missing, deferred, TXN_ID), _entity) self.assertEqual(_called_with[0][0], ()) self.assertEqual(_called_with[0][1]['keys'], [key]) self.assertIs(_called_with[0][1]['missing'], missing) self.assertIs(_called_with[0][1]['deferred'], deferred) self.assertEqual(_called_with[0][1]['transaction'], TXN_ID)