diff --git a/tests/adapters/interpreters/wrappers/test_joining_interpreter.py b/tests/adapters/interpreters/wrappers/test_joining_interpreter.py index 38382b371..02f88a98a 100644 --- a/tests/adapters/interpreters/wrappers/test_joining_interpreter.py +++ b/tests/adapters/interpreters/wrappers/test_joining_interpreter.py @@ -1,3 +1,5 @@ +from typing import AnyStr, AsyncIterable, List + import pytest from mock import AsyncMock @@ -5,8 +7,8 @@ @pytest.mark.asyncio -async def test_joining_interpreterconcatenates_as_expected(): - async def multi_resp(msgs): +async def test_joining_interpreter_concatenates_as_expected(): + async def multi_resp(msgs: List[AnyStr]) -> AsyncIterable[AnyStr]: for msg in msgs: yield msg diff --git a/tickit/adapters/interpreters/wrappers/joining_interpreter.py b/tickit/adapters/interpreters/wrappers/joining_interpreter.py index 61bad33dc..af4b7320f 100644 --- a/tickit/adapters/interpreters/wrappers/joining_interpreter.py +++ b/tickit/adapters/interpreters/wrappers/joining_interpreter.py @@ -50,8 +50,7 @@ async def _combine_responses( """ response_list = [response async for response in responses] response = self.response_delimiter.join(response_list) - resp = wrap_as_async_iterable(response) - return resp + return wrap_as_async_iterable(response) async def handle( self, adapter: Adapter, message: AnyStr