From 839c6646a9760bf626edca9abeb27fc9c68296c2 Mon Sep 17 00:00:00 2001
From: Stuart McCulloch <stuart.mcculloch@datadoghq.com>
Date: Thu, 29 Oct 2026 17:15:28 +0000
Subject: [PATCH] Fix trie builder benchmark corpus loading

---
 .../instrument/utils/ClassNameTrieBuilderBenchmark.java   | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/utils/src/jmh/java/datadog/instrument/utils/ClassNameTrieBuilderBenchmark.java b/utils/src/jmh/java/datadog/instrument/utils/ClassNameTrieBuilderBenchmark.java
index 53fff23..4586f58 100644
--- a/utils/src/jmh/java/datadog/instrument/utils/ClassNameTrieBuilderBenchmark.java
+++ b/utils/src/jmh/java/datadog/instrument/utils/ClassNameTrieBuilderBenchmark.java
@@ -3,9 +3,7 @@
 import static java.util.concurrent.TimeUnit.MILLISECONDS;
 import static org.openjdk.jmh.annotations.Mode.AverageTime;
 
-import java.io.IOException;
-import java.nio.file.Files;
-import java.nio.file.Paths;
+import datadog.instrument.testing.SampleClasses;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
@@ -39,7 +37,7 @@ public class ClassNameTrieBuilderBenchmark {
   private int[] values;
 
   @Setup
-  public void setup() throws IOException {
+  public void setup() {
     List<String> names = new ArrayList<>();
     List<Integer> numbers = new ArrayList<>();
     switch (dataset) {
@@ -51,7 +49,7 @@ public void setup() throws IOException {
             });
         break;
       case "classNames":
-        names.addAll(Files.readAllLines(Paths.get("src/test/resources/sample_class_names.txt")));
+        names.addAll(SampleClasses.loadClassNames());
         break;
       case "longJumps":
         char[] chars = new char[7700];