otel-collector/opamp/simple_client_test.go

package opamp

import (
	"context"
	"testing"

	"github.com/SigNoz/signoz-otel-collector/signozcol"
	"go.opentelemetry.io/collector/otelcol"
	"go.uber.org/zap"
)

func TestNopClientWithCollector(t *testing.T) {
	coll := signozcol.New(signozcol.WrappedCollectorSettings{
		ConfigPaths: []string{"testdata/simple/config.yaml"},
		Version:     "0.0.1",
		Desc:        "test",
		LoggingOpts: []zap.Option{zap.AddStacktrace(zap.ErrorLevel)},
	})

	client := NewSimpleClient(coll)

	err := client.Start(context.Background())
	if err != nil {
		t.Errorf("unexpected error: %v", err)
	}

	if coll.GetState() != otelcol.StateRunning {
		t.Errorf("expected collector to be run")
	}

	err = client.Stop(context.Background())
	if err != nil {
		t.Errorf("unexpected error: %v", err)
	}
}

func TestNopClientWithCollectorError(t *testing.T) {
	coll := signozcol.New(signozcol.WrappedCollectorSettings{
		ConfigPaths: []string{"testdata/invalid.yaml"},
		Version:     "0.0.1",
		Desc:        "test",
		LoggingOpts: []zap.Option{zap.AddStacktrace(zap.ErrorLevel)},
	})

	client := NewSimpleClient(coll)

	err := client.Start(context.Background())
	if err == nil {
		t.Errorf("expected error")
	}

	if coll.GetState() != otelcol.StateClosed {
		t.Errorf("expected collector to be in closed state")
	}

	err = client.Stop(context.Background())
	if err == nil {
		t.Errorf("expected error")
	}
}

func TestNopClientWithCollectorErrorRead(t *testing.T) {
	coll := signozcol.New(signozcol.WrappedCollectorSettings{
		ConfigPaths: []string{"testdata/invalid.yaml"},
		Version:     "0.0.1",
		Desc:        "test",
		LoggingOpts: []zap.Option{zap.AddStacktrace(zap.ErrorLevel)},
	})

	client := NewSimpleClient(coll)

	err := client.Start(context.Background())
	if err == nil {
		t.Errorf("expected error")
	}

	if coll.GetState() != otelcol.StateClosed {
		t.Errorf("expected collector to be in closed state")
	}

	err = client.Error()
	if err == nil {
		t.Errorf("expected error")
	}
}

func TestNopClientWithCollectorErrorChanMultipleTime(t *testing.T) {
	coll := signozcol.New(signozcol.WrappedCollectorSettings{
		ConfigPaths: []string{"testdata/invalid.yaml"},
		Version:     "0.0.1",
		Desc:        "test",
		LoggingOpts: []zap.Option{zap.AddStacktrace(zap.ErrorLevel)},
	})

	client := NewSimpleClient(coll)

	err := client.Start(context.Background())
	if err == nil {
		t.Errorf("expected error")
	}

	if coll.GetState() != otelcol.StateClosed {
		t.Errorf("expected collector to be in closed state")
	}

	err = client.Error()
	if err == nil {
		t.Errorf("expected error")
	}

	err = client.Error()
	if err != nil {
		t.Errorf("expected no error")
	}
}