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

Use QUORUM consistency for cassandra.concurrent test

Browse Source
This commit is contained in:
Tyler Hobbs
2014-05-08 16:18:40 -05:00
parent 9206341d9d
commit cc26613bdb
1 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/integration/standard/test_concurrent.py
View File

@@ -19,12 +19,13 @@ except ImportError:
from itertools import cycle from itertools import cycle
from cassandra import InvalidRequest from cassandra import InvalidRequest, ConsistencyLevel
from cassandra.cluster import Cluster from cassandra.cluster import Cluster
from cassandra.concurrent import (execute_concurrent, from cassandra.concurrent import (execute_concurrent,
execute_concurrent_with_args) execute_concurrent_with_args)
from cassandra.policies import HostDistance from cassandra.policies import HostDistance
from cassandra.decoder import tuple_factory from cassandra.decoder import tuple_factory
from cassandra.query import SimpleStatement
class ClusterTests(unittest.TestCase): class ClusterTests(unittest.TestCase):
@@ -38,7 +39,10 @@ class ClusterTests(unittest.TestCase):
def test_execute_concurrent(self): def test_execute_concurrent(self):
for num_statements in (0, 1, 2, 7, 10, 99, 100, 101, 199, 200, 201): for num_statements in (0, 1, 2, 7, 10, 99, 100, 101, 199, 200, 201):
# write # write
statements = cycle(("INSERT INTO test3rf.test (k, v) VALUES (%s, %s)", )) statement = SimpleStatement(
"INSERT INTO test3rf.test (k, v) VALUES (%s, %s)",
consistency_level=ConsistencyLevel.QUORUM)
statements = cycle((statement, ))
parameters = [(i, i) for i in range(num_statements)] parameters = [(i, i) for i in range(num_statements)]
results = execute_concurrent(self.session, zip(statements, parameters)) results = execute_concurrent(self.session, zip(statements, parameters))
@@ -46,7 +50,10 @@ class ClusterTests(unittest.TestCase):
self.assertEqual([(True, None)] * num_statements, results) self.assertEqual([(True, None)] * num_statements, results)
# read # read
statements = cycle(("SELECT v FROM test3rf.test WHERE k=%s", )) statement = SimpleStatement(
"SELECT v FROM test3rf.test WHERE k=%s",
consistency_level=ConsistencyLevel.QUORUM)
statements = cycle((statement, ))
parameters = [(i, ) for i in range(num_statements)] parameters = [(i, ) for i in range(num_statements)]
results = execute_concurrent(self.session, zip(statements, parameters)) results = execute_concurrent(self.session, zip(statements, parameters))
@@ -55,7 +62,9 @@ class ClusterTests(unittest.TestCase):
def test_execute_concurrent_with_args(self): def test_execute_concurrent_with_args(self):
for num_statements in (0, 1, 2, 7, 10, 99, 100, 101, 199, 200, 201): for num_statements in (0, 1, 2, 7, 10, 99, 100, 101, 199, 200, 201):
statement = "INSERT INTO test3rf.test (k, v) VALUES (%s, %s)" statement = SimpleStatement(
"INSERT INTO test3rf.test (k, v) VALUES (%s, %s)",
consistency_level=ConsistencyLevel.QUORUM)
parameters = [(i, i) for i in range(num_statements)] parameters = [(i, i) for i in range(num_statements)]
results = execute_concurrent_with_args(self.session, statement, parameters) results = execute_concurrent_with_args(self.session, statement, parameters)
@@ -63,7 +72,9 @@ class ClusterTests(unittest.TestCase):
self.assertEqual([(True, None)] * num_statements, results) self.assertEqual([(True, None)] * num_statements, results)
# read # read
statement = "SELECT v FROM test3rf.test WHERE k=%s" statement = SimpleStatement(
"SELECT v FROM test3rf.test WHERE k=%s",
consistency_level=ConsistencyLevel.QUORUM)
parameters = [(i, ) for i in range(num_statements)] parameters = [(i, ) for i in range(num_statements)]
results = execute_concurrent_with_args(self.session, statement, parameters) results = execute_concurrent_with_args(self.session, statement, parameters)
@@ -82,7 +93,10 @@ class ClusterTests(unittest.TestCase):
execute_concurrent, self.session, zip(statements, parameters), raise_on_first_error=True) execute_concurrent, self.session, zip(statements, parameters), raise_on_first_error=True)
def test_first_failure_client_side(self): def test_first_failure_client_side(self):
statements = cycle(("INSERT INTO test3rf.test (k, v) VALUES (%s, %s)", )) statement = SimpleStatement(
"INSERT INTO test3rf.test (k, v) VALUES (%s, %s)",
consistency_level=ConsistencyLevel.QUORUM)
statements = cycle((statement, ))
parameters = [(i, i) for i in range(100)] parameters = [(i, i) for i in range(100)]
# the driver will raise an error when binding the params # the driver will raise an error when binding the params
@@ -93,7 +107,10 @@ class ClusterTests(unittest.TestCase):
execute_concurrent, self.session, zip(statements, parameters), raise_on_first_error=True) execute_concurrent, self.session, zip(statements, parameters), raise_on_first_error=True)
def test_no_raise_on_first_failure(self): def test_no_raise_on_first_failure(self):
statements = cycle(("INSERT INTO test3rf.test (k, v) VALUES (%s, %s)", )) statement = SimpleStatement(
"INSERT INTO test3rf.test (k, v) VALUES (%s, %s)",
consistency_level=ConsistencyLevel.QUORUM)
statements = cycle((statement, ))
parameters = [(i, i) for i in range(100)] parameters = [(i, i) for i in range(100)]
# we'll get an error back from the server # we'll get an error back from the server
@@ -109,7 +126,10 @@ class ClusterTests(unittest.TestCase):
self.assertEqual(None, result) self.assertEqual(None, result)
def test_no_raise_on_first_failure_client_side(self): def test_no_raise_on_first_failure_client_side(self):
statements = cycle(("INSERT INTO test3rf.test (k, v) VALUES (%s, %s)", )) statement = SimpleStatement(
"INSERT INTO test3rf.test (k, v) VALUES (%s, %s)",
consistency_level=ConsistencyLevel.QUORUM)
statements = cycle((statement, ))
parameters = [(i, i) for i in range(100)] parameters = [(i, i) for i in range(100)]
# the driver will raise an error when binding the params # the driver will raise an error when binding the params