Added unit tests for execution/deployment counts

SAP · Sep 16, 2024 · 46298ac · 46298ac
1 parent cf8458f
commit 46298ac
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/core/src/test/java/com/sap/ai/sdk/core/client/DeploymentUnitTest.java b/core/src/test/java/com/sap/ai/sdk/core/client/DeploymentUnitTest.java
@@ -331,4 +331,25 @@ void patchDeploymentConfiguration() {
                           }
                           """)));
   }
+
+  @Test
+  void getDeploymentCount() {
+    wireMockServer.stubFor(
+        get(urlPathEqualTo("/lm/deployments/$count"))
+            .withHeader("AI-Resource-Group", equalTo("default"))
+            .willReturn(
+                aResponse()
+                    .withStatus(HttpStatus.SC_OK)
+                    .withHeader("content-type", "application/json")
+                    .withBody(
+                        """
+                        1
+                        """)));
+
+    final int count = new DeploymentApi(getClient(destination)).deploymentCount("default");
+
+    assertThat(count).isEqualTo(1);
+  }
+
+
 }
diff --git a/core/src/test/java/com/sap/ai/sdk/core/client/ExecutionUnitTest.java b/core/src/test/java/com/sap/ai/sdk/core/client/ExecutionUnitTest.java
@@ -290,4 +290,23 @@ void patchExecution() {
             .withHeader("AI-Resource-Group", equalTo("default"))
             .withRequestBody(equalToJson("{\"targetStatus\":\"STOPPED\"}")));
   }
+
+  @Test
+  void getExecutionCount() {
+    wireMockServer.stubFor(
+        get(urlPathEqualTo("/lm/executions/$count"))
+            .withHeader("AI-Resource-Group", equalTo("default"))
+            .willReturn(
+                aResponse()
+                    .withStatus(HttpStatus.SC_OK)
+                    .withHeader("content-type", "application/json")
+                    .withBody(
+                        """
+                        1
+                        """)));
+
+    final int count = new ExecutionApi(getClient(destination)).executionCount("default");
+
+    assertThat(count).isEqualTo(1);
+  }
 }