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 4

Mock PublishClient instead of TCPPubClient

Browse source
This commit is contained in:
Daniel A. Wozniak 2023-12-23 09:35:15 -07:00 • committed by Daniel Wozniak
parent 449cca258f
commit 5bc09e5470
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/pytests/unit/conftest.py
View file

@ -68,10 +68,10 @@ def syndic_opts(tmp_path):
@pytest.fixture
def mocked_tcp_pub_client():
transport = MagicMock(spec=salt.transport.tcp.TCPPubClient)
transport = MagicMock(spec=salt.transport.tcp.PublishClient)
transport.connect = MagicMock()
future = asyncio.Future()
transport.connect.return_value = future
future.set_result(True)
with patch("salt.transport.tcp.TCPPubClient", transport):
with patch("salt.transport.tcp.PublishClient", transport):
yield