class JSONRedisMappingTest(TestCase):
def setUp(self):
self.r = JSONRedis(StrictRedis(db=15), encode=Cat.encode, decode=Cat.decode)
self.r.flushdb()
self.objects = OrderedDict([
('cat:0', Cat('cat:0', 'Happy')),
('cat:1', Cat('cat:1', 'Grumpy')),
('cat:2', Cat('cat:2', 'Long')),
('cat:3', Cat('cat:3', 'Monorail')),
('cat:4', Cat('cat:4', 'Ceiling'))
])
self.r.omset(self.objects)
self.r.rpush('cats', *self.objects.keys())
self.cats = JSONRedisMapping(self.r, 'cats')
def test_getitem(self):
self.assertEqual(self.cats['cat:0'], self.objects['cat:0'])
def test_iter(self):
# Use list to also compare order
self.assertEqual(list(iter(self.cats)), list(iter(self.objects)))
def test_len(self):
self.assertEqual(len(self.cats), len(self.objects))
def test_contains(self):
self.assertTrue('cat:0' in self.cats)
self.assertFalse('foo' in self.cats)
def setUp(self):
self.r = JSONRedis(StrictRedis(db=15), encode=Cat.encode, decode=Cat.decode)
self.r.flushdb()
self.objects = OrderedDict([
('cat:0', Cat('cat:0', 'Happy')),
('cat:1', Cat('cat:1', 'Grumpy')),
('cat:2', Cat('cat:2', 'Long')),
('cat:3', Cat('cat:3', 'Monorail')),
('cat:4', Cat('cat:4', 'Ceiling'))
])
self.r.omset(self.objects)
self.r.rpush('cats', *self.objects.keys())
self.cats = JSONRedisMapping(self.r, 'cats')
def setUp(self):
self.r = JSONRedis(StrictRedis(db=15), encode=Cat.encode, decode=Cat.decode)
self.r.flushdb()
class JSONRedisTest(TestCase):
def setUp(self):
self.r = JSONRedis(StrictRedis(db=15), encode=Cat.encode, decode=Cat.decode)
self.r.flushdb()
def test_oset_oget(self):
cat = Cat('cat:0', 'Happy')
self.r.oset('cat:0', cat)
got_cat = self.r.oget('cat:0')
self.assertIsInstance(got_cat, Cat)
self.assertEqual(got_cat, cat)
self.assertEqual(got_cat.instance_id, cat.instance_id)
def test_oset_oget_caching_disabled(self):
self.r.caching = False
cat = Cat('cat:0', 'Happy')
self.r.oset('cat:0', cat)
got_cat = self.r.oget('cat:0')
self.assertIsInstance(got_cat, Cat)
self.assertEqual(got_cat, cat)
self.assertNotEqual(got_cat.instance_id, cat.instance_id)
def test_oget_object_destroyed(self):
cat = Cat('cat:0', 'Happy')
self.r.oset('cat:0', cat)
destroyed_instance_id = cat.instance_id
del cat
got_cat = self.r.oget('cat:0')
self.assertNotEqual(got_cat.instance_id, destroyed_instance_id)
def test_oget_cache_empty(self):
self.r.set('cat:0', json.dumps(Cat.encode(Cat('cat:0', 'Happy'))))
got_cat = self.r.oget('cat:0')
same_cat = self.r.oget('cat:0')
self.assertEqual(same_cat, got_cat)
self.assertEqual(same_cat.instance_id, got_cat.instance_id)
def test_oget_cache_empty_caching_disabled(self):
self.r.set('cat:0', json.dumps(Cat.encode(Cat('cat:0', 'Happy'))))
self.r.caching = False
got_cat = self.r.oget('cat:0')
same_cat = self.r.oget('cat:0')
self.assertEqual(same_cat, got_cat)
self.assertNotEqual(same_cat.instance_id, got_cat.instance_id)
def test_oget_key_nonexistant(self):
self.assertIsNone(self.r.oget('foo'))
def test_oget_value_not_json(self):
self.r.set('not-json', 'not-json')
with self.assertRaises(ResponseError):
self.r.oget('not-json')
def test_omget_omset(self):
cats = {'cat:0': Cat('cat:0', 'Happy'), 'cat:1': Cat('cat:1', 'Grumpy')}
self.r.omset(cats)
got_cats = self.r.omget(cats.keys())
self.assertEqual(got_cats, list(cats.values()))
