Update tests

Max Hniebergall · Max Hniebergall · commit 5d430eb0e676 · 2025-01-09T09:50:15.000-05:00
diff --git a/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceCrudIT.java b/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceCrudIT.java
@@ -58,7 +58,7 @@ public void testCRUD() throws IOException {
         }
 
         var getAllModels = getAllModels();
-        int numModels = 12;
+        int numModels = 13;
         assertThat(getAllModels, hasSize(numModels));
 
         var getSparseModels = getModels("_all", TaskType.SPARSE_EMBEDDING);
@@ -543,8 +543,8 @@ private static String expectedResult(String input) {
         }
     }
 
-    public void testGetZeroModels() throws IOException {
+    public void testGetCompletionModels() throws IOException {
         var models = getModels("_all", TaskType.COMPLETION);
-        assertThat(models, empty());
+        assertEquals(models.size(), 1);
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ public void testCRUD() throws IOException {`
`58`	`58`	`}`
`59`	`59`
`60`	`60`	`var getAllModels = getAllModels();`
`61`		`- int numModels = 12;`
	`61`	`+ int numModels = 13;`
`62`	`62`	`assertThat(getAllModels, hasSize(numModels));`
`63`	`63`
`64`	`64`	`var getSparseModels = getModels("_all", TaskType.SPARSE_EMBEDDING);`
`@@ -543,8 +543,8 @@ private static String expectedResult(String input) {`
`543`	`543`	`}`
`544`	`544`	`}`
`545`	`545`
`546`		`- public void testGetZeroModels() throws IOException {`
	`546`	`+ public void testGetCompletionModels() throws IOException {`
`547`	`547`	`var models = getModels("_all", TaskType.COMPLETION);`
`548`		`- assertThat(models, empty());`
	`548`	`+ assertEquals(models.size(), 1);`
`549`	`549`	`}`
`550`	`550`	`}`