def test_agent_due(self): in_two_hours = datetime.datetime.now() + datetime.timedelta(hours=2) agent1 = TestAgent.new('Agent 1', cron_entry='*/10 * * * *') agent2 = TestAgent.new('Agent 2', cron_entry='%s %s * * *' % (in_two_hours.minute, in_two_hours.hour)) d1 = datetime.datetime.now() + datetime.timedelta(seconds=601) d2 = datetime.datetime.now() + datetime.timedelta(hours=3) d3 = datetime.datetime.now() - datetime.timedelta(seconds=10) a1 = AgentStore.due(d1) a2 = AgentStore.due(d2) a3 = AgentStore.due(d3) self.assertEqual(len(a1), 1) self.assertEqual(len(a2), 2) self.assertEqual(len(a3), 0)
def get(self): now = datetime.datetime.now() agents = AgentStore.due(now) for agent in agents: if not agent.cron_entry == models.ON_NEW_EVENT: agent.run_taskqueue()
def get(self): now = datetime.datetime.now() agents = AgentStore.due(now) for agent in agents: agent.run_taskqueue()