Add OTel to tests

ionide · Apr 15, 2024 · f83ff26 · f83ff26
1 parent b846ac3
commit f83ff26
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/test/FsAutoComplete.Tests.Lsp/Program.fs b/test/FsAutoComplete.Tests.Lsp/Program.fs
@@ -121,9 +121,29 @@ let tests = testList "FSAC" [
     SnapshotTests.snapshotTests loaders toolsPath
   ]
 
+open OpenTelemetry
+open OpenTelemetry.Resources
+open OpenTelemetry.Trace
+open OpenTelemetry.Logs
+open OpenTelemetry.Metrics
+open System.Diagnostics
+open FsAutoComplete.Telemetry
 
 [<EntryPoint>]
 let main args =
+  let serviceName = "FsAutoComplete.Tests.Lsp"
+  use traceProvider =
+    let version = FsAutoComplete.Utils.Version.info().Version
+    Sdk
+      .CreateTracerProviderBuilder()
+      .AddSource(FsAutoComplete.Utils.Tracing.serviceName, Tracing.fscServiceName, serviceName)
+      .SetResourceBuilder(
+        ResourceBuilder
+          .CreateDefault()
+          .AddService(serviceName = serviceName, serviceVersion = version)
+      )
+      .AddOtlpExporter()
+      .Build()
   let outputTemplate =
     "[{Timestamp:HH:mm:ss} {Level:u3}] [{SourceContext}] {Message:lj}{NewLine}{Exception}"
 
@@ -225,12 +245,16 @@ let main args =
   let fixedUpArgs = args |> Array.except argsToRemove
 
   let cts = new CancellationTokenSource(testTimeout)
+  use activitySource = new ActivitySource(serviceName)
 
   let cliArgs =
     [
       CLIArguments.Printer(Expecto.Impl.TestPrinters.summaryWithLocationPrinter defaultConfig.printer)
       CLIArguments.Verbosity Expecto.Logging.LogLevel.Info
       CLIArguments.Parallel
+      CLIArguments.ActivitySource activitySource
     ]
-
+  // let trace = traceProvider.GetTracer("FsAutoComplete.Tests.Lsp")
+  // use span =  trace.StartActiveSpan("runTests", SpanKind.Internal)
+  use span = activitySource.StartActivity("runTests")
   runTestsWithCLIArgsAndCancel cts.Token cliArgs fixedUpArgs tests
diff --git a/test/FsAutoComplete.Tests.Lsp/paket.references b/test/FsAutoComplete.Tests.Lsp/paket.references
@@ -13,6 +13,7 @@ Serilog
 Destructurama.FSharp
 Serilog.Sinks.Async
 Serilog.Sinks.Console
+OpenTelemetry.Exporter.OpenTelemetryProtocol
 
 Microsoft.Build copy_local: false
 Microsoft.Build.Framework copy_local: false