Another fix to job status tests to accomodate new types.

k8ssandra · Sep 26, 2023 · c0e1b61 · c0e1b61
1 parent b6a042d
commit c0e1b61
Showing 1 changed file with 11 additions and 19 deletions.
diff --git a/management-api-server/src/test/java/com/datastax/mgmtapi/NonDestructiveOpsIT.java b/management-api-server/src/test/java/com/datastax/mgmtapi/NonDestructiveOpsIT.java
@@ -794,16 +794,14 @@ public void testAsyncRepair() throws IOException, URISyntaxException, Interrupte
                 return;
               }
               assertThat(getJobDetailsResponse.getLeft()).isEqualTo(HttpStatus.SC_OK);
-              Map<String, String> jobDetails =
+              Job jobDetails =
                   new JsonMapper()
                       .readValue(
                           getJobDetailsResponse.getRight(),
-                          new TypeReference<Map<String, String>>() {});
-              assertThat(jobDetails)
-                  .hasEntrySatisfying("id", value -> assertThat(value).isEqualTo(jobId))
-                  .hasEntrySatisfying("type", value -> assertThat(value).isEqualTo("repair"))
-                  .hasEntrySatisfying(
-                      "status", value -> assertThat(value).isIn("COMPLETED", "ERROR"));
+                          new TypeReference<Job>() {});
+              assertThat(jobDetails.getJobId()).isEqualTo(jobId);
+              assertThat(jobDetails.getJobType()).isEqualTo("repair");
+              assertThat(jobDetails.getStatus()).isIn("COMPLETED", "ERROR");
             });
   }
 
@@ -1019,18 +1017,12 @@ public void testMoveNode() throws IOException, URISyntaxException {
                       .thenApply(this::responseAsCodeAndBody)
                       .join();
               assertThat(getJobDetailsResponse.getLeft()).isEqualTo(HttpStatus.SC_OK);
-              Map<String, String> jobDetails =
+              Job jobDetails =
                   new JsonMapper()
-                      .readValue(
-                          getJobDetailsResponse.getRight(),
-                          new TypeReference<Map<String, String>>() {});
-              assertThat(jobDetails)
-                  .hasEntrySatisfying("id", value -> assertThat(value).isEqualTo(jobId))
-                  .hasEntrySatisfying("type", value -> assertThat(value).isEqualTo("move"))
-                  // if the server has only one token, the job will be completed, otherwise it will
-                  // end up in error
-                  .hasEntrySatisfying(
-                      "status", value -> assertThat(value).isIn("COMPLETED", "ERROR"));
+                      .readValue(getJobDetailsResponse.getRight(), new TypeReference<Job>() {});
+              assertThat(jobDetails.getJobId()).isEqualTo(jobId);
+              assertThat(jobDetails.getJobType()).isEqualTo("move");
+              assertThat(jobDetails.getStatus()).isIn("COMPLETED", "ERROR");
             });
   }
-}
+}