diff --git a/simple_message_channels/__init__.py b/simple_message_channels/__init__.py index 96eda65..037bc65 100644 --- a/simple_message_channels/__init__.py +++ b/simple_message_channels/__init__.py @@ -1,3 +1,2 @@ """simple-message-channels module.""" - from simple_message_channels.smc import SimpleMessageChannel # noqa diff --git a/simple_message_channels/smc.py b/simple_message_channels/smc.py index 58b99ba..3e7b2b1 100644 --- a/simple_message_channels/smc.py +++ b/simple_message_channels/smc.py @@ -1,11 +1,10 @@ """Sans I/O wire protocol for Hypercore""" - -__all__ = ["SimpleMessageChannel"] - from typing import List, Optional import attr +import pyvarint +__all__ = ["SimpleMessageChannel"] @attr.s(auto_attribs=True) class SimpleMessageChannel: diff --git a/test/test_smc.py b/test/test_smc.py index 30223bd..384364b 100644 --- a/test/test_smc.py +++ b/test/test_smc.py @@ -3,15 +3,3 @@ async def test_send_produces_bytes(smc): payload = await smc.send(channel, type, message) assert isinstance(payload, bytes) - - -# async def test_recv_context_manager(smc): -# channel, type, message = 0, 0, b"abc" - -# payload = await smc.send(channel, type, message) -# assert isinstance(payload, bytes) - -# async with smc.recv(payload) as response: -# assert response.channel == channel -# assert response.type == type -# assert response.message == message