Coverage for faststream / rabbit / helpers / state.py: 80%

1from dataclasses import dataclass (empty)

2from typing import TYPE_CHECKING, Protocol (empty)

3 

4from faststream.exceptions import IncorrectState (empty)

5 

6if TYPE_CHECKING: (empty)

7 from aio_pika import RobustConnection 

8 

9 

10class ConnectionState(Protocol): (empty)

11 @property (empty)

12 def connection(self) -> "RobustConnection": ... (empty)

13 

14 

15class EmptyConnectionState: (empty)

16 __slots__ = () (empty)

17 

18 @property (empty)

19 def connection(self) -> "RobustConnection": (empty)

20 msg = "You should connect broker first." 

21 raise IncorrectState(msg) 

22 

23 

24@dataclass(slots=True) (empty)

25class ConnectedState: (empty)

26 connection: "RobustConnection" (empty)