add a stream publisher for the client

praekeltfoundation · Nov 20, 2024 · 3144bac · 3144bac
1 parent 3fc9d56
commit 3144bac
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/vumi2/transports/smpp/client.py b/src/vumi2/transports/smpp/client.py
@@ -21,6 +21,7 @@
 from smpp.pdu.pdu_encoding import PDUEncoder  # type: ignore
 from smpp.pdu.pdu_types import PDU, AddrNpi, AddrTon, CommandStatus  # type: ignore
 from trio import (
+    MemoryReceiveChannel,
     MemorySendChannel,
     Nursery,
     SocketStream,
@@ -98,6 +99,8 @@ def __init__(
         self.buffer = bytearray()
         self.responses: dict[int, MemorySendChannel] = {}
         self.encoder = PDUEncoder()
+        self.pdu_send_channel: MemorySendChannel
+        self.pdu_receive_channel: MemoryReceiveChannel
         self.pdu_send_channel, self.pdu_receive_channel = open_memory_channel(0)
 
     async def start(self) -> None:
@@ -243,6 +246,8 @@ async def send_pdu(
             await self.pdu_send_channel.send(pdu)
             return None
 
+        send_channel: MemorySendChannel
+        receive_channel: MemoryReceiveChannel
         send_channel, receive_channel = open_memory_channel(0)
         self.responses[pdu.seqNum] = send_channel