Fix known issues v2

src/pydo/operations/_patch.py

@@ -9,6 +9,8 @@
 from typing import TYPE_CHECKING
 
 from ._operations import DropletsOperations as Droplets
+from ._operations import KubernetesOperations as Kubernetes
+from ._operations import InvoicesOperations as Invoices
 
 if TYPE_CHECKING:
     # pylint: disable=unused-import,ungrouped-imports
@@ -25,3 +27,27 @@ def patch_sdk():
     you can't accomplish using the techniques described in
     https://aka.ms/azsdk/python/dpcodegen/python/customize
     """
+
+    # Fix kubernetes.get_kubeconfig to return raw YAML content instead of trying to parse as JSON
+    def _get_kubeconfig(self, cluster_id, **kwargs):
+        """Get a Kubernetes config file for the specified cluster."""
+        # Call the original method but with raw response
+        response = self._client.get(
+            f"/v2/kubernetes/clusters/{cluster_id}/kubeconfig",
+            **kwargs
+        )
+        return response.content
+
+    Kubernetes.get_kubeconfig = _get_kubeconfig
+
+    # Fix invoices.get_pdf_by_uuid to return raw PDF content instead of trying to parse as JSON
+    def _get_pdf_by_uuid(self, invoice_uuid, **kwargs):
+        """Get a PDF invoice by UUID."""
+        # Call the original method but with raw response
+        response = self._client.get(
+            f"/v2/customers/my/invoices/{invoice_uuid}/pdf",
+            **kwargs
+        )
+        return response.content
+
+    Invoices.get_pdf_by_uuid = _get_pdf_by_uuid

tests/mocked/test_billing.py

@@ -203,12 +203,11 @@ def test_get_invoice_pdf_by_uuid(mock_client: Client, mock_client_url):
     responses.add(
         responses.GET,
         f"{mock_client_url}/v2/customers/my/invoices/1/pdf",
-        json=expected,
+        body=expected,
     )
     invoices = mock_client.invoices.get_pdf_by_uuid(invoice_uuid=1)
-    list_in = list(invoices)
 
-    assert "group_description" in str(list_in)
+    assert "group_description" in str(invoices)
 
 
 @responses.activate

tests/mocked/test_kubernetes.py

@@ -211,9 +211,6 @@ def test_kubernetes_get_kubeconfig(mock_client: Client, mock_client_url):
     )
 
     config_resp = mock_client.kubernetes.get_kubeconfig(cluster_id)
-    pytest.skip("The operation currently fails to return content.")
-    # TODO: investigate why the generated client doesn't return the response content
-    # It seems to be something to do with the yaml content type.
     assert config_resp.decode("utf-8") == expected