def test_consume_timeout_without_consumers( self, fake_parse_url, fake_ensure_connection): deadline = time.time() + 3 conn = kafka_driver.Connection( self.conf, '', kafka_driver.PURPOSE_LISTEN) conn.consumer = mock.MagicMock(return_value=None) self.assertRaises(driver_common.Timeout, conn.consume, timeout=3) self.assertEqual(0, int(deadline - time.time()))
def test_consume_with_default_timeout( self, fake_parse_url, fake_ensure_connection): deadline = time.time() + 1 conn = kafka_driver.Connection( self.conf, '', kafka_driver.PURPOSE_LISTEN) conn.consumer = mock.MagicMock() conn.consumer.fetch_messages = mock.MagicMock(return_value=iter([])) self.assertRaises(driver_common.Timeout, conn.consume) self.assertEqual(0, int(deadline - time.time()))
def test_consume(self, fake_parse_url, fake_ensure_connection): fake_message = { "context": {"fake": "fake_context_1"}, "message": {"fake": "fake_message_1"}} conn = kafka_driver.Connection( self.conf, '', kafka_driver.PURPOSE_LISTEN) conn.consumer = mock.MagicMock() conn.consumer.fetch_messages = mock.MagicMock( return_value=iter([jsonutils.dumps(fake_message)])) self.assertEqual(fake_message, jsonutils.loads(conn.consume()[0])) self.assertEqual(1, len(conn.consumer.fetch_messages.mock_calls))