update unit test

star-whale · Sep 23, 2022 · 08fa1a1 · 08fa1a1
1 parent 0d3954a
commit 08fa1a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/controller/src/test/java/ai/starwhale/mlops/api/DataStoreControllerTest.java b/server/controller/src/test/java/ai/starwhale/mlops/api/DataStoreControllerTest.java
@@ -641,6 +641,7 @@ public void testNullTableName() {
         @Test
         public void testTableNotExists() {
             this.req.setTableName("table not exists");
+            this.req.setIgnoreNonExistingTable(false);
             assertThrows(SwValidationException.class,
                     () -> DataStoreControllerTest.this.controller.queryTable(this.req),
                     "");
@@ -649,7 +650,6 @@ public void testTableNotExists() {
         @Test
         public void testTableNotExistsIgnore() {
             this.req.setTableName("table not exists");
-            this.req.setIgnoreNonExistingTable(true);
 
             var resp = DataStoreControllerTest.this.controller.queryTable(this.req);
             assertThat("test", resp.getStatusCode().is2xxSuccessful(), is(true));
@@ -1216,6 +1216,7 @@ public void testNullTableName() {
         @Test
         public void testInvalidTableName() {
             this.req.getTables().get(0).setTableName("invalid");
+            this.req.setIgnoreNonExistingTable(false);
             assertThrows(SwValidationException.class,
                     () -> DataStoreControllerTest.this.controller.scanTable(this.req),
                     "");
@@ -1224,7 +1225,6 @@ public void testInvalidTableName() {
         @Test
         public void testTableNotExistsIgnore() {
             this.req.getTables().get(0).setTableName("invalid");
-            this.req.setIgnoreNonExistingTable(true);
 
             var resp = DataStoreControllerTest.this.controller.scanTable(this.req);
             assertThat("test", resp.getStatusCode().is2xxSuccessful(), is(true));