def test_connecting_with_keyspace(self):
# this conn is just for creating the keyspace
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True)
curs = conn.cursor()
with self.with_keyspace(curs, conn.cql_version) as ksname:
curs.execute('create table blah1_%s (a int primary key, b int);' % self.randstr)
conn2 = cql.connect(TEST_HOST, TEST_NATIVE_PORT, keyspace=ksname,
native=True, cql_version=conn.cql_version)
curs2 = conn2.cursor()
curs2.execute('select * from blah1_%s;' % self.randstr)
conn2.close()
def test_connecting_with_keyspace(self):
# this conn is just for creating the keyspace
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True)
curs = conn.cursor()
with self.with_keyspace(curs, conn.cql_version) as ksname:
curs.execute('create table blah1_%s (a int primary key, b int);' % self.randstr)
conn2 = cql.connect(TEST_HOST, TEST_NATIVE_PORT, keyspace=ksname,
native=True, cql_version=conn.cql_version)
curs2 = conn2.cursor()
curs2.execute('select * from blah1_%s;' % self.randstr)
conn2.close()
def test_connecting_without_compression(self):
conn = cql.connect(TEST_HOST,
TEST_NATIVE_PORT,
native=True,
compression=False)
self.assertEqual(conn.compressor, None)
self.try_basic_stuff(conn)
def test_execution_fails_after_close(self):
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True)
curs = conn.cursor()
with self.with_keyspace(curs, conn.cql_version) as ksname:
curs.execute('create table blah (a int primary key, b int);')
curs.execute('select * from blah;')
conn.close()
self.assertRaises(cql.ProgrammingError, curs.execute, 'select * from blah;')
def test_make_many_queries(self):
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True)
curs = conn.cursor()
with self.with_keyspace(curs, conn.cql_version) as ksname:
curs.execute('create table blah (a int primary key, b int);')
for i in range(200):
curs.execute('select * from blah;')
conn.close()
def test_execution_fails_after_close(self):
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True)
curs = conn.cursor()
with self.with_keyspace(curs, conn.cql_version) as ksname:
curs.execute('create table blah (a int primary key, b int);')
curs.execute('select * from blah;')
conn.close()
self.assertRaises(cql.ProgrammingError, curs.execute, 'select * from blah;')
def test_connecting_with_compression(self):
try:
import snappy
except ImportError:
if hasattr(unittest, 'skipTest'):
unittest.skipTest('Snappy compression not available')
else:
return
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True, compression=True)
self.assertEqual(conn.compressor, snappy.compress)
self.try_basic_stuff(conn)
def test_connecting_with_compression(self):
try:
import snappy
except ImportError:
if hasattr(unittest, 'skipTest'):
unittest.skipTest('Snappy compression not available')
else:
return
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True, compression=True)
self.assertEqual(conn.compressor, snappy.compress)
self.try_basic_stuff(conn)
def test_connecting_without_compression(self):
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True, compression=False)
self.assertEqual(conn.compressor, None)
self.try_basic_stuff(conn)
def test_make_too_many_concurrent_queries(self):
conn = cql.connect(TEST_HOST, TEST_NATIVE_PORT, native=True)
for i in range(128):
conn.make_reqid()
self.assertRaises(cql.ProgrammingError, conn.make_reqid)
conn.close()
