def testMessagesOrdered(self): """Tests that the returned messages are ordered by creation date.""" # seed a couple of messages for the connection message1 = connection_utils.seed_new_connection_message( self.connection.key, created=datetime.now()) message2 = connection_utils.seed_new_connection_message( self.connection.key, created=timeline_utils.past(delta=100)) message3 = connection_utils.seed_new_connection_message( self.connection.key, created=timeline_utils.past(delta=50)) messages = connection_logic.getConnectionMessages(self.connection.key) self.assertListEqual([message2, message3, message1], messages)
def testCorrectMessagesReturned(self): """Tests that correct messages are returned.""" # seed a couple of messages for the connection message1 = connection_utils.seed_new_connection_message( self.connection.key, author=self.profile.key) message2 = connection_utils.seed_new_connection_message(self.connection.key) # seed another organization and a connection other_org = org_utils.seedOrganization(self.program.key()) other_connection = connection_utils.seed_new_connection( self.profile.key, other_org.key) # create a few messages for the other connection for _ in range(4): connection_utils.seed_new_connection_message( other_connection.key, author=self.profile.key) # check that only correct messages are returned messages = connection_logic.getConnectionMessages(self.connection.key) expected_keys = set([message1.key, message2.key]) actual_keys = set([message.key for message in messages]) self.assertEqual(actual_keys, expected_keys)
def testCorrectMessagesReturned(self): """Tests that correct messages are returned.""" # seed a couple of messages for the connection message1 = connection_utils.seed_new_connection_message( self.connection.key, author=self.profile.key) message2 = connection_utils.seed_new_connection_message( self.connection.key) # seed another organization and a connection other_org = org_utils.seedOrganization(self.program.key()) other_connection = connection_utils.seed_new_connection( self.profile.key, other_org.key) # create a few messages for the other connection for _ in range(4): connection_utils.seed_new_connection_message( other_connection.key, author=self.profile.key) # check that only correct messages are returned messages = connection_logic.getConnectionMessages(self.connection.key) expected_keys = set([message1.key, message2.key]) actual_keys = set([message.key for message in messages]) self.assertEqual(actual_keys, expected_keys)