chore: pytestify requests

test/request_factory/test_datasource_requests.py

@@ -1,15 +1,13 @@
-import unittest
 import tableauserverclient as TSC
 import tableauserverclient.server.request_factory as TSC_RF
 from tableauserverclient import DatasourceItem
 
 
-class DatasourceRequestTests(unittest.TestCase):
-    def test_generate_xml(self):
-        datasource_item: TSC.DatasourceItem = TSC.DatasourceItem("name")
-        datasource_item.name = "a ds"
-        datasource_item.description = "described"
-        datasource_item.use_remote_query_agent = False
-        datasource_item.ask_data_enablement = DatasourceItem.AskDataEnablement.Enabled
-        datasource_item.project_id = "testval"
-        TSC_RF.RequestFactory.Datasource._generate_xml(datasource_item)
+def test_generate_xml():
+    datasource_item: TSC.DatasourceItem = TSC.DatasourceItem("name")
+    datasource_item.name = "a ds"
+    datasource_item.description = "described"
+    datasource_item.use_remote_query_agent = False
+    datasource_item.ask_data_enablement = DatasourceItem.AskDataEnablement.Enabled
+    datasource_item.project_id = "testval"
+    TSC_RF.RequestFactory.Datasource._generate_xml(datasource_item)

test/request_factory/test_workbook_requests.py

@@ -1,55 +1,58 @@
-import unittest
 import tableauserverclient as TSC
 import tableauserverclient.server.request_factory as TSC_RF
 from tableauserverclient.helpers.strings import redact_xml
 import pytest
 import sys
 
 
-class WorkbookRequestTests(unittest.TestCase):
-    def test_embedded_extract_req(self):
-        include_all = True
-        embedded_datasources = None
-        xml_result = TSC_RF.RequestFactory.Workbook.embedded_extract_req(include_all, embedded_datasources)
-
-    def test_generate_xml(self):
-        workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
-        TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item)
-
-    def test_generate_xml_invalid_connection(self):
-        workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
-        conn = TSC.ConnectionItem()
-        with self.assertRaises(ValueError):
-            request = TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item, connections=[conn])
-
-    def test_generate_xml_invalid_connection_credentials(self):
-        workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
-        conn = TSC.ConnectionItem()
-        conn.server_address = "address"
-        creds = TSC.ConnectionCredentials("username", "password")
-        creds.name = None
-        conn.connection_credentials = creds
-        with self.assertRaises(ValueError):
-            request = TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item, connections=[conn])
-
-    def test_generate_xml_valid_connection_credentials(self):
-        workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
-        conn = TSC.ConnectionItem()
-        conn.server_address = "address"
-        creds = TSC.ConnectionCredentials("username", "DELETEME")
-        conn.connection_credentials = creds
+def test_embedded_extract_req() -> None:
+    include_all = True
+    embedded_datasources = None
+    xml_result = TSC_RF.RequestFactory.Workbook.embedded_extract_req(include_all, embedded_datasources)
+
+
+def test_generate_xml() -> None:
+    workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
+    TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item)
+
+
+def test_generate_xml_invalid_connection() -> None:
+    workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
+    conn = TSC.ConnectionItem()
+    with pytest.raises(ValueError):
         request = TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item, connections=[conn])
-        assert request.find(b"DELETEME") > 0
-
-    def test_redact_passwords_in_xml(self):
-        if sys.version_info < (3, 7):
-            pytest.skip("Redaction is only implemented for 3.7+.")
-        workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
-        conn = TSC.ConnectionItem()
-        conn.server_address = "address"
-        creds = TSC.ConnectionCredentials("username", "DELETEME")
-        conn.connection_credentials = creds
+
+
+def test_generate_xml_invalid_connection_credentials() -> None:
+    workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
+    conn = TSC.ConnectionItem()
+    conn.server_address = "address"
+    creds = TSC.ConnectionCredentials("username", "password")
+    creds.name = None
+    conn.connection_credentials = creds
+    with pytest.raises(ValueError):
         request = TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item, connections=[conn])
-        redacted = redact_xml(request)
-        assert request.find(b"DELETEME") > 0, request
-        assert redacted.find(b"DELETEME") == -1, redacted
+
+
+def test_generate_xml_valid_connection_credentials() -> None:
+    workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
+    conn = TSC.ConnectionItem()
+    conn.server_address = "address"
+    creds = TSC.ConnectionCredentials("username", "DELETEME")
+    conn.connection_credentials = creds
+    request = TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item, connections=[conn])
+    assert request.find(b"DELETEME") > 0
+
+
+def test_redact_passwords_in_xml() -> None:
+    if sys.version_info < (3, 7):
+        pytest.skip("Redaction is only implemented for 3.7+.")
+    workbook_item: TSC.WorkbookItem = TSC.WorkbookItem("name", "project_id")
+    conn = TSC.ConnectionItem()
+    conn.server_address = "address"
+    creds = TSC.ConnectionCredentials("username", "DELETEME")
+    conn.connection_credentials = creds
+    request = TSC_RF.RequestFactory.Workbook._generate_xml(workbook_item, connections=[conn])
+    redacted = redact_xml(request)
+    assert request.find(b"DELETEME") > 0, request
+    assert redacted.find(b"DELETEME") == -1, redacted

test/test_datasource.py

@@ -895,7 +895,8 @@ def test_publish_description(server: TSC.Server) -> None:
     ds_elem = body.find(".//datasource")
     assert ds_elem is not None
     assert ds_elem.attrib["description"] == "Sample description"
-
+
+
 def test_get_datasource_no_owner(server: TSC.Server) -> None:
     with requests_mock.mock() as m:
         m.get(server.datasources.baseurl, text=GET_NO_OWNER.read_text())

test/test_requests.py

@@ -1,61 +1,69 @@
-import re
-import unittest
+from urllib.parse import parse_qs
 
+import pytest
 import requests
 import requests_mock
 
 import tableauserverclient as TSC
 from tableauserverclient.server.endpoint.exceptions import InternalServerError, NonXMLResponseError
 
 
-class RequestTests(unittest.TestCase):
-    def setUp(self):
-        self.server = TSC.Server("http://test", False)
-
-        # Fake sign in
-        self.server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
-        self.server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
-
-        self.baseurl = self.server.workbooks.baseurl
-
-    def test_make_get_request(self):
-        with requests_mock.mock() as m:
-            m.get(requests_mock.ANY)
-            url = "http://test/api/2.3/sites/dad65087-b08b-4603-af4e-2887b8aafc67/workbooks"
-            opts = TSC.RequestOptions(pagesize=13, pagenumber=15)
-            resp = self.server.workbooks.get_request(url, request_object=opts)
-
-            self.assertTrue(re.search("pagesize=13", resp.request.query))
-            self.assertTrue(re.search("pagenumber=15", resp.request.query))
-
-    def test_make_post_request(self):
-        with requests_mock.mock() as m:
-            m.post(requests_mock.ANY)
-            url = "http://test/api/2.3/sites/dad65087-b08b-4603-af4e-2887b8aafc67/workbooks"
-            resp = self.server.workbooks._make_request(
-                requests.post,
-                url,
-                content=b"1337",
-                auth_token="j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM",
-                content_type="multipart/mixed",
-            )
-            self.assertEqual(resp.request.headers["x-tableau-auth"], "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM")
-            self.assertEqual(resp.request.headers["content-type"], "multipart/mixed")
-            self.assertTrue(re.search("Tableau Server Client", resp.request.headers["user-agent"]))
-            self.assertEqual(resp.request.body, b"1337")
-
-    # Test that 500 server errors are handled properly
-    def test_internal_server_error(self):
-        self.server.version = "3.2"
-        server_response = "500: Internal Server Error"
-        with requests_mock.mock() as m:
-            m.register_uri("GET", self.server.server_info.baseurl, status_code=500, text=server_response)
-            self.assertRaisesRegex(InternalServerError, server_response, self.server.server_info.get)
-
-    # Test that non-xml server errors are handled properly
-    def test_non_xml_error(self):
-        self.server.version = "3.2"
-        server_response = "this is not xml"
-        with requests_mock.mock() as m:
-            m.register_uri("GET", self.server.server_info.baseurl, status_code=499, text=server_response)
-            self.assertRaisesRegex(NonXMLResponseError, server_response, self.server.server_info.get)
+@pytest.fixture(scope="function")
+def server():
+    """Fixture to create a TSC.Server instance for testing."""
+    server = TSC.Server("http://test", False)
+
+    # Fake signin
+    server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
+    server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
+
+    return server
+
+
+def test_make_get_request(server: TSC.Server) -> None:
+    with requests_mock.mock() as m:
+        m.get(requests_mock.ANY)
+        url = "http://test/api/2.3/sites/dad65087-b08b-4603-af4e-2887b8aafc67/workbooks"
+        opts = TSC.RequestOptions(pagesize=13, pagenumber=15)
+        resp = server.workbooks.get_request(url, request_object=opts)
+
+        query = parse_qs(resp.request.query)
+        assert query.get("pagesize") == ["13"]
+        assert query.get("pagenumber") == ["15"]
+
+
+def test_make_post_request(server: TSC.Server) -> None:
+    with requests_mock.mock() as m:
+        m.post(requests_mock.ANY)
+        url = "http://test/api/2.3/sites/dad65087-b08b-4603-af4e-2887b8aafc67/workbooks"
+        resp = server.workbooks._make_request(
+            requests.post,
+            url,
+            content=b"1337",
+            auth_token="j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM",
+            content_type="multipart/mixed",
+        )
+        assert resp.request.headers["x-tableau-auth"] == "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
+        assert resp.request.headers["content-type"] == "multipart/mixed"
+        assert "Tableau Server Client" in resp.request.headers["user-agent"]
+        assert resp.request.body == b"1337"
+
+
+# Test that 500 server errors are handled properly
+def test_internal_server_error(server: TSC.Server) -> None:
+    server.version = "3.2"
+    server_response = "500: Internal Server Error"
+    with requests_mock.mock() as m:
+        m.register_uri("GET", server.server_info.baseurl, status_code=500, text=server_response)
+        with pytest.raises(InternalServerError, match=server_response):
+            server.server_info.get()
+
+
+# Test that non-xml server errors are handled properly
+def test_non_xml_error(server: TSC.Server) -> None:
+    server.version = "3.2"
+    server_response = "this is not xml"
+    with requests_mock.mock() as m:
+        m.register_uri("GET", server.server_info.baseurl, status_code=499, text=server_response)
+        with pytest.raises(NonXMLResponseError, match=server_response):
+            server.server_info.get()