class TestNTSerializer(unittest.TestCase):
def setUp(self):
self.ss = SortedSet(
redis.Redis(),
self.__class__.__name__,
serializer=NamedtupleSerializer(DiscoTask),
)
def tearDown(self):
self.ss.clear()
def test_nt_serializer(self):
dt = DiscoTask(tiger='larry', woods='david')
self.ss.add(dt)
assert len(self.ss) == 1
assert dt in self.ss
self.assertEqual(
dt,
self.ss.pop(),
)
class TestNTSerializer(unittest.TestCase):
def setUp(self):
self.ss = SortedSet(
redis.Redis(),
self.__class__.__name__,
serializer=NamedtupleSerializer(DiscoTask),
)
def tearDown(self):
self.ss.clear()
def test_nt_serializer(self):
dt = DiscoTask(tiger='larry', woods='david')
self.ss.add(dt)
assert len(self.ss) == 1
assert dt in self.ss
self.assertEqual(
dt,
self.ss.pop(),
)
class SortedSetTest(unittest.TestCase):
def setUp(self):
self.key = 'ss_test'
self.ss = SortedSet(redis.Redis(), self.key)
def tearDown(self):
self.ss.clear()
def test_repr(self):
"""Just make sure it doesn't blow up."""
str(self.ss)
def test_length(self):
for i in range(5):
self.ss.add(i)
self.assertEquals(
len(self.ss),
5,
)
def test_add_with_score(self):
item = 'samere'
score = 123
self.ss.add(item, score)
assert self.ss.score(item) == score
def test_and_and_update_score(self):
item = 'samere'
score = 123
self.ss.add(item, score)
new_score = 456
self.ss.add(item, new_score)
assert self.ss.score(item) == new_score
def test_contains(self):
for i in range(5):
self.ss.add(i)
self.assertTrue(0 in self.ss)
self.assertFalse(-1 in self.ss)
def test_ordering(self):
for i in range(5):
self.ss.add(i, score=i)
self.assertEquals(
[str(i) for i in range(5)],
[self.ss.pop() for __ in range(5)],
)
def test_empty_pop(self):
with self.assertRaises(KeyError):
self.ss.pop()
def test_empty_peek(self):
with self.assertRaises(KeyError):
self.ss.peek()
def test_add_dup(self):
for i in range(5):
self.ss.add(i)
dup_added_at = 10
self.ss.add(0, score=dup_added_at)
self.assertEquals(
len(self.ss),
5,
)
self.assertEquals(
int(self.ss.score(0)),
int(dup_added_at),
)
def test_clear(self):
self.assertFalse(self.ss.clear())
for i in range(5):
self.ss.add(i)
self.assertTrue(self.ss.clear())
self.assertEquals(
len(self.ss),
0,
)
def test_discard(self):
self.ss.add(0)
self.assertTrue(self.ss.discard(0))
self.assertFalse(self.ss.discard(0))
def test_peek(self):
with self.assertRaises(KeyError):
self.ss.peek()
self.ss.add(0)
for __ in range(2):
self.assertEquals(
self.ss.peek(),
'0',
)
with self.assertRaises(KeyError):
self.ss.peek(position=1)
self.ss.add(1)
for __ in range(2):
self.assertEquals(
self.ss.peek(position=1),
'1',
)
def test_take(self):
for i in range(5):
self.ss.add(i)
self.assertEquals(
set([str(i) for i in range(2)]),
set(self.ss.take(2)),
)
self.assertEquals(
set([str(i + 2) for i in range(3)]),
set(self.ss.take(100)),
)
self.assertEquals(
len(self.ss),
0,
)
self.assertEquals(
self.ss.take(0),
[],
)
self.assertEquals(
self.ss.take(-1),
[],
)
class SerializerTest(unittest.TestCase):
class FakeJsonSerializer(Serializer):
"""
Handles JSON serialization.
"""
def dumps(self, item):
return json.dumps(item)
def loads(self, item):
if 'uhoh' in item:
raise Exception("omg unserializing failed!")
return json.loads(item)
def setUp(self):
self.key = 'json_ss_test'
self.ss = SortedSet(
redis.Redis(),
self.key,
serializer=self.FakeJsonSerializer(),
)
# has a bad serializer
self.ss2 = SortedSet(
redis.Redis(),
self.key + '2',
serializer=object(),
)
def tearDown(self):
self.ss.clear()
self.ss2.clear()
def test_add_and_pop(self):
self.ss.add({'yo': 'json'}, score=1)
self.ss.add({'yo': 'yaml'}, score=0)
self.assertTrue({'yo': 'json'} in self.ss)
self.assertEqual(
self.ss.pop(),
{'yo': 'yaml'},
)
self.assertEqual(
self.ss.pop(),
{'yo': 'json'},
)
self.assertEqual(
0,
len(self.ss),
)
def test_cant_deserialize(self):
self.ss.add({'yo': 'foo'}, score=0)
self.ss.add({'yo': 'uhoh!'}, score=1)
self.ss.add({'yo': 'hey'}, score=2)
self.assertEquals(
self.ss.take(3),
[{
'yo': 'foo'
}, {
'yo': 'hey'
}],
)
def test_bad_serializer(self):
self.ss2.add(1, score=0)
self.ss2.add(2, score=1)
assert '2' in self.ss2
# gets deserialied as a str, not an int
self.assertEquals(
'1',
self.ss2.pop(),
)
class SerializerTest(unittest.TestCase):
class FakeJsonSerializer(Serializer):
"""
Handles JSON serialization.
"""
def dumps(self, item):
return json.dumps(item)
def loads(self, item):
if 'uhoh' in item:
raise Exception("omg unserializing failed!")
return json.loads(item)
def setUp(self):
self.key = 'json_ss_test'
self.ss = SortedSet(
redis.Redis(),
self.key,
serializer=self.FakeJsonSerializer(),
)
# has a bad serializer
self.ss2 = SortedSet(
redis.Redis(),
self.key + '2',
serializer=object(),
)
def tearDown(self):
self.ss.clear()
self.ss2.clear()
def test_add_and_pop(self):
self.ss.add({'yo': 'json'}, score=1)
self.ss.add({'yo': 'yaml'}, score=0)
self.assertTrue(
{'yo': 'json'} in self.ss
)
self.assertEqual(
self.ss.pop(),
{'yo': 'yaml'},
)
self.assertEqual(
self.ss.pop(),
{'yo': 'json'},
)
self.assertEqual(
0,
len(self.ss),
)
def test_cant_deserialize(self):
self.ss.add({'yo': 'foo'}, score=0)
self.ss.add({'yo': 'uhoh!'}, score=1)
self.ss.add({'yo': 'hey'}, score=2)
self.assertEquals(
self.ss.take(3),
[{'yo': 'foo'},
{'yo': 'hey'}],
)
def test_bad_serializer(self):
self.ss2.add(1, score=0)
self.ss2.add(2, score=1)
assert '2' in self.ss2
# gets deserialied as a str, not an int
self.assertEquals(
'1',
self.ss2.pop(),
)
class SortedSetTest(unittest.TestCase):
def setUp(self):
self.key = 'ss_test'
self.ss = SortedSet(redis.Redis(), self.key)
def tearDown(self):
self.ss.clear()
def test_repr(self):
"""Just make sure it doesn't blow up."""
str(self.ss)
def test_length(self):
for i in range(5):
self.ss.add(i)
self.assertEquals(
len(self.ss),
5,
)
def test_add_with_score(self):
item = 'samere'
score = 123
self.ss.add(item, score)
assert self.ss.score(item) == score
def test_and_and_update_score(self):
item = 'samere'
score = 123
self.ss.add(item, score)
new_score = 456
self.ss.add(item, new_score)
assert self.ss.score(item) == new_score
def test_contains(self):
for i in range(5):
self.ss.add(i)
self.assertTrue(
0 in self.ss
)
self.assertFalse(
-1 in self.ss
)
def test_ordering(self):
for i in range(5):
self.ss.add(i, score=i)
self.assertEquals(
[str(i) for i in range(5)],
[self.ss.pop() for __ in range(5)],
)
def test_empty_pop(self):
with self.assertRaises(KeyError):
self.ss.pop()
def test_empty_peek(self):
with self.assertRaises(KeyError):
self.ss.peek()
def test_add_dup(self):
for i in range(5):
self.ss.add(i)
dup_added_at = 10
self.ss.add(0, score=dup_added_at)
self.assertEquals(
len(self.ss),
5,
)
self.assertEquals(
int(self.ss.score(0)),
int(dup_added_at),
)
def test_clear(self):
self.assertFalse(self.ss.clear())
for i in range(5):
self.ss.add(i)
self.assertTrue(self.ss.clear())
self.assertEquals(
len(self.ss),
0,
)
def test_discard(self):
self.ss.add(0)
self.assertTrue(self.ss.discard(0))
self.assertFalse(self.ss.discard(0))
def test_peek(self):
with self.assertRaises(KeyError):
self.ss.peek()
self.ss.add(0)
for __ in range(2):
self.assertEquals(
self.ss.peek(),
'0',
)
with self.assertRaises(KeyError):
self.ss.peek(position=1)
self.ss.add(1)
for __ in range(2):
self.assertEquals(
self.ss.peek(position=1),
'1',
)
def test_take(self):
for i in range(5):
self.ss.add(i)
self.assertEquals(
set([str(i) for i in range(2)]),
set(self.ss.take(2)),
)
self.assertEquals(
set([str(i + 2) for i in range(3)]),
set(self.ss.take(100)),
)
self.assertEquals(
len(self.ss),
0,
)
self.assertEquals(
self.ss.take(0),
[],
)
self.assertEquals(
self.ss.take(-1),
[],
)
