fix: mypy issues

Author: jorwoods

test/test_endpoint.py

@@ -3,6 +3,7 @@
 import requests
 
 import tableauserverclient as TSC
+from tableauserverclient.server.endpoint import Endpoint
 
 import requests_mock
 
@@ -23,7 +24,7 @@ def server():
 
 def test_fallback_request_logic(server: TSC.Server) -> None:
     url = "http://test/"
-    endpoint = TSC.server.Endpoint(server)
+    endpoint = Endpoint(server)
     with requests_mock.mock() as m:
         m.get(url)
         response = endpoint.get_request(url=url)
@@ -32,7 +33,7 @@ def test_fallback_request_logic(server: TSC.Server) -> None:
 
 def test_user_friendly_request_returns(server: TSC.Server) -> None:
     url = "http://test/"
-    endpoint = TSC.server.Endpoint(server)
+    endpoint = Endpoint(server)
     with requests_mock.mock() as m:
         m.get(url)
         response = endpoint.send_request_while_show_progress_threaded(
@@ -44,14 +45,14 @@ def test_user_friendly_request_returns(server: TSC.Server) -> None:
 def test_blocking_request_raises_request_error(server: TSC.Server) -> None:
     with pytest.raises(requests.exceptions.ConnectionError):
         url = "http://test/"
-        endpoint = TSC.server.Endpoint(server)
+        endpoint = Endpoint(server)
         response = endpoint._blocking_request(endpoint.parent_srv.session.get, url=url)
         assert response is not None
 
 
 def test_get_request_stream(server: TSC.Server) -> None:
     url = "http://test/"
-    endpoint = TSC.server.Endpoint(server)
+    endpoint = Endpoint(server)
     with requests_mock.mock() as m:
         m.get(url, headers={"Content-Type": "application/octet-stream"})
         response = endpoint.get_request(url, parameters={"stream": True})
@@ -65,15 +66,15 @@ class FakeResponse:
         content = b"\x1337" * 1000
         status_code = 200
 
-    endpoint = TSC.server.Endpoint(server)
+    endpoint = Endpoint(server)
     server_response = FakeResponse()
-    log = endpoint.log_response_safely(server_response)
+    log = endpoint.log_response_safely(server_response)  # type: ignore
     assert log.find("[Truncated File Contents]") > 0
 
 
 def test_set_user_agent_from_options_headers(server: TSC.Server) -> None:
     params = {"User-Agent": "1", "headers": {"User-Agent": "2"}}
-    result = TSC.server.Endpoint.set_user_agent(params)
+    result = Endpoint.set_user_agent(params)
     # it should use the value under 'headers' if more than one is given
     print(result)
     print(result["headers"]["User-Agent"])
@@ -82,11 +83,11 @@ def test_set_user_agent_from_options_headers(server: TSC.Server) -> None:
 
 def test_set_user_agent_from_options(server: TSC.Server) -> None:
     params = {"headers": {"User-Agent": "2"}}
-    result = TSC.server.Endpoint.set_user_agent(params)
+    result = Endpoint.set_user_agent(params)
     assert result["headers"]["User-Agent"] == "2"
 
 
 def test_set_user_agent_when_blank(server: TSC.Server) -> None:
-    params = {"headers": {}}
-    result = TSC.server.Endpoint.set_user_agent(params)
+    params = {"headers": {}}  # type: ignore
+    result = Endpoint.set_user_agent(params)
     assert result["headers"]["User-Agent"].startswith("Tableau Server Client")