Feat: configure audit fields and metrics for onix adapter and add local configuration for onix adapterZ
This commit is contained in:
@@ -2,9 +2,9 @@ package telemetry
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
@@ -13,16 +13,31 @@ func TestNewProviderAndMetrics(t *testing.T) {
|
||||
provider, err := NewTestProvider(ctx)
|
||||
require.NoError(t, err)
|
||||
require.NotNil(t, provider)
|
||||
require.NotNil(t, provider.MetricsHandler)
|
||||
require.NotNil(t, provider.MeterProvider, "MeterProvider should be set")
|
||||
|
||||
metrics, err := GetMetrics(ctx)
|
||||
require.NoError(t, err)
|
||||
require.NotNil(t, metrics)
|
||||
|
||||
rec := httptest.NewRecorder()
|
||||
req := httptest.NewRequest("GET", "/metrics", nil)
|
||||
provider.MetricsHandler.ServeHTTP(rec, req)
|
||||
require.Equal(t, 200, rec.Code)
|
||||
|
||||
require.NoError(t, provider.Shutdown(context.Background()))
|
||||
require.NoError(t, provider.Shutdown(ctx))
|
||||
}
|
||||
|
||||
func TestNewProviderAndTraces(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
provider, sr, err := NewTestProviderWithTrace(ctx)
|
||||
require.NoError(t, err)
|
||||
require.NotNil(t, provider)
|
||||
require.NotNil(t, provider.MeterProvider, "MeterProvider should be set")
|
||||
require.NotNil(t, provider.TraceProvider, "TraceProvider should be set")
|
||||
require.NotNil(t, sr, "SpanRecorder should be set")
|
||||
|
||||
tracer := provider.TraceProvider.Tracer("test-instrumentation")
|
||||
_, span := tracer.Start(ctx, "test-span")
|
||||
span.End()
|
||||
|
||||
ended := sr.Ended()
|
||||
require.Len(t, ended, 1, "exactly one span should be recorded")
|
||||
assert.Equal(t, "test-span", ended[0].Name(), "recorded span should have expected name")
|
||||
|
||||
require.NoError(t, provider.Shutdown(ctx))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user