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

fixing protocol version

Browse Source
This commit is contained in:
Jon Haddad
2014-08-13 13:22:08 -07:00
parent 30dbc6e4d9
commit 503e2d1ecf
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
bin/test.py
View File

@@ -9,19 +9,23 @@ sys.path.append("")
# setup cassandra # setup cassandra
from cqlengine import connection from cqlengine import connection
try:
CASSANDRA_VERSION = os.environ["CASSANDRA_VERSION"]
except:
print "CASSANDRA_VERSION must be set as an environment variable. One of (12, 20, 21)"
raise
if os.environ.get('CASSANDRA_TEST_HOST'): if os.environ.get('CASSANDRA_TEST_HOST'):
CASSANDRA_TEST_HOST = os.environ['CASSANDRA_TEST_HOST'] CASSANDRA_TEST_HOST = os.environ['CASSANDRA_TEST_HOST']
else: else:
CASSANDRA_TEST_HOST = 'localhost' CASSANDRA_TEST_HOST = 'localhost'
protocol_version = int(os.environ.get("CASSANDRA_PROTOCOL_VERSION", 2)) if CASSANDRA_VERSION < 20:
protocol_version = 1
else:
protocol_version = 2
connection.setup([CASSANDRA_TEST_HOST], protocol_version=protocol_version, default_keyspace='cqlengine_test') connection.setup([CASSANDRA_TEST_HOST], protocol_version=protocol_version, default_keyspace='cqlengine_test')
try:
c_version = os.environ["CASSANDRA_VERSION"]
except:
print "CASSANDRA_VERSION must be set as an environment variable. One of (12, 20, 21)"
raise
nose.main() nose.main()