/
githubmirror
/
salt
Обзор
Документация
Войти
/
githubmirror
/
salt
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/pytests/functional/channel/test_client.py
25 строк
853 B
Daniel A. Wozniak
Use context manager to ensure channel is closed properly
14 ноя 2023, 20:52
14 ноя 2023, 20:52
2ddd5fc
Код
Авторство
О чём код?
import salt.channel.client from tests.support.mock import MagicMock, patch async def test_async_pub_channel_connect_cb(minion_opts): """ Validate connect_callback closes the request channel it creates. """ minion_opts["master_uri"] = "tcp://127.0.0.1:4506" minion_opts["master_ip"] = "127.0.0.1" with salt.channel.client.AsyncPubChannel.factory(minion_opts) as channel: async def send_id(*args): return channel.send_id = send_id 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.__exit__.assert_called_once()