-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathtest_callback_subscription.py
More file actions
48 lines (35 loc) · 1.36 KB
/
test_callback_subscription.py
File metadata and controls
48 lines (35 loc) · 1.36 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import asyncio
import uuid
from natsrpy import CallbackSubscription, Nats
async def test_callback_unsubscribe(nats: Nats) -> None:
subj = uuid.uuid4().hex
async def callback(msg: object) -> None:
pass
sub = await nats.subscribe(subject=subj, callback=callback)
assert isinstance(sub, CallbackSubscription)
await sub.unsubscribe()
async def test_callback_unsubscribe_with_limit(nats: Nats) -> None:
subj = uuid.uuid4().hex
received: list[bytes] = []
event = asyncio.Event()
async def callback(msg: object) -> None:
received.append(msg.payload) # type: ignore[attr-defined]
if len(received) >= 2:
event.set()
sub = await nats.subscribe(subject=subj, callback=callback)
assert isinstance(sub, CallbackSubscription)
await sub.unsubscribe(limit=2)
await nats.publish(subj, b"msg-1")
await nats.publish(subj, b"msg-2")
await asyncio.wait_for(event.wait(), timeout=5.0)
assert set(received) == {b"msg-1", b"msg-2"}
async def test_callback_drain(nats_url: str) -> None:
client = Nats(addrs=[nats_url])
await client.startup()
subj = uuid.uuid4().hex
async def callback(msg: object) -> None:
pass
sub = await client.subscribe(subject=subj, callback=callback)
assert isinstance(sub, CallbackSubscription)
await sub.drain()
await client.shutdown()