codemie-test-harness 0.1.122__py3-none-any.whl → 0.1.123__py3-none-any.whl

tests/test_data/integrations_test_data.py

@@ -91,6 +91,10 @@ valid_integrations = [
         CredentialTypes.SERVICE_NOW,
         CredentialsUtil.servicenow_credentials(),
     ),
+    (
+        CredentialTypes.KUBERNETES,
+        CredentialsUtil.kubernetes_credentials(),
+    ),
 ]
 
 testable_integrations = [
@@ -146,6 +150,10 @@ testable_integrations = [
         CredentialTypes.SERVICE_NOW,
         CredentialsUtil.servicenow_credentials(),
     ),
+    (
+        CredentialTypes.KUBERNETES,
+        CredentialsUtil.kubernetes_credentials(),
+    ),
 ]
 
 invalid_integrations = [
@@ -204,4 +212,9 @@ invalid_integrations = [
         CredentialsUtil.invalid_servicenow_credentials(),
         'ServiceNow tool exception. Status: 401. Response: {"error":{"message":"User Not Authenticated","detail":"Required to provide Auth information"}',
     ),
+    (
+        CredentialTypes.KUBERNETES,
+        CredentialsUtil.invalid_kubernetes_credentials(),
+        "Error: (401)\nReason: Unauthorized",
+    ),
 ]

tests/utils/aws_parameters_store.py

@@ -537,3 +537,24 @@ class CredentialsUtil:
             if cred.key == "api_key":
                 cred.value = "wrong_token"
         return credentials
+
+    @staticmethod
+    def kubernetes_credentials() -> List[CredentialValues]:
+        kubernetes_creds = AwsParameterStore.get_cloud_provider_credentials(
+            "kubernetes"
+        )["sandbox"]["kubernetes"]
+        return [
+            CredentialValues(key="url", value=CredentialsUtil.AUTO_GENERATED),
+            CredentialValues(key="kubernetes_url", value=kubernetes_creds.get("url")),
+            CredentialValues(
+                key="kubernetes_token", value=kubernetes_creds.get("bearerToken")
+            ),
+        ]
+
+    @staticmethod
+    def invalid_kubernetes_credentials() -> List[CredentialValues]:
+        credentials = CredentialsUtil.kubernetes_credentials()
+        for cred in credentials:
+            if cred.key == "kubernetes_token":
+                cred.value = "wrong_token"
+        return credentials

tests/workflow/assistant_tools/cloud/test_workflow_with_assistant_cloud_tools.py

@@ -12,6 +12,7 @@ from tests.test_data.cloud_tools_test_data import cloud_test_data
         f"{Toolkit.CLOUD}_{CloudTool.AWS}",
         f"{Toolkit.CLOUD}_{CloudTool.AZURE}",
         f"{Toolkit.CLOUD}_{CloudTool.GCP}",
+        f"{Toolkit.CLOUD}_{CloudTool.KUBERNETES}",
     ],
 )
 def test_workflow_with_cloud_tools(

tests/workflow/direct_tools_calling/test_workflow_with_cloud_tools.py

@@ -17,6 +17,7 @@ from tests.utils.base_utils import get_random_name
         f"{Toolkit.CLOUD}_{CloudTool.AWS}",
         f"{Toolkit.CLOUD}_{CloudTool.AZURE}",
         f"{Toolkit.CLOUD}_{CloudTool.GCP}",
+        f"{Toolkit.CLOUD}_{CloudTool.KUBERNETES}",
     ],
 )
 def test_workflow_with_cloud_tools_direct(
@@ -53,6 +54,7 @@ def test_workflow_with_cloud_tools_direct(
         f"{Toolkit.CLOUD}_{CloudTool.AWS}_hardcoded",
         f"{Toolkit.CLOUD}_{CloudTool.AZURE}_hardcoded",
         f"{Toolkit.CLOUD}_{CloudTool.GCP}_hardcoded",
+        f"{Toolkit.CLOUD}_{CloudTool.KUBERNETES}_hardcoded",
     ],
 )
 def test_workflow_with_cloud_tools_with_hardcoded_args(
@@ -87,6 +89,7 @@ def test_workflow_with_cloud_tools_with_hardcoded_args(
         f"{Toolkit.CLOUD}_{CloudTool.AWS}_overriding",
         f"{Toolkit.CLOUD}_{CloudTool.AZURE}_overriding",
         f"{Toolkit.CLOUD}_{CloudTool.GCP}_overriding",
+        f"{Toolkit.CLOUD}_{CloudTool.KUBERNETES}_overriding",
     ],
 )
 def test_workflow_with_cloud_tools_with_overriding_args(

tests/workflow/virtual_assistant_tools/cloud/test_workflow_with_cloud_tools.py

@@ -16,6 +16,7 @@ from tests.utils.base_utils import get_random_name
         f"{Toolkit.CLOUD}_{CloudTool.AWS}",
         f"{Toolkit.CLOUD}_{CloudTool.AZURE}",
         f"{Toolkit.CLOUD}_{CloudTool.GCP}",
+        f"{Toolkit.CLOUD}_{CloudTool.KUBERNETES}",
     ],
 )
 def test_workflow_with_cloud_tools(