32 lines
660 B
Python
Executable File
32 lines
660 B
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import os
|
|
import nose
|
|
import sys
|
|
|
|
sys.path.append("")
|
|
|
|
# setup cassandra
|
|
from cqlengine import connection
|
|
|
|
try:
|
|
CASSANDRA_VERSION = int(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'):
|
|
CASSANDRA_TEST_HOST = os.environ['CASSANDRA_TEST_HOST']
|
|
else:
|
|
CASSANDRA_TEST_HOST = 'localhost'
|
|
|
|
if CASSANDRA_VERSION < 20:
|
|
protocol_version = 1
|
|
else:
|
|
protocol_version = 2
|
|
|
|
connection.setup([CASSANDRA_TEST_HOST], protocol_version=protocol_version, default_keyspace='cqlengine_test')
|
|
|
|
|
|
nose.main()
|