- with mock.patch('websockets.connect', base.AsyncMock(return_value=ws)):
- await con.open()
- actual_responses = []
- for i in range(3):
- actual_responses.append(await con.rpc({'version': 1}))
- assert actual_responses == expected_responses
+ try:
+ with mock.patch('websockets.connect', base.AsyncMock(return_value=ws)):
+ await con.open()
+ actual_responses = []
+ for i in range(3):
+ actual_responses.append(await con.rpc({'version': 1}))
+ assert actual_responses == expected_responses
+ finally:
+ await con.close()