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())