def clear(self, force=False): assert force, 'clear_keyspace must be called with force' assert self.keyspace != settings.CASSANDRA_KEYSPACE, 'Cannot erase the keyspace in settings' management.delete_keyspace(self.keyspace) self.tear_down() return self.setup()
def destroy_db(connection): if None is not connection: keyspace_names = [ key for key in settings.DATABASES['default']['KEYSPACES'].keys() ] for keyspace in keyspace_names: delete_keyspace(keyspace)
def test_create_drop_succeeeds(self): cluster = get_cluster() keyspace_ss = 'test_ks_ss' self.assertFalse(keyspace_ss in cluster.metadata.keyspaces) management.create_keyspace_simple(keyspace_ss, 2) self.assertTrue(keyspace_ss in cluster.metadata.keyspaces) management.drop_keyspace(keyspace_ss) self.assertFalse(keyspace_ss in cluster.metadata.keyspaces) with warnings.catch_warnings(record=True) as w: management.create_keyspace(keyspace_ss, strategy_class="SimpleStrategy", replication_factor=1) self.assertEqual(len(w), 1) self.assertEqual(w[-1].category, DeprecationWarning) self.assertTrue(keyspace_ss in cluster.metadata.keyspaces) management.drop_keyspace(keyspace_ss) self.assertFalse(keyspace_ss in cluster.metadata.keyspaces) keyspace_nts = 'test_ks_nts' self.assertFalse(keyspace_nts in cluster.metadata.keyspaces) management.create_keyspace_simple(keyspace_nts, 2) self.assertTrue(keyspace_nts in cluster.metadata.keyspaces) with warnings.catch_warnings(record=True) as w: management.delete_keyspace(keyspace_nts) self.assertEqual(len(w), 1) self.assertEqual(w[-1].category, DeprecationWarning) self.assertFalse(keyspace_nts in cluster.metadata.keyspaces)
def tearDown(self): delete_keyspace(self.keyspace)