Fix tests

Author: rockdaboot

x-pack/plugin/profiling/src/test/java/org/elasticsearch/xpack/profiling/action/TransportGetFlamegraphActionTests.java

@@ -62,19 +62,21 @@ public void testCreateFlamegraph() {
         assertEquals(1.0d, response.getSamplingRate(), 0.001d);
         assertEquals(List.of(1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L), response.getCountInclusive());
         assertEquals(List.of(0L, 0L, 0L, 0L, 0L, 0L, 0L, 0L, 0L, 0L, 0L, 1L), response.getCountExclusive());
+        // {145245218=3}, {145245218=4}, {145245218=5}, {145245218=6}, {145245218=7}, {145245218=8}, {145245218=9}, {145245218=10},
+        // {145245218=11}
         assertEquals(
             List.of(
                 Map.of("1220", 1),
                 Map.of("1219", 2),
-                Map.of("174640828", 3),
-                Map.of("181190529", 4),
-                Map.of("181192637", 5),
-                Map.of("180652335", 6),
-                Map.of("180479184", 7),
-                Map.of("180475689", 8),
-                Map.of("174846197", 9),
-                Map.of("175515318", 10),
-                Map.of("178196121", 11),
+                Map.of("145245218", 3),
+                Map.of("145245218", 4),
+                Map.of("145245218", 5),
+                Map.of("145245218", 6),
+                Map.of("145245218", 7),
+                Map.of("145245218", 8),
+                Map.of("145245218", 9),
+                Map.of("145245218", 10),
+                Map.of("145245218", 11),
                 Map.of()
             ),
             response.getEdges()

x-pack/plugin/profiling/src/test/java/org/elasticsearch/xpack/profiling/action/TransportGetTopNFunctionsActionTests.java

@@ -60,26 +60,13 @@ public void testCreateAllTopNFunctions() {
         GetTopNFunctionsResponse response = TransportGetTopNFunctionsAction.buildTopNFunctions(stacktraces, null);
         assertNotNull(response);
         assertEquals(1, response.getSelfCount());
-        assertEquals(9, response.getTotalCount());
+        assertEquals(1, response.getTotalCount());
 
         List<TopNFunction> topNFunctions = response.getTopN();
         assertNotNull(topNFunctions);
-        assertEquals(9, topNFunctions.size());
-
-        assertEquals(
-            List.of(
-                topN("178196121", 1, 16339645, 1L, 1L, 0.3d, 0.3d, 2.7d, 2.7d),
-                topN("181192637", 2, 19336161, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("181190529", 3, 19334053, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("180652335", 4, 18795859, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("180479184", 5, 18622708, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("180475689", 6, 18619213, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("175515318", 7, 13658842, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("174846197", 8, 12989721, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("174640828", 9, 12784352, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d)
-            ),
-            topNFunctions
-        );
+        assertEquals(1, topNFunctions.size());
+
+        assertEquals(List.of(topN("145245218", 1, 12784352, 1L, 1L, 0.3d, 0.3d, 2.7d, 2.7d)), topNFunctions);
     }
 
     public void testCreateTopNFunctionsWithLimit() {
@@ -129,20 +116,13 @@ public void testCreateTopNFunctionsWithLimit() {
         GetTopNFunctionsResponse response = TransportGetTopNFunctionsAction.buildTopNFunctions(stacktraces, 3);
         assertNotNull(response);
         assertEquals(1, response.getSelfCount());
-        assertEquals(9, response.getTotalCount());
+        assertEquals(1, response.getTotalCount());
 
         List<TopNFunction> topNFunctions = response.getTopN();
         assertNotNull(topNFunctions);
-        assertEquals(3, topNFunctions.size());
+        assertEquals(1, topNFunctions.size());
 
-        assertEquals(
-            List.of(
-                topN("178196121", 1, 16339645, 1L, 1L, 0.3d, 0.3d, 2.7d, 2.7d),
-                topN("181192637", 2, 19336161, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d),
-                topN("181190529", 3, 19334053, 0L, 1L, 0.0d, 0.3d, 0.0d, 2.7d)
-            ),
-            topNFunctions
-        );
+        assertEquals(List.of(topN("145245218", 1, 12784352, 1L, 1L, 0.3d, 0.3d, 2.7d, 2.7d)), topNFunctions);
     }
 
     private TopNFunction topN(