tests: open telemetry

* add test suite; PASS

Signed-off-by: Alex Aizman <[email protected]>

tracing/tracing_on.go

@@ -132,6 +132,7 @@ func Shutdown() {
 	if err := tp.Shutdown(context.Background()); err != nil {
 		cos.ExitLog(err)
 	}
+	tp = nil
 }
 
 func NewTraceableHandler(handler http.Handler, operation string) http.Handler {

tracing/unit_suite_test.go

@@ -0,0 +1,19 @@
+// Package test provides tests for common low-level types and utilities for all aistore projects
+/*
+ * Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
+ */
+package tracing_test
+
+import (
+	"testing"
+
+	"github.com/NVIDIA/aistore/tools"
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+)
+
+func TestTracing(t *testing.T) {
+	tools.CheckSkip(t, &tools.SkipTestArgs{Long: true})
+	RegisterFailHandler(Fail)
+	RunSpecs(t, t.Name())
+}

tracing/unit_test.go

@@ -6,6 +6,9 @@
  */
 package tracing_test
 
+// usage:
+// go test -v -tags="debug oteltracing"
+
 import (
 	"io"
 	"net/http"
@@ -25,8 +28,6 @@ var _ = Describe("Tracing", func() {
 	const aisVersion = "v3.33"
 
 	var (
-		exporter *tracetest.InMemoryExporter
-
 		dummySnode = &meta.Snode{DaeID: "test", DaeType: "proxy"}
 
 		newTestHandler = http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
@@ -59,11 +60,12 @@ var _ = Describe("Tracing", func() {
 			tracing.Shutdown()
 		})
 		It("should export server trace when tracing enabled", func() {
+			exporter := tracetest.NewInMemoryExporter()
 			tracing.Init(&cmn.TracingConf{
 				ExporterEndpoint:   "dummy",
 				Enabled:            true,
 				SamplerProbability: 1.0,
-			}, dummySnode, tracetest.NewInMemoryExporter(), aisVersion)
+			}, dummySnode, exporter, aisVersion)
 			Expect(tracing.IsEnabled()).To(BeTrue())
 
 			server := httptest.NewServer(tracing.NewTraceableHandler(newTestHandler, "testendpoint"))
@@ -80,9 +82,10 @@ var _ = Describe("Tracing", func() {
 		})
 
 		It("should do nothing when tracing disabled", func() {
+			exporter := tracetest.NewInMemoryExporter()
 			tracing.Init(&cmn.TracingConf{
 				Enabled: false,
-			}, dummySnode, tracetest.NewInMemoryExporter(), aisVersion)
+			}, dummySnode, exporter, aisVersion)
 			Expect(tracing.IsEnabled()).To(BeFalse())
 
 			server := httptest.NewServer(tracing.NewTraceableHandler(newTestHandler, "testendpoint"))
@@ -100,11 +103,12 @@ var _ = Describe("Tracing", func() {
 			tracing.Shutdown()
 		})
 		It("should export client trace when tracing enabled", func() {
+			exporter := tracetest.NewInMemoryExporter()
 			tracing.Init(&cmn.TracingConf{
 				ExporterEndpoint:   "dummy",
 				Enabled:            true,
 				SamplerProbability: 1.0,
-			}, dummySnode, tracetest.NewInMemoryExporter(), aisVersion)
+			}, dummySnode, exporter, aisVersion)
 			Expect(tracing.IsEnabled()).To(BeTrue())
 
 			server := httptest.NewServer(newTestHandler)
@@ -128,9 +132,10 @@ var _ = Describe("Tracing", func() {
 		})
 
 		It("should do nothing when tracing disabled", func() {
+			exporter := tracetest.NewInMemoryExporter()
 			tracing.Init(&cmn.TracingConf{
 				Enabled: false,
-			}, dummySnode, tracetest.NewInMemoryExporter(), aisVersion)
+			}, dummySnode, exporter, aisVersion)
 			Expect(tracing.IsEnabled()).To(BeFalse())
 
 			server := httptest.NewServer(newTestHandler)