def test_week(self): self.CONF.set_override('time_unit', 'w', group='timed_strategy') self.strategy = timed_strategy.TimedStrategy() day = dparser.parse('2013-10-31', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20131027000000', self.strategy.get_index()) day = dparser.parse('2013-11-1', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20131027000000', self.strategy.get_index()) day = dparser.parse('2013-11-3', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20131103000000', self.strategy.get_index()) day = dparser.parse('2014-09-12', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20140907000000', self.strategy.get_index()) day = dparser.parse('monkey 2014-07-10 love 12:34:56', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20140706000000', self.strategy.get_index()) # test input integer the_int = int(time.mktime(day.timetuple())) self.strategy.set_time(the_int) self.assertEqual('20140706000000', self.strategy.get_index()) # test the str input self.strategy.set_time('Nov 15, 2014') self.assertEqual('20141109000000', self.strategy.get_index())
def test_send_messages_with_id(self): self.CONF.set_override('id_field', 'id', group='es_conn') self.CONF.set_override('uri', 'http://fake', group='es_conn') self.CONF.set_override('time_unit', 'h', group='timed_strategy') strategy = timed_strategy.TimedStrategy() conn = es_conn.ESConnection('alarms', strategy, 'pre_') req_result = mock.Mock() req_result.status_code = 204 msg = {'id': 'whatever'} with mock.patch.object(requests, 'post', return_value=req_result): conn.send_messages(json.dumps(msg)) self.assertTrue(requests.post.called)
def test_hour(self): self.CONF.set_override('time_unit', 'h', group='timed_strategy') self.strategy = timed_strategy.TimedStrategy() day = dparser.parse('monkey 2014-07-10 love 12:34:56', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20140710120000', self.strategy.get_index()) # test input integer the_int = int(time.mktime(day.timetuple())) self.strategy.set_time(the_int) self.assertEqual('20140710120000', self.strategy.get_index()) # test the str input self.strategy.set_time('Nov 15, 2014 00:27:05') self.assertEqual('20141115000000', self.strategy.get_index())
def test_month(self): self.CONF.set_override('time_unit', 'm', group='timed_strategy') self.strategy = timed_strategy.TimedStrategy() day = dparser.parse('2014-10-31', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20141001000000', self.strategy.get_index()) day = dparser.parse('2014-11-1', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20141101000000', self.strategy.get_index()) day = dparser.parse('2014-11-15', fuzzy=True) self.strategy.set_time(day) self.assertEqual('20141101000000', self.strategy.get_index()) # test input integer the_int = int(time.mktime(day.timetuple())) self.strategy.set_time(the_int) self.assertEqual('20141101000000', self.strategy.get_index()) # test the str input self.strategy.set_time('Nov 15, 2014') self.assertEqual('20141101000000', self.strategy.get_index())