Mock PublishClient instead of TCPPubClient

saltstack · Dec 28, 2023 · 5bc09e5 · 5bc09e5
1 parent 449cca2
commit 5bc09e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/pytests/unit/conftest.py b/tests/pytests/unit/conftest.py
@@ -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