diff --git a/src/packetserver/common/__init__.py b/src/packetserver/common/__init__.py index 98aede6..0350575 100644 --- a/src/packetserver/common/__init__.py +++ b/src/packetserver/common/__init__.py @@ -82,23 +82,6 @@ class PacketServerConnection(Connection): def query_accept(cls, port, call_from, call_to): return True - -class DummyPacketServerConnection(PacketServerConnection): - - def __init__(self, call_from: str, call_to: str, incoming=False): - super().__init__(0, call_from, call_to, incoming=incoming) - self.sent_data = Unpacker() - self._state = ConnectionState.CONNECTED - - @property - def state(self): - return self._state - - def send_data(self, data: Union[bytes, bytearray]): - self.sent_data.feed(data) - logging.debug(f"Sender added {data} to self.sent_data.feed") - - class Message: """Base class for communication encapsulated in msgpack objects.""" diff --git a/src/packetserver/common/testing.py b/src/packetserver/common/testing.py new file mode 100644 index 0000000..cfa9618 --- /dev/null +++ b/src/packetserver/common/testing.py @@ -0,0 +1,21 @@ +from . import PacketServerConnection +from msgpack import Unpacker +from msgpack import packb, unpackb +from pe.connect import Connection, ConnectionState +import logging +from typing import Union + +class DummyPacketServerConnection(PacketServerConnection): + + def __init__(self, call_from: str, call_to: str, incoming=False): + super().__init__(0, call_from, call_to, incoming=incoming) + self.sent_data = Unpacker() + self._state = ConnectionState.CONNECTED + + @property + def state(self): + return self._state + + def send_data(self, data: Union[bytes, bytearray]): + self.sent_data.feed(data) + logging.debug(f"Sender added {data} to self.sent_data.feed") \ No newline at end of file