def test_delivery_default_user_destination(self): response = Response(self.mock_firefly, self.user_message, self.hostmask, self.user_destination) message1 = 'Hello world! Message 1' message2 = 'performs an action test 2' message3 = 'Hello world! Notice 3' response.add_message(message1) response.add_action(message2) response.add_notice(message3) mock_message = mock.Mock() mock_action = mock.Mock() mock_notice = mock.Mock() self.mock_firefly.msg = mock_message self.mock_firefly.describe = mock_action self.mock_firefly.notice = mock_notice self.assertEqual(response._destination, response.DEST_USER) self.assertIs(response.destination, self.hostmask) response.send() mock_message.assert_called_once_with(self.hostmask, message1) mock_action.assert_called_once_with(self.hostmask, message2) mock_notice.assert_called_once_with(self.hostmask, message3)
def test_add_actions(self): response = Response(self.mock_firefly, self.channel_message, self.hostmask, self.channel_destination) message1 = 'performs an action test 1' message2 = 'performs an action test 2' message3 = 'performs an action test 3' response.add_action(message1) response.add_action(message2) response.add_action(message3) self.assertListEqual(response._messages, [ ('action', message1, None), ('action', message2, None), ('action', message3, None) ]) self.assertListEqual(response.messages, []) self.assertListEqual(response.actions, [message1, message2, message3]) self.assertListEqual(response.notices, [])
def test_delivery_mixed_destinations(self): response = Response(self.mock_firefly, self.channel_message, self.hostmask, self.channel_destination) message1 = 'Hello world! Default message 1' message2 = 'performs an action channel test 2' message3 = 'Hello world! User notice 3' response.add_message(message1) response.add_action(message2, response.DEST_CHANNEL) response.add_notice(message3, response.DEST_USER) mock_message = mock.Mock() mock_action = mock.Mock() mock_notice = mock.Mock() self.mock_firefly.msg = mock_message self.mock_firefly.describe = mock_action self.mock_firefly.notice = mock_notice response.send() mock_message.assert_called_once_with(self.channel_destination, message1) mock_action.assert_called_once_with(self.channel_destination, message2) mock_notice.assert_called_once_with(self.hostmask, message3)
def test_delivery_timestamps(self): response = Response(self.mock_firefly, self.channel_message, self.hostmask, self.channel_destination) message1 = 'Hello world! Message 1' message2 = 'performs an action test 2' message3 = 'Hello world! Notice 3' response.add_message(message1) response.add_action(message2) response.add_notice(message3) mock_message = mock.Mock() mock_action = mock.Mock() mock_notice = mock.Mock() self.mock_firefly.msg = mock_message self.mock_firefly.describe = mock_action self.mock_firefly.notice = mock_notice response.send() self.assertIsInstance(response._delivered[0][2], arrow.Arrow) self.assertIsInstance(response._delivered[1][2], arrow.Arrow) self.assertIsInstance(response._delivered[2][2], arrow.Arrow)
def test_send_messages(self): response = Response(self.mock_firefly, self.channel_message, self.hostmask, self.channel_destination) message1 = 'Hello world! Message 1' message2 = 'performs an action test 2' message3 = 'Hello world! Notice 3' response.add_message(message1) response.add_action(message2) response.add_notice(message3) mock_message = mock.Mock() mock_action = mock.Mock() mock_notice = mock.Mock() self.mock_firefly.msg = mock_message self.mock_firefly.describe = mock_action self.mock_firefly.notice = mock_notice response.send() mock_message.assert_called_once_with(self.channel_destination, message1) mock_action.assert_called_once_with(self.channel_destination, message2) mock_notice.assert_called_once_with(self.channel_destination, message3)