mock.sentinel.message

Here are the examples of the python api mock.sentinel.message taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.

4 Examples 7

Example 1

Project: kafka-python Source File: test_protocol_legacy.py
    @contextmanager
    def mock_create_message_fns(self):
        import kafka.protocol
        with patch.object(kafka.protocol.legacy, "create_message",
                               return_value=sentinel.message):
            with patch.object(kafka.protocol.legacy, "create_gzip_message",
                                   return_value=sentinel.gzip_message):
                with patch.object(kafka.protocol.legacy, "create_snappy_message",
                                       return_value=sentinel.snappy_message):
                    yield

Example 2

Project: mailbot Source File: test_mailbot.py
    def test_process_message_no_trigger(self):
        callback = Mock()
        callback.check_rules.return_value = False
        callback_class = Mock(return_value=callback)

        res = self.bot.process_message(sentinel.message, callback_class,
                                       sentinel.rules)

        callback.check_rules.assert_called_once_with()
        self.assertEqual(res, None)

Example 3

Project: ZenPacks.zenoss.Microsoft.Windows Source File: testEventLogDataSource.py
Function: test_onsuccess
    def test_onSuccess(self):
        plugin = EventLogPlugin()

        res = plugin.onSuccess([
            {'Category': u'(0)',
             'EntryType': u'Information',
             'EventID': u'10120',
             'InstanceId': u'468872',
             'MachineName': u'machine',
             'Message': sentinel.message,
             'Source': u'WinRM',
             'TimeGenerated': u'05/30/2014 18:27:22',
             'UserName': u''}
        ], Mock(
            id=sentinel.id,
            datasources=[Mock(params={'eventlog': sentinel.eventlog})],
        ))

        self.assertEquals(len(res['events']), 2)
        self.assertEquals(res['events'][0]['summary'], sentinel.message)
        self.assertEquals(res['events'][0]['eventGroup'], sentinel.eventlog)

Example 4

Project: kafka-python Source File: test_protocol_legacy.py
    def test_create_message_set(self):
        messages = [(1, "k1"), (2, "k2"), (3, "k3")]

        # Default codec is CODEC_NONE. Expect list of regular messages.
        expect = [sentinel.message] * len(messages)
        with self.mock_create_message_fns():
            message_set = create_message_set(messages)
        self.assertEqual(message_set, expect)

        # CODEC_NONE: Expect list of regular messages.
        expect = [sentinel.message] * len(messages)
        with self.mock_create_message_fns():
            message_set = create_message_set(messages, CODEC_NONE)
        self.assertEqual(message_set, expect)

        # CODEC_GZIP: Expect list of one gzip-encoded message.
        expect = [sentinel.gzip_message]
        with self.mock_create_message_fns():
            message_set = create_message_set(messages, CODEC_GZIP)
        self.assertEqual(message_set, expect)

        # CODEC_SNAPPY: Expect list of one snappy-encoded message.
        expect = [sentinel.snappy_message]
        with self.mock_create_message_fns():
            message_set = create_message_set(messages, CODEC_SNAPPY)
        self.assertEqual(message_set, expect)

        # Unknown codec should raise UnsupportedCodecError.
        with self.assertRaises(UnsupportedCodecError):
            create_message_set(messages, -1)