def test_remove_after(self): _, agent_id = self._insert_agent_to_stat_agent() stats = { dt(2012, 1, 1): { agent_id: { 'login_time': timedelta(minutes=15), 'pause_time': timedelta(minutes=13) }, }, dt(2012, 1, 2): { agent_id: { 'login_time': timedelta(minutes=20), 'pause_time': timedelta(minutes=13) }, }, dt(2012, 1, 3): { agent_id: { 'login_time': timedelta(minutes=25), 'pause_time': timedelta(minutes=13) }, }, } self.session.begin() for period_start, agents_stats in stats.iteritems(): stat_agent_periodic_dao.insert_stats(self.session, agents_stats, period_start) self.session.commit() stat_agent_periodic_dao.remove_after(self.session, dt(2012, 1, 2)) res = self.session.query(StatAgentPeriodic.time) self.assertEqual(res.count(), 1) self.assertEqual(res[0].time, dt(2012, 1, 1))
def test_remove_after(self): _, agent_id = self._insert_agent_to_stat_agent() stats = { dt(2012, 1, 1, tzinfo=UTC): { agent_id: { 'login_time': timedelta(minutes=15), 'pause_time': timedelta(minutes=13) }, }, dt(2012, 1, 2, tzinfo=UTC): { agent_id: { 'login_time': timedelta(minutes=20), 'pause_time': timedelta(minutes=13) }, }, dt(2012, 1, 3, tzinfo=UTC): { agent_id: { 'login_time': timedelta(minutes=25), 'pause_time': timedelta(minutes=13) }, }, } with flush_session(self.session): for period_start, agents_stats in six.iteritems(stats): stat_agent_periodic_dao.insert_stats(self.session, agents_stats, period_start) stat_agent_periodic_dao.remove_after(self.session, dt(2012, 1, 2, tzinfo=UTC)) res = self.session.query(StatAgentPeriodic.time) self.assertEqual(res.count(), 1) self.assertEqual(res[0].time, dt(2012, 1, 1, tzinfo=UTC))
def remove_after_start(dao_sess, date): logger.info('Removing agent cache after %s', date) stat_agent_periodic_dao.remove_after(dao_sess, date)