openstack/deb-python-cassandra-driver
Code Issues Proposed changes

Add some tests for cqlengine queryset slicing

Browse Source
This commit is contained in:
Alan Boudreault
2016-03-03 14:41:49 -05:00
parent 3e379abf8f
commit 84c9c8c04b
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/integration/cqlengine/query/test_queryset.py
View File

@@ -554,16 +554,31 @@ class TestQuerySetSlicing(BaseQuerySetUsage):
def test_slicing_works_properly(self): def test_slicing_works_properly(self):
q = TestModel.objects(test_id=0).order_by('attempt_id') q = TestModel.objects(test_id=0).order_by('attempt_id')
expected_order = [0, 1, 2, 3] expected_order = [0, 1, 2, 3]
for model, expect in zip(q[1:3], expected_order[1:3]): for model, expect in zip(q[1:3], expected_order[1:3]):
assert model.attempt_id == expect self.assertEqual(model.attempt_id, expect)
for model, expect in zip(q[0:3:2], expected_order[0:3:2]):
self.assertEqual(model.attempt_id, expect)
def test_negative_slicing(self): def test_negative_slicing(self):
q = TestModel.objects(test_id=0).order_by('attempt_id') q = TestModel.objects(test_id=0).order_by('attempt_id')
expected_order = [0, 1, 2, 3] expected_order = [0, 1, 2, 3]
for model, expect in zip(q[-3:], expected_order[-3:]): for model, expect in zip(q[-3:], expected_order[-3:]):
assert model.attempt_id == expect self.assertEqual(model.attempt_id, expect)
for model, expect in zip(q[:-1], expected_order[:-1]): for model, expect in zip(q[:-1], expected_order[:-1]):
assert model.attempt_id == expect self.assertEqual(model.attempt_id, expect)
for model, expect in zip(q[1:-1], expected_order[1:-1]):
self.assertEqual(model.attempt_id, expect)
for model, expect in zip(q[-3:-1], expected_order[-3:-1]):
self.assertEqual(model.attempt_id, expect)
for model, expect in zip(q[-3:-1:2], expected_order[-3:-1:2]):
self.assertEqual(model.attempt_id, expect)
class TestQuerySetValidation(BaseQuerySetUsage): class TestQuerySetValidation(BaseQuerySetUsage):