Moving dummy packet connection class to separate module to fix bug.

This commit is contained in:
Michael Woods
2025-02-15 15:12:22 -05:00
parent f2f9cf4a8f
commit 7f295e8fbb
2 changed files with 21 additions and 17 deletions

View File

@@ -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."""

View File

@@ -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")