class TestDatabase(unittest.TestCase):
def setUp(self):
self.mc = Memcached(prefix="mgtest-")
self.db = CassandraPool().dbget("mgtest", self.mc)
self.cleanup()
def tearDown(self):
self.cleanup()
pass
def cleanup(self):
conn = self.db.pool.cget()
try:
conn.cass.system_drop_keyspace("mgtest")
except Exception as e:
pass
self.db.pool.success(conn)
self.mc.delete("Cassandra-KS-mgtest")
self.mc.delete("Cassandra-CF-mgtest-Family1")
self.mc.delete("Cassandra-CF-mgtest-Family2")
self.mc.delete("Cassandra-CF-mgtest-Family3")
def testputget(self):
timestamp = time.time() * 1000
self.db.insert("1", ColumnParent(column_family="Family1"), Column(name="email", value="[email protected] - проверка", timestamp=timestamp), ConsistencyLevel.QUORUM)
self.db.get_slice("1", ColumnParent(column_family="Family1"), SlicePredicate(slice_range=SliceRange(start="", finish="")), ConsistencyLevel.QUORUM)
def testerror(self):
try:
self.db.insert("", ColumnParent(column_family="Family1"), Column(name="email", value="somevalue", timestamp=0), ConsistencyLevel.QUORUM)
self.assertTrue(False)
except DatabaseError as e:
self.assertEqual(e.why, "Key may not be empty")
else:
self.assertTrue(False)
class TestDatabase(unittest.TestCase):
def setUp(self):
self.mc = Memcached(prefix="mgtest-")
self.db = CassandraPool().dbget("mgtest", self.mc)
self.cleanup()
def tearDown(self):
self.cleanup()
pass
def cleanup(self):
conn = self.db.pool.cget()
try:
conn.cass.system_drop_keyspace("mgtest")
except Exception as e:
pass
self.db.pool.success(conn)
self.mc.delete("Cassandra-KS-mgtest")
self.mc.delete("Cassandra-CF-mgtest-Family1")
self.mc.delete("Cassandra-CF-mgtest-Family2")
self.mc.delete("Cassandra-CF-mgtest-Family3")
def testputget(self):
timestamp = time.time() * 1000
self.db.insert(
"1", ColumnParent(column_family="Family1"),
Column(name="email",
value="[email protected] - проверка",
timestamp=timestamp), ConsistencyLevel.QUORUM)
self.db.get_slice(
"1", ColumnParent(column_family="Family1"),
SlicePredicate(slice_range=SliceRange(start="", finish="")),
ConsistencyLevel.QUORUM)
def testerror(self):
try:
self.db.insert(
"", ColumnParent(column_family="Family1"),
Column(name="email", value="somevalue", timestamp=0),
ConsistencyLevel.QUORUM)
self.assertTrue(False)
except DatabaseError as e:
self.assertEqual(e.why, "Key may not be empty")
else:
self.assertTrue(False)
