def tearDownClass(cls): execute_with_long_wait_retry(cls.setup_session, "DROP KEYSPACE {0}".format(cls.keyspace1)) execute_with_long_wait_retry(cls.setup_session, "DROP KEYSPACE {0}".format(cls.keyspace2)) models.DEFAULT_KEYSPACE = DEFAULT_KEYSPACE cls.setup_cluster.shutdown() setup_connection(DEFAULT_KEYSPACE) models.DEFAULT_KEYSPACE
def tearDownClass(cls): super(ManagementConnectionTests, cls).tearDownClass() # reset the default connection conn.unregister_connection('fake_cluster') conn.unregister_connection('cluster') setup_connection(DEFAULT_KEYSPACE)
def tearDownClass(cls): super(BatchQueryConnectionTests, cls).tearDownClass() # reset the default connection conn.unregister_connection('fake_cluster') conn.unregister_connection('cluster') setup_connection(DEFAULT_KEYSPACE) drop_keyspace('ks1')
def tearDownClass(cls): super(UsingDescriptorTests, cls).tearDownClass() # reset the default connection conn.unregister_connection('fake_cluster') conn.unregister_connection('cluster') setup_connection(DEFAULT_KEYSPACE) for ks in cls.keyspaces: drop_keyspace(ks)
def tearDownClass(cls): super(ContextQueryConnectionTests, cls).tearDownClass() with ContextQuery(TestModel, connection='cluster') as tm: drop_table(tm) drop_keyspace('ks1', connections=['cluster']) # reset the default connection conn.unregister_connection('fake_cluster') conn.unregister_connection('cluster') setup_connection(DEFAULT_KEYSPACE)
def tearDownClass(cls): conn.unregister_connection('cluster') setup_connection(DEFAULT_KEYSPACE) super(BaseConnectionTestNoDefault, cls).tearDownClass()
def tearDownClass(cls): models.DEFAULT_KEYSPACE = cls.default_keyspace setup_connection(models.DEFAULT_KEYSPACE) super(TestNamedWithMV, cls).tearDownClass()
def tearDown(self): models.DEFAULT_KEYSPACE = self.default_keyspace setup_connection(models.DEFAULT_KEYSPACE)