File tree Expand file tree Collapse file tree 2 files changed +21
-4
lines changed Expand file tree Collapse file tree 2 files changed +21
-4
lines changed Original file line number Diff line number Diff line change @@ -46,11 +46,11 @@ def environment_auth(pytestconfig):
46
46
uri = "amqp://localhost:5672" ,
47
47
oauth2_options = OAuth2Options (token = token_string ),
48
48
)
49
- # try:
50
- return environment
49
+ try :
50
+ yield environment
51
51
52
- # finally:
53
- # environment.close()
52
+ finally :
53
+ environment .close ()
54
54
55
55
56
56
@pytest .fixture ()
Original file line number Diff line number Diff line change 4
4
from rabbitmq_amqp_python_client import (
5
5
ConnectionClosed ,
6
6
Environment ,
7
+ QuorumQueueSpecification ,
7
8
RecoveryConfiguration ,
8
9
StreamSpecification ,
9
10
ValidationCodeException ,
@@ -50,6 +51,22 @@ def test_connection_auth(environment_auth: Environment) -> None:
50
51
connection .close ()
51
52
52
53
54
+ def test_connection_auth_with_timeout (environment_auth : Environment ) -> None :
55
+
56
+ connection = environment_auth .connection ()
57
+ connection .dial ()
58
+ time .sleep (3 )
59
+ raised = False
60
+ try :
61
+ management = connection .management ()
62
+ management .declare_queue (QuorumQueueSpecification (name = "test-queue" ))
63
+ except Exception :
64
+ raised = True
65
+
66
+ assert raised is True
67
+ connection .close ()
68
+
69
+
53
70
def test_environment_connections_management () -> None :
54
71
55
72
environment = Environment (uri = "amqp://guest:guest@localhost:5672/" )
You can’t perform that action at this time.
0 commit comments