From d886878f15c08564825fbfac1301dd342e82a050 Mon Sep 17 00:00:00 2001
From: Jacob Henner <code@ventricle.us>
Date: Sun, 5 Jan 2025 04:25:56 +0000
Subject: [PATCH] Add e2e custom object test

---
 kubernetes_asyncio/e2e_test/test_client.py | 90 +++++++++++++++++++++-
 1 file changed, 89 insertions(+), 1 deletion(-)

diff --git a/kubernetes_asyncio/e2e_test/test_client.py b/kubernetes_asyncio/e2e_test/test_client.py
index 8fb22f38..ecf3910a 100644
--- a/kubernetes_asyncio/e2e_test/test_client.py
+++ b/kubernetes_asyncio/e2e_test/test_client.py
@@ -17,7 +17,9 @@
 from unittest import IsolatedAsyncioTestCase
 
 from kubernetes_asyncio.client import api_client
-from kubernetes_asyncio.client.api import core_v1_api
+from kubernetes_asyncio.client.api import (
+    apiextensions_v1_api, core_v1_api, custom_objects_api,
+)
 from kubernetes_asyncio.e2e_test import base
 from kubernetes_asyncio.stream import WsApiClient
 
@@ -187,6 +189,92 @@ async def test_service_apis(self):
             name=name, body={}, namespace="default"
         )
 
+    async def test_custom_objects_api(self):
+        client = api_client.ApiClient(configuration=self.config)
+
+        apiextensions_api_client = apiextensions_v1_api.ApiextensionsV1Api(client)
+        custom_objects_api_client = custom_objects_api.CustomObjectsApi(client)
+
+        name = 'clusterchangemes.apps.example.com'
+        crd_manifest = {
+            "apiVersion": "apiextensions.k8s.io/v1",
+            "kind": "CustomResourceDefinition",
+            "metadata": {
+                "name": name,
+            },
+            "spec": {
+                "group": "apps.example.com",
+                "names": {
+                    "kind": "ClusterChangeMe",
+                    "listKind": "ClusterChangeMeList",
+                    "plural": "clusterchangemes",
+                    "singular": "clusterchangeme",
+                },
+                "scope": "Cluster",
+                "versions": [
+                    {
+                        "name": "v1",
+                        "served": True,
+                        "storage": True,
+                        "schema": {
+                            "openAPIV3Schema": {
+                                "type": "object",
+                                "properties": {
+                                    "spec": {
+                                        "type": "object",
+                                        "properties": {
+                                            "size": {"type": "integer"}
+                                        },
+                                    }
+                                },
+                            }
+                        },
+                    }
+                ],
+            },
+        }
+        custom_object_manifest = {
+            'apiVersion': 'apps.example.com/v1',
+            'kind': 'ClusterChangeMe',
+            'metadata': {
+                'name': "changeme-name",
+            },
+            'spec': {}
+        }
+
+        await apiextensions_api_client.create_custom_resource_definition(
+            crd_manifest
+        )
+
+        await apiextensions_api_client.read_custom_resource_definition(
+            crd_manifest["metadata"]["name"]
+        )
+
+        await custom_objects_api_client.create_cluster_custom_object(
+            crd_manifest["spec"]["group"],
+            crd_manifest["spec"]["versions"][0]["name"],
+            crd_manifest["spec"]["names"]["plural"],
+            custom_object_manifest
+        )
+
+        # json merge patch (implied)
+        resp = await custom_objects_api_client.patch_cluster_custom_object(
+            group=crd_manifest["spec"]["group"],
+            version=crd_manifest["spec"]["versions"][0]["name"],
+            plural=crd_manifest["spec"]["names"]["plural"],
+            name=custom_object_manifest["metadata"]["name"],
+            body={
+                "spec": {
+                    "size": 0
+                }
+            },
+        )
+        self.assertEqual(resp["spec"]["size"], 0)
+
+        await apiextensions_api_client.delete_custom_resource_definition(
+            crd_manifest["metadata"]["name"]
+        )
+
     async def test_replication_controller_apis(self):
         client = api_client.ApiClient(configuration=self.config)
         api = core_v1_api.CoreV1Api(client)