Пример #1
0
    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)
Пример #2
0
    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, [])
Пример #3
0
    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)
Пример #4
0
    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)
Пример #5
0
    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)