From 1c64a6df6c63375f55ef97330a8eba9968d92f78 Mon Sep 17 00:00:00 2001 From: Michael Woods Date: Sat, 15 Feb 2025 15:20:34 -0500 Subject: [PATCH] Moved dummyconnection back out of common/__init__.py --- src/packetserver/common/__init__.py | 15 --------------- src/packetserver/common/testing.py | 20 ++++++++++++++++++++ 2 files changed, 20 insertions(+), 15 deletions(-) create mode 100644 src/packetserver/common/testing.py diff --git a/src/packetserver/common/__init__.py b/src/packetserver/common/__init__.py index 2902993..fcf7953 100644 --- a/src/packetserver/common/__init__.py +++ b/src/packetserver/common/__init__.py @@ -82,21 +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..dd93d44 --- /dev/null +++ b/src/packetserver/common/testing.py @@ -0,0 +1,20 @@ +from . import PacketServerConnection +from pe.connect import ConnectionState +from msgpack import Unpacker +from typing import Union, Self +import logging + +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