def test_get_message_count_given_from_and_to_date(self):
message_attr = self.message.copy()
message_attr['received_at'] = datetime.datetime(
2014, 11, 17, 16, 0, 49, 807000)
RapidProMessageBase(**message_attr).save()
RapidProMessageBase(**self.message).save()
location_messages_count = RapidProMessageBase.count_(**{
'from': '2014-09-17',
'to': '2014-10-17'
})
self.assertEqual(1, location_messages_count)
location_messages_count = RapidProMessageBase.count_(**{
'from': '2014-08-17',
'to': '2014-12-17'
})
self.assertEqual(2, location_messages_count)
location_messages_count = RapidProMessageBase.count_(
**{'from': '2014-08-17'})
self.assertEqual(2, location_messages_count)
location_messages_count = RapidProMessageBase.count_(**{
'from': None,
'to': None
})
self.assertEqual(2, location_messages_count)
def test_get_message_count_given_from_and_to_date(self):
message_attr = self.message.copy()
message_attr['received_at'] = datetime.datetime(2014, 11, 17, 16, 0, 49, 807000)
RapidProMessageBase(**message_attr).save()
RapidProMessageBase(**self.message).save()
location_messages_count = RapidProMessageBase.count_(**{'from': '2014-09-17', 'to': '2014-10-17'})
self.assertEqual(1, location_messages_count)
location_messages_count = RapidProMessageBase.count_(**{'from': '2014-08-17', 'to': '2014-12-17'})
self.assertEqual(2, location_messages_count)
location_messages_count = RapidProMessageBase.count_(**{'from': '2014-08-17'})
self.assertEqual(2, location_messages_count)
location_messages_count = RapidProMessageBase.count_(**{'from': None, 'to': None})
self.assertEqual(2, location_messages_count)
