add one test case

apache · Jun 19, 2021 · bd2ab46 · bd2ab46
1 parent d08243d
commit bd2ab46
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/...etes/src/test/java/org/apache/flink/kubernetes/kubeclient/handlers/PodOOMHandlerTest.java b/...etes/src/test/java/org/apache/flink/kubernetes/kubeclient/handlers/PodOOMHandlerTest.java
@@ -41,6 +41,7 @@ public void testOOMCount() {
                 new PodOOMHandler(
                         UnregisteredMetricGroups.createUnregisteredResourceManagerMetricGroup());
 
+        String finishTime = "2020-03-19T13:14:48Z";
         KubernetesPod kubernetesPod =
                 new KubernetesPod(
                         new PodBuilder()
@@ -60,7 +61,7 @@ public void testOOMCount() {
                                                                             new ContainerStateBuilder()
                                                                                     .withNewTerminated()
                                                                                     .withFinishedAt(
-                                                                                            "2020-10-19T13:14:48Z")
+                                                                                            finishTime)
                                                                                     .withReason(
                                                                                             "OOMKilled")
                                                                                     .endTerminated()
@@ -71,14 +72,13 @@ public void testOOMCount() {
                                 .build());
         podOOMHandler.handle(kubernetesPod);
         Assert.assertEquals(1, podOOMHandler.getTombstones().size());
-        Assert.assertTrue(
-                podOOMHandler
-                        .getTombstones()
-                        .getIfPresent("2020-10-19T13:14:48Z")
-                        .contains("pod-1"));
+        Assert.assertTrue(podOOMHandler.getTombstones().getIfPresent(finishTime).contains("pod-1"));
         // test ignore same events
         podOOMHandler.handle(kubernetesPod);
-        Assert.assertEquals(
-                1, podOOMHandler.getTombstones().getIfPresent("2020-10-19T13:14:48Z").size());
+        Assert.assertEquals(1, podOOMHandler.getTombstones().getIfPresent(finishTime).size());
+
+        kubernetesPod.getInternalResource().getMetadata().setName("pod-2");
+        podOOMHandler.handle(kubernetesPod);
+        Assert.assertEquals(2, podOOMHandler.getTombstones().getIfPresent(finishTime).size());
     }
 }