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): {
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)
