Try avoiding flakyness in ObjectSampleDumpSmokeTest

DataDog · Feb 21, 2025 · 4595318 · 4595318
1 parent e1b03ad
commit 4595318
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/ddprof-test/src/test/java/com/datadoghq/profiler/jfr/JfrDumpTest.java b/ddprof-test/src/test/java/com/datadoghq/profiler/jfr/JfrDumpTest.java
@@ -16,10 +16,14 @@ public void runTest(String eventName) throws Exception {
     }
 
     public void runTest(String eventName, String ... patterns) throws Exception {
+        runTest(eventName, 10, patterns);
+    }
+
+    public void runTest(String eventName, int dumpCnt, String ... patterns) throws Exception {
         Assumptions.assumeTrue(Platform.isJavaVersionAtLeast(11));
         Assumptions.assumeFalse(Platform.isJ9());
 
-        for (int j = 0; j < 10; j++) {
+        for (int j = 0; j < dumpCnt; j++) {
             Path recording = Files.createTempFile("dump-", ".jfr");
             try {
                 for (int i = 0; i < 50; i++) {

diff --git a/ddprof-test/src/test/java/com/datadoghq/profiler/jfr/ObjectSampleDumpSmokeTest.java b/ddprof-test/src/test/java/com/datadoghq/profiler/jfr/ObjectSampleDumpSmokeTest.java
@@ -21,6 +21,6 @@ protected String getProfilerCommand() {
     @RetryingTest(5)
     @Timeout(value = 300)
     public void test() throws Exception {
-        runTest("datadog.ObjectSample", "method3");
+        runTest("datadog.ObjectSample", 3, "method3");
     }
 }