saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Use context manager to ensure channel is closed properly

Browse source
This commit is contained in:
Daniel A. Wozniak 2023-11-14 02:52:55 -07:00 • committed by Pedro Algarvio
parent 6e5a301ca6
commit 2ddd5fce46
1 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/pytests/functional/channel/test_client.py
View file

@ -8,18 +8,18 @@ async def test_async_pub_channel_connect_cb(minion_opts):
"""
minion_opts["master_uri"] = "tcp://127.0.0.1:4506"
minion_opts["master_ip"] = "127.0.0.1"
channel = salt.channel.client.AsyncPubChannel.factory(minion_opts)
with salt.channel.client.AsyncPubChannel.factory(minion_opts) as channel:
async def send_id(*args):
return
async def send_id(*args):
return
channel.send_id = send_id
channel._reconnected = True
channel.send_id = send_id
channel._reconnected = True
mock = MagicMock(salt.channel.client.AsyncReqChannel)
mock.__enter__ = lambda self: mock
mock = MagicMock(salt.channel.client.AsyncReqChannel)
mock.__enter__ = lambda self: mock
with patch("salt.channel.client.AsyncReqChannel.factory", return_value=mock):
await channel.connect_callback(None)
mock.send.assert_called_once()
mock.__exit__.assert_called_once()
with patch("salt.channel.client.AsyncReqChannel.factory", return_value=mock):
await channel.connect_callback(None)
mock.send.assert_called_once()
mock.__exit__.assert_called_once()