def test_init(self):
# assert example good encoding
q = Base(MagicMock(), 'key')
q1 = Base(MagicMock(), 'key', pickle)
q2 = Base(MagicMock(), 'key', ujson)
q3 = Base(MagicMock(), 'key', json)
# assert bad encodings
with self.assertRaises(NotImplementedError) as e:
q2 = Base(MagicMock(), 'key', copy)
def test_encode(self):
q = Base(MagicMock(), 'key', pickle)
# python pickling is different between versions
data = pickle.dumps('cool', protocol=-1)
self.assertEquals(q._encode_item('cool'), data)
q2 = Base(MagicMock(), 'key', ujson)
self.assertEquals(q2._encode_item('cool2'), '"cool2"')
class TestBase(TestCase):
def setUp(self):
self.q = Base(MagicMock(), 'k')
def test_init(self):
# assert example good encoding
q = Base(MagicMock(), 'key')
q1 = Base(MagicMock(), 'key', pickle)
q2 = Base(MagicMock(), 'key', ujson)
q3 = Base(MagicMock(), 'key', json)
# assert bad encodings
with self.assertRaises(NotImplementedError) as e:
q2 = Base(MagicMock(), 'key', copy)
def test_encode(self):
q = Base(MagicMock(), 'key', pickle)
# python pickling is different between versions
data = pickle.dumps('cool', protocol=-1)
self.assertEquals(q._encode_item('cool'), data)
q2 = Base(MagicMock(), 'key', ujson)
self.assertEquals(q2._encode_item('cool2'), '"cool2"')
def test_decode(self):
q = Base(MagicMock(), 'key', pickle)
self.assertEquals(q._decode_item(b"\x80\x02U\x04coolq\x00."), 'cool')
q2 = Base(MagicMock(), 'key', ujson)
self.assertEquals(q2._decode_item('"cool2"'), 'cool2')
def test_len(self):
with self.assertRaises(NotImplementedError):
len(self.q)
def test_push(self):
with self.assertRaises(NotImplementedError):
self.q.push("blah")
def test_pop(self):
with self.assertRaises(NotImplementedError):
self.q.pop()
def test_clear(self):
self.q.clear()
def test_decode(self):
q = Base(MagicMock(), 'key', pickle)
self.assertEquals(q._decode_item(b"\x80\x02U\x04coolq\x00."), 'cool')
q2 = Base(MagicMock(), 'key', ujson)
self.assertEquals(q2._decode_item('"cool2"'), 'cool2')
def setUp(self):
self.q = Base(MagicMock(), 'k')