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