Add SampleHelpers.CreateScope (#2427)

Lets you create a Datadog Scope in integration tests without having to reference the Datadog.Trace library directly
lachmatt · Feb 13, 2022 · 6989e88 · 6989e88
1 parent cda56dd
commit 6989e88
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tracer/test/test-applications/Samples.Shared/SampleHelpers.cs b/tracer/test/test-applications/Samples.Shared/SampleHelpers.cs
@@ -9,6 +9,9 @@ namespace Samples
     public class SampleHelpers
     {
         private static readonly Type NativeMethodsType = Type.GetType("Datadog.Trace.ClrProfiler.NativeMethods, Datadog.Trace");
+        private static readonly Type TracerType = Type.GetType("Datadog.Trace.Tracer, Datadog.Trace");
+        private static readonly MethodInfo GetTracerInstance = TracerType.GetProperty("Instance").GetMethod;
+        private static readonly MethodInfo StartActiveMethod = TracerType.GetMethod("StartActive", types: new[] { typeof(string) });
 
         public static bool IsProfilerAttached()
         {
@@ -52,6 +55,12 @@ public static void RunShutDownTasks(object caller)
             }
         }
 
+        public static IDisposable CreateScope(string operationName)
+        {
+            var tracer = GetTracerInstance.Invoke(null, Array.Empty<object>());
+            return (IDisposable) StartActiveMethod.Invoke(tracer, new object[] { operationName });
+        }
+
         public static IEnumerable<KeyValuePair<string,string>> GetDatadogEnvironmentVariables()
         {
             var prefixes = new[] { "COR_", "CORECLR_", "DD_", "DATADOG_" };