fix: correct NatsJS rpc

Lancetnik · Aug 21, 2023 · 8f3c384 · 8f3c384
1 parent f6f6acb
commit 8f3c384
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/propan/brokers/nats/nats_js_broker.py b/propan/brokers/nats/nats_js_broker.py
@@ -283,9 +283,9 @@ async def publish(
 
             token = client._nuid.next()
             token.extend(token_hex(2).encode())
-
+            reply_to = token.decode()
             future: asyncio.Future[Msg] = asyncio.Future()
-            sub = await client.subscribe(token.decode(), future=future, max_msgs=1)
+            sub = await client.subscribe(reply_to, future=future, max_msgs=1)
             await sub.unsubscribe(limit=1)
 
         if raise_timeout:

diff --git a/tests/brokers/base/rpc.py b/tests/brokers/base/rpc.py
@@ -18,7 +18,7 @@ async def m():  # pragma: no cover
             await full_broker.start()
 
             r = await full_broker.publish(
-                "hello", queue, callback_timeout=3, callback=True
+                "hello", queue, callback_timeout=3, callback=True, raise_timeout=True
             )
             assert r == "1"