def clean_db(): with session_scope() as dao_sess: stat_call_on_queue_dao.clean_table(dao_sess) stat_agent_periodic_dao.clean_table(dao_sess) stat_queue_periodic_dao.clean_table(dao_sess) stat_agent_dao.clean_table(dao_sess) stat_queue_dao.clean_table(dao_sess)
def test_clean_table(self): queue_name, queue_id = self._insert_queue_to_stat_queue() stats = {queue_id: {'full': 4, 'total': 10}} period_start = dt(2012, 1, 1, 00, 00, 00, tzinfo=UTC) stat_queue_periodic_dao.insert_stats(self.session, stats, period_start) stat_queue_periodic_dao.clean_table(self.session) total = self.session.query(func.count(StatQueuePeriodic.time))[0][0] self.assertEqual(total, 0)
def test_clean_table(self): queue_name, queue_id = self._insert_queue_to_stat_queue() stats = { queue_id: { 'full': 4, 'total': 10 } } period_start = datetime.datetime(2012, 01, 01, 00, 00, 00) stat_queue_periodic_dao.insert_stats(self.session, stats, period_start) stat_queue_periodic_dao.clean_table(self.session) total = self.session.query(func.count(StatQueuePeriodic.time))[0][0] self.assertEqual(total, 0)
def clean_db(): stat_call_on_queue_dao.clean_table(dao_sess) stat_agent_periodic_dao.clean_table(dao_sess) stat_queue_periodic_dao.clean_table(dao_sess) stat_agent_dao.clean_table(dao_sess) stat_queue_dao.clean_table(dao_sess)