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

Fix test when using context manager is used

Browse source
This commit is contained in:
Daniel A. Wozniak 2023-11-08 14:42:56 -07:00 • committed by Pedro Algarvio
parent 6615b5a530
commit 6e5a301ca6
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -17,7 +17,9 @@ async def test_async_pub_channel_connect_cb(minion_opts):
channel._reconnected = True
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.close.assert_called_once()
mock.__exit__.assert_called_once()