async def test_failure_on_subscribe():
convo = SubscribeToStream("my-stream")
await drop_subscription(convo, SubscriptionDropReason.SubscriberMaxCountReached)
with pytest.raises(exn.SubscriptionCreationFailed):
await convo.result
async def test_subscribe_with_context_manager():
conversation_id = uuid4()
convo = SubscribeToStream("my-stream", conversation_id=conversation_id)
queue = TeeQueue()
# Create a subscription with three events in it
await confirm_subscription(convo, output_queue=queue)
for _ in range(0, 3):
await respond_to(convo, event_appeared(uuid4()))
# While reading the events it should not send any messages
async with (await convo.result) as subscription:
events_seen = 0
async for _ in subscription.events:
events_seen += 1
assert not queue.items
if events_seen == 3:
break
# Having exited the context manager it should send
# an unsubscribe message
message = await queue.get()
assert message.command == TcpCommand.UnsubscribeFromStream
assert message.conversation_id == conversation_id
async def test_erroring_from_a_context_manager():
queue = TeeQueue()
convo = SubscribeToStream("my-stream")
await confirm_subscription(convo, output_queue=queue)
async with (await convo.result):
await drop_subscription(convo, SubscriptionDropReason.SubscriberMaxCountReached)
assert queue.items == []
async def test_event_appeared():
convo = SubscribeToStream("my-stream")
event_id = uuid4()
await confirm_subscription(convo, event_number=10, commit_pos=10)
subscription = convo.result.result()
await respond_to(convo, event_appeared(event_id))
event = await subscription.events.anext()
assert event.id == event_id
async def test_confirmation():
convo = SubscribeToStream("my-stream")
await confirm_subscription(convo, event_number=10, commit_pos=10)
subscription = convo.result.result()
assert subscription.stream == "my-stream"
assert subscription.last_event_number == 10
assert subscription.last_commit_position == 10
assert subscription.first_event_number == 10
assert subscription.first_commit_position == 10
async def test_unsubscription():
correlation_id = uuid4()
queue = TeeQueue()
convo = SubscribeToStream("my-stream", conversation_id=correlation_id)
sub = await confirm_subscription(convo, output_queue=queue)
await sub.unsubscribe()
[message] = queue.items
assert message.command == TcpCommand.UnsubscribeFromStream
assert message.conversation_id == correlation_id
async def test_subscription_dropped_mid_stream():
convo = SubscribeToStream("my-stream")
event_id = uuid4()
await confirm_subscription(convo, event_number=10, commit_pos=10)
subscription = convo.result.result()
await respond_to(convo, event_appeared(event_id))
await drop_subscription(convo)
events = [e async for e in subscription.events]
assert len(events) == 1
async def test_subscribe_to_stream():
output = TeeQueue()
convo = SubscribeToStream("my-stream")
await convo.start(output)
[request] = output.items
payload = proto.SubscribeToStream()
payload.ParseFromString(request.payload)
assert request.command == TcpCommand.SubscribeToStream
assert payload.event_stream_id == "my-stream"
assert payload.resolve_link_tos is True
async def test_subscription_failure_mid_stream():
convo = SubscribeToStream("my-stream")
event_id = uuid4()
await confirm_subscription(convo, event_number=10, commit_pos=10)
subscription = convo.result.result()
await respond_to(convo, event_appeared(event_id))
await drop_subscription(convo, SubscriptionDropReason.SubscriberMaxCountReached)
with pytest.raises(exn.SubscriptionFailed):
event = await anext(subscription.events)
assert event.id == event_id
await anext(subscription.events)