From 5bc09e5470a4e59270af2bfa2b0da289eba88a91 Mon Sep 17 00:00:00 2001
From: "Daniel A. Wozniak" <dwozniak@vmware.com>
Date: Sat, 23 Dec 2023 09:35:15 -0700
Subject: [PATCH] Mock PublishClient instead of TCPPubClient

---
 tests/pytests/unit/conftest.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/pytests/unit/conftest.py b/tests/pytests/unit/conftest.py
index e19db6cf5a3..b3dc4bb32eb 100644
--- 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