Fix retrieving source files logic

Author: sarahchen6

components/context/src/main/resources/META-INF/services/org.junit.jupiter.api.extension.Extension

@@ -1 +1 @@
-datadog.context.TestSourceFileExtension
+datadog.context.GatherSourceFileInfoExtension

components/context/src/main/resources/META-INF/services/org.junit.platform.launcher.TestExecutionListener

@@ -1 +1 @@
-datadog.context.SourceFileExtension
+datadog.context.InsertSourceFileExtension

components/context/src/test/java/datadog/context/GatherSourceFileInfoExtension.java

@@ -0,0 +1,31 @@
+package datadog.context;
+
+import java.nio.file.Paths;
+import java.util.HashMap;
+import java.util.Map;
+import org.junit.jupiter.api.extension.ExtensionContext;
+import org.junit.jupiter.api.extension.TestInstancePostProcessor;
+
+public class GatherSourceFileInfoExtension implements TestInstancePostProcessor {
+  private static Map<String, String> sourceFiles = new HashMap<String, String>();
+
+  @Override
+  public void postProcessTestInstance(Object testInstance, ExtensionContext context) {
+    getTestData(context);
+  }
+
+  public static Map<String, String> getSourceFiles() {
+    return sourceFiles;
+  }
+
+  private static void getTestData(ExtensionContext context) {
+    // get test class name and source file
+    String testClassName = context.getTestClass().get().getName();
+    String testClassPath = testClassName.replace(".", "/") + ".java";
+    String absolutePath = Paths.get("").toAbsolutePath() + "/src/test/java/" + testClassPath;
+    String subPath =
+        absolutePath.substring(absolutePath.indexOf("dd-trace-java") + "dd-trace-java".length());
+    // add info to sourceFiles map
+    sourceFiles.put(testClassName, subPath);
+  }
+}

…datadog/context/SourceFileExtension.java …g/context/InsertSourceFileExtension.javacomponents/context/src/test/java/datadog/context/SourceFileExtension.java renamed to components/context/src/test/java/datadog/context/InsertSourceFileExtension.java components/context/src/test/java/datadog/context/SourceFileExtension.java renamed to components/context/src/test/java/datadog/context/InsertSourceFileExtension.java

@@ -13,7 +13,7 @@
 import org.junit.platform.launcher.TestIdentifier;
 import org.junit.platform.launcher.TestPlan;
 
-public class SourceFileExtension implements TestExecutionListener {
+public class InsertSourceFileExtension implements TestExecutionListener {
   public void executionStarted(TestIdentifier testIdentifier) {
     System.out.println("EXECUTIONSTARTED.");
   }
@@ -24,10 +24,9 @@ public void testPlanExecutionStarted(TestPlan testPlan) {
 
   public void executionFinished(
       TestIdentifier testIdentifier, TestExecutionResult testExecutionResult) {
-    // this should happen in testPlanExecutionFinished after all tests are run, but this method is
-    // easier to test locally for now
+    // should this happen in testPlanExecutionFinished after all tests are run?
 
-    Map<String, String> sourceFiles = TestSourceFileExtension.getSourceFiles();
+    Map<String, String> sourceFiles = GatherSourceFileInfoExtension.getSourceFiles();
 
     // for each test
     for (String sourceFile : sourceFiles.keySet()) {
@@ -69,10 +68,13 @@ public void executionFinished(
         matcher.appendTail(result);
 
         // set old filePath to new xml result
-        //        System.out.println("result: " + result);
+        System.out.println("result: " + result.substring(0, 1000));
+
+        // i think this logic gets re-overwritten by xml reports
         BufferedWriter writer = new BufferedWriter(new FileWriter(pathString));
         writer.write(result.toString());
         writer.close();
+
       } catch (Exception e) {
         System.out.println("Modifying XML files did not work.");
       }
@@ -81,8 +83,5 @@ public void executionFinished(
 
   public void testPlanExecutionFinished(TestPlan testPlan) {
     System.out.println("TESTPLANEXECUTIONFINISHED.");
-    // get xml report file
-    // modify report with test source file info
-    // save resulting file
   }
 }