From 0ede71eda54ab48ec69dacb46a69a80c99d971ae Mon Sep 17 00:00:00 2001 From: "Daniel A. Wozniak" Date: Tue, 4 Jul 2023 00:38:19 -0700 Subject: [PATCH] Fix batch test --- tests/pytests/functional/cli/test_batch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pytests/functional/cli/test_batch.py b/tests/pytests/functional/cli/test_batch.py index e721b729cfe..50619735c87 100644 --- a/tests/pytests/functional/cli/test_batch.py +++ b/tests/pytests/functional/cli/test_batch.py @@ -64,7 +64,7 @@ class MockSubscriber: def __init__(self, *args, **kwargs): return - def read(self, timeout=None): + def recv(self, timeout=None): """ Mock IPCMessageSubcriber read method. @@ -170,7 +170,7 @@ def test_batch_issue_56273(): "extension_modules": "", "failhard": True, } - with patch("salt.transport.ipc.IPCMessageSubscriber", MockSubscriber): + with patch("salt.transport.tcp.TCPPubClient", MockSubscriber): batch = salt.cli.batch.Batch(opts, quiet=True) with patch.object(batch.local, "pub", Mock(side_effect=mock_pub)): with patch.object(