Coverage for examples / fastapi_integration / testing.py: 86%

1import pytest (empty)

2import pytest_asyncio (empty)

3from fastapi.exceptions import RequestValidationError (empty)

4 

5from faststream.rabbit import TestRabbitBroker (empty)

6 

7from .app import handler, publisher, router (empty)

8 

9 

10@pytest_asyncio.fixture (empty)

11async def broker(): (empty)

12 async with TestRabbitBroker(router.broker) as br: 

13 yield br 

14 

15 

16@pytest.mark.asyncio() (empty)

17async def test_incorrect(broker): (empty)

18 with pytest.raises(RequestValidationError): (empty)

19 await broker.publish("user-id", "test-q") (empty)

20 

21 

22@pytest.mark.asyncio() (empty)

23async def test_handler(broker): (empty)

24 user_id = 1 (empty)

25 

26 await broker.publish(user_id, "test-q") (empty)

27 

28 handler.mock.assert_called_once_with(user_id) (empty)

29 publisher.mock.assert_called_once_with(f"{user_id} created") (empty)