From caba861dda5beaa02324f16975d3ce7beee3abd4 Mon Sep 17 00:00:00 2001
From: Alan Rominger <arominge@redhat.com>
Date: Wed, 13 Nov 2024 16:27:46 -0500
Subject: [PATCH] Update unit test

---
 awxkit/test/test_ws.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/awxkit/test/test_ws.py b/awxkit/test/test_ws.py
index c2e69fbc51d1..f40767b35720 100644
--- a/awxkit/test/test_ws.py
+++ b/awxkit/test/test_ws.py
@@ -10,11 +10,10 @@
 
 
 def test_explicit_hostname():
-    client = WSClient("token", "some-hostname", 556, False)
+    client = WSClient(hostname="some-hostname", port=556, secure=False)
     assert client.port == 556
     assert client.hostname == "some-hostname"
     assert client._use_ssl == False
-    assert client.token == "token"
 
 
 def test_websocket_suffix():
@@ -35,7 +34,7 @@ def test_urlparsing(url, result):
     with patch("awxkit.ws.config") as mock_config:
         mock_config.base_url = url
 
-        client = WSClient("token")
+        client = WSClient(hostname=None)
         assert client.port == result.port
         assert client.hostname == result.hostname
         assert client._use_ssl == result.secure