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

Merge pull request #730 from datastax/python-688-test

Browse Source 
Added some more asserts to test_map_update_remove in tests
This commit is contained in:
Jaume Marhuenda
2017-04-05 10:14:47 -04:00
committed by GitHub
parent 048e11a42c 6f16e166c3
commit 0578dbbe8b
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/integration/cqlengine/query/test_updates.py
View File

@@ -21,7 +21,7 @@ from cassandra.cqlengine import columns
from tests.integration.cqlengine import is_prepend_reversed
from tests.integration.cqlengine.base import BaseCassEngTestCase
from tests.integration.cqlengine import execute_count
from tests.integration import greaterthancass20
class TestQueryUpdateModel(Model):
@@ -242,6 +242,7 @@ class QueryUpdateTests(BaseCassEngTestCase):
obj = TestQueryUpdateModel.objects.get(partition=partition, cluster=cluster)
self.assertEqual(obj.text_map, {"foo": '1'})
@greaterthancass20
@execute_count(5)
def test_map_update_remove(self):
"""
@@ -257,13 +258,14 @@ class QueryUpdateTests(BaseCassEngTestCase):
text_map={"foo": '1', "bar": '2'}
)
TestQueryUpdateModel.objects(partition=partition, cluster=cluster).update(
text_map__remove={"bar"}
text_map__remove={"bar"},
text_map__update={"foz": '4', "foo": '2'}
)
obj = TestQueryUpdateModel.objects.get(partition=partition, cluster=cluster)
self.assertEqual(obj.text_map, {"foo": '1'})
self.assertEqual(obj.text_map, {"foo": '2', "foz": '4'})
TestQueryUpdateModel.objects(partition=partition, cluster=cluster).update(
text_map__remove={"foo"}
text_map__remove={"foo", "foz"}
)
self.assertEqual(
TestQueryUpdateModel.objects.get(partition=partition, cluster=cluster).text_map,