def test_apply_message_translator_error(dummy_provider): translator = StringMessageTranslator() translator.translate = mock.Mock(return_value={'content': '', 'metadata': {}}) route = Route(dummy_provider, mock.Mock(), message_translator=translator) with pytest.raises(ValueError): route.apply_message_translator('message') assert translator.translate.called translator.translate.assert_called_once_with('message')
def test_apply_message_translator_error(dummy_provider): translator = StringMessageTranslator() translator.translate = mock.Mock(return_value={ "content": "", "metadata": {} }) route = Route(dummy_provider, mock.Mock(), message_translator=translator) with pytest.raises(ValueError): route.apply_message_translator("message") assert translator.translate.called translator.translate.assert_called_once_with("message")
def test_apply_message_translator(dummy_provider): translator = StringMessageTranslator() translator.translate = mock.Mock(return_value={'content': 'foobar', 'metadata': {}}) route = Route(dummy_provider, mock.Mock(), message_translator=translator) translated = route.apply_message_translator('message') assert translated['content'] == 'foobar' assert translated['metadata'] == {} assert translator.translate.called translator.translate.assert_called_once_with('message')
async def test_deliver_with_message_translator(dummy_provider): mock_handler = CoroutineMock(return_value=True) route = Route(dummy_provider, mock_handler) route.apply_message_translator = mock.Mock(return_value={'content': 'whatever', 'metadata': {}}) result = await route.deliver('test') assert result is True assert route.apply_message_translator.called assert mock_handler.called mock_handler.assert_called_once_with('whatever', {})
def test_apply_message_translator(dummy_provider): translator = StringMessageTranslator() translator.translate = mock.Mock(return_value={ "content": "foobar", "metadata": {} }) route = Route(dummy_provider, mock.Mock(), message_translator=translator) translated = route.apply_message_translator("message") assert translated["content"] == "foobar" assert translated["metadata"] == {} assert translator.translate.called translator.translate.assert_called_once_with("message")
async def test_deliver_with_message_translator(dummy_provider): mock_handler = AsyncMock(return_value=True) route = Route(dummy_provider, mock_handler) route.apply_message_translator = mock.Mock(return_value={ "content": "whatever", "metadata": {} }) result = await route.deliver("test") assert result is True assert route.apply_message_translator.called assert mock_handler.called mock_handler.assert_called_once_with("whatever", {})
def test_apply_message_translator(dummy_provider): translator = StringMessageTranslator() translator.translate = mock.Mock(return_value={ 'content': 'foobar', 'metadata': {} }) route = Route(dummy_provider, mock.Mock(), message_translator=translator) translated = route.apply_message_translator('message') assert translated['content'] == 'foobar' assert translated['metadata'] == {} assert translator.translate.called translator.translate.assert_called_once_with('message')
async def test_deliver_with_message_translator(dummy_provider): mock_handler = CoroutineMock(return_value=True) route = Route(dummy_provider, mock_handler) route.apply_message_translator = mock.Mock(return_value={ 'content': 'whatever', 'metadata': {} }) result = await route.deliver('test') assert result is True assert route.apply_message_translator.called assert mock_handler.called mock_handler.assert_called_once_with('whatever', {})