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)