def setUp(self): self.sm = SystemManager('localhost') self.sm.create_keyspace(TEST_KEYSPACE) self.sm.create_column_family(TEST_KEYSPACE, TEST_TABLE, key_validation_class=types.UTF8Type, comparator_type=types.UTF8Type, default_validation_class=types.UTF8Type)
class SystemManagerTest(unittest.TestCase): def setUp(self): self.sm = SystemManager('localhost') self.sm.create_keyspace(TEST_KEYSPACE) self.sm.create_column_family(TEST_KEYSPACE, TEST_TABLE, key_validation_class=types.UTF8Type, comparator_type=types.UTF8Type, default_validation_class=types.UTF8Type) def tearDown(self): self.sm.drop_keyspace(TEST_KEYSPACE) self.sm.close() def test_create_existing_keyspace(self): with self.assertRaises(AlreadyExists): self.sm.create_keyspace(TEST_KEYSPACE) def test_create_existing_column_family(self): with self.assertRaises(AlreadyExists): self.sm.create_column_family(TEST_KEYSPACE, TEST_TABLE, key_validation_class=types.UTF8Type, comparator_type=types.UTF8Type, default_validation_class=types.UTF8Type)