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): _, agent_id = self._insert_agent_to_stat_agent() stats = { agent_id: { 'login_time': timedelta(minutes=15), 'pause_time': ONE_HOUR }, } stat_agent_periodic_dao.insert_stats(self.session, stats, dt(2012, 1, 1)) stat_agent_periodic_dao.clean_table(self.session) total = self.session.query(func.count(StatAgentPeriodic.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)