Unit tests

Signed-off-by: Ruben Vargas <[email protected]>

Author: rubenvp8510

pkg/crdmetrics/collector_test.go

@@ -0,0 +1,239 @@
+package crdmetrics
+
+import (
+	"context"
+	"github.com/open-telemetry/opentelemetry-operator/apis/v1beta1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/sdk/metric"
+	"go.opentelemetry.io/otel/sdk/metric/metricdata"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/client-go/kubernetes/scheme"
+	"sigs.k8s.io/controller-runtime/pkg/client/fake"
+
+	"github.com/open-telemetry/opentelemetry-operator/apis/v1alpha1"
+)
+
+type expectedMetric struct {
+	name   string
+	labels []attribute.KeyValue
+	value  int64
+}
+
+func assertLabelAndValues(t *testing.T, name string, metrics metricdata.ResourceMetrics, expectedAttrs []attribute.KeyValue, expectedValue int64) {
+	var matchingMetric metricdata.Metrics
+	found := false
+	for _, sm := range metrics.ScopeMetrics {
+		for _, m := range sm.Metrics {
+			if m.Name == name {
+				matchingMetric = m
+				found = true
+				break
+			}
+		}
+	}
+
+	assert.True(t, found, "Metric %s not found", name)
+
+	gauge, ok := matchingMetric.Data.(metricdata.Gauge[int64])
+	assert.True(t, ok,
+		"Metric %s doesn't have expected type %T, got %T", metricdata.Gauge[int64]{}, matchingMetric.Data,
+	)
+
+	expectedAttrSet := attribute.NewSet(expectedAttrs...)
+	var matchingDP metricdata.DataPoint[int64]
+	found = false
+	for _, dp := range gauge.DataPoints {
+		if expectedAttrSet.Equals(&dp.Attributes) {
+			matchingDP = dp
+			found = true
+			break
+		}
+	}
+
+	assert.True(t, found, "Metric %s doesn't have expected attributes %v", expectedAttrs)
+	assert.Equal(t, expectedValue, matchingDP.Value,
+		"Metric %s doesn't have expected value %d, got %d", name, expectedValue, matchingDP.Value)
+}
+
+func newExpectedMetric(name string, keyPair attribute.KeyValue, value int64) expectedMetric {
+	return expectedMetric{
+		name: name,
+		labels: []attribute.KeyValue{
+			keyPair,
+		},
+		value: value,
+	}
+}
+
+func TestValueObservedMetrics(t *testing.T) {
+	s := scheme.Scheme
+
+	// Add jaeger to schema
+	s.AddKnownTypes(v1alpha1.GroupVersion,
+		&v1alpha1.OpenTelemetryCollector{},
+		&v1alpha1.OpenTelemetryCollectorList{},
+		&v1beta1.OpenTelemetryCollector{},
+		&v1beta1.OpenTelemetryCollectorList{},
+	)
+
+	collector1 := v1alpha1.OpenTelemetryCollector{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "collector1",
+			Namespace: "otel",
+		},
+		Spec: v1alpha1.OpenTelemetryCollectorSpec{
+
+			Mode: v1alpha1.ModeSidecar,
+			Config: `
+				receivers:
+				  otlp:
+					protocols:
+					  grpc:
+					  http:
+				processors:
+				  memory_limiter:
+					check_interval: 1s
+					limit_percentage: 75
+					spike_limit_percentage: 15
+				  batch:
+					send_batch_size: 10000
+					timeout: 10s
+			
+				exporters:
+				  debug:
+			
+				service:
+				  pipelines:
+					traces:
+					  receivers: [otlp]
+					  processors: []
+					  exporters: [debug]
+			`,
+		},
+	}
+
+	collector2 := v1alpha1.OpenTelemetryCollector{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "collector1",
+			Namespace: "otel",
+		},
+		Spec: v1alpha1.OpenTelemetryCollectorSpec{
+
+			Mode: v1alpha1.ModeSidecar,
+			Config: `
+				receivers:
+				  otlp:
+					protocols:
+					  grpc:
+					  http:
+				processors:
+				  memory_limiter:
+					check_interval: 1s
+					limit_percentage: 75
+					spike_limit_percentage: 15
+				  batch:
+					send_batch_size: 10000
+					timeout: 10s
+			
+				exporters:
+				  debug:
+			
+				service:
+				  pipelines:
+					traces:
+					  receivers: [otlp]
+					  processors: []
+					  exporters: [debug]
+			`,
+		},
+	}
+
+	collector3 := v1beta1.OpenTelemetryCollector{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "collector1",
+			Namespace: "otel",
+		},
+		Spec: v1beta1.OpenTelemetryCollectorSpec{
+
+			Mode: v1beta1.ModeSidecar,
+			Config: v1beta1.Config{
+				Receivers: v1beta1.AnyConfig{map[string]interface{}{
+					""
+				}},
+			},
+		},
+	}
+
+	objs := []runtime.Object{
+		&tempoGCS,
+		&tempoS3,
+		&tempoOtherS3,
+		&tempoUnmanaged,
+		&tempoUnmanaged2,
+		&tempoTenantOpenshift,
+		&tempoTenantStatic,
+	}
+	expected := []expectedMetric{
+		newExpectedMetric(managedMetric, attribute.String("state", string(v1alpha1.ManagementStateManaged)), 5),
+		newExpectedMetric(managedMetric, attribute.String("state", string(v1alpha1.ManagementStateUnmanaged)), 2),
+		newExpectedMetric(storageBackendMetric, attribute.String("type", string(v1alpha1.ObjectStorageSecretGCS)), 1),
+		newExpectedMetric(storageBackendMetric, attribute.String("type", string(v1alpha1.ObjectStorageSecretS3)), 4),
+		newExpectedMetric(storageBackendMetric, attribute.String("type", string(v1alpha1.ObjectStorageSecretAzure)), 2),
+		newExpectedMetric(multitenancy, attribute.String("type", "disabled"), 5),
+		newExpectedMetric(multitenancy, attribute.String("type", string(v1alpha1.ModeOpenShift)), 1),
+		newExpectedMetric(multitenancy, attribute.String("type", string(v1alpha1.ModeStatic)), 1),
+		newExpectedMetric(jaegerUIUsage, attribute.String("enabled", "true"), 1),
+		newExpectedMetric(jaegerUIUsage, attribute.String("enabled", "false"), 6),
+	}
+
+	cl := fake.NewClientBuilder().WithScheme(s).WithRuntimeObjects(objs...).Build()
+
+	reader := metric.NewManualReader()
+	provider := metric.NewMeterProvider(metric.WithReader(reader))
+	otel.SetMeterProvider(provider)
+
+	instancesObservedValue := newTempoStackMetrics(cl)
+	err := instancesObservedValue.Setup()
+	require.NoError(t, err)
+
+	metrics := metricdata.ResourceMetrics{}
+	err = reader.Collect(context.Background(), &metrics)
+	require.NoError(t, err)
+
+	for _, e := range expected {
+		assertLabelAndValues(t, e.name, metrics, e.labels, e.value)
+	}
+
+	// Test deleting GCS storage
+	err = cl.Delete(context.Background(), &tempoGCS)
+	require.NoError(t, err)
+
+	// Reset measurement batches
+	err = provider.ForceFlush(context.Background())
+	require.NoError(t, err)
+	metrics = metricdata.ResourceMetrics{}
+	err = reader.Collect(context.Background(), &metrics)
+	require.NoError(t, err)
+
+	// Set new numbers
+	expected = []expectedMetric{
+		newExpectedMetric(managedMetric, attribute.String("state", string(v1alpha1.ManagementStateManaged)), 4),
+		newExpectedMetric(managedMetric, attribute.String("state", string(v1alpha1.ManagementStateUnmanaged)), 2),
+		newExpectedMetric(storageBackendMetric, attribute.String("type", string(v1alpha1.ObjectStorageSecretGCS)), 0),
+		newExpectedMetric(storageBackendMetric, attribute.String("type", string(v1alpha1.ObjectStorageSecretS3)), 4),
+		newExpectedMetric(storageBackendMetric, attribute.String("type", string(v1alpha1.ObjectStorageSecretAzure)), 2),
+		newExpectedMetric(multitenancy, attribute.String("type", string(v1alpha1.ModeOpenShift)), 1),
+		newExpectedMetric(multitenancy, attribute.String("type", string(v1alpha1.ModeStatic)), 1),
+		newExpectedMetric(jaegerUIUsage, attribute.String("enabled", "true"), 0),
+		newExpectedMetric(jaegerUIUsage, attribute.String("enabled", "false"), 6),
+	}
+	for _, e := range expected {
+		assertLabelAndValues(t, e.name, metrics, e.labels, e.value)
+	}
+}