示例#1
0
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')
示例#2
0
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")
示例#3
0
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')
示例#4
0
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', {})
示例#5
0
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")
示例#6
0
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", {})
示例#7
0
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')
示例#8
0
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', {})