def test_clean_route(self, stub_broker, postgres_state_middleware):
client = app.test_client()
backend = postgres_state_middleware.backend
backend.set_state(State("id0"))
assert len(backend.get_states()) == 1
client.delete("/messages/states")
assert len(backend.get_states()) == 0
def api_client(state_middleware):
with app.test_client() as client:
yield client
def test_request_cancel(self, stub_broker, cancel_backend):
with app.test_client() as client:
message_id = "1"
stub_broker.add_middleware(Cancel(backend=cancel_backend))
client.post(f"/messages/cancel/{message_id}")
assert cancel_backend.is_canceled(message_id, None)