mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-10-13 03:21:28 +08:00
ab4a8dfbea
### Summary A config package based on https://github.com/open-telemetry/opentelemetry-collector/blob/main/confmap/confmap.go for signoz. #### Related Issues / PR's This is a part of https://github.com/SigNoz/signoz/pull/5710
67 lines
2.5 KiB
Go
67 lines
2.5 KiB
Go
package inmemoryreceiver
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
"go.opentelemetry.io/collector/component/componenttest"
|
|
"go.opentelemetry.io/collector/confmap"
|
|
"go.opentelemetry.io/collector/consumer/consumertest"
|
|
"go.opentelemetry.io/collector/receiver"
|
|
)
|
|
|
|
func TestReceiverLifecycle(t *testing.T) {
|
|
require := require.New(t)
|
|
testReceiverId := uuid.NewString()
|
|
|
|
// Should be able to get a hold of the receiver after starting it.
|
|
require.Nil(GetReceiverInstance(testReceiverId), "receiver instance should not exist before Start()")
|
|
|
|
constructed, err := makeTestLogReceiver(testReceiverId)
|
|
require.Nil(err, "could not make test receiver")
|
|
|
|
err = constructed.Start(context.Background(), componenttest.NewNopHost())
|
|
require.Nil(err, "could not start test receiver")
|
|
|
|
testReceiver := GetReceiverInstance(testReceiverId)
|
|
require.NotNil(testReceiver, "could not get receiver instance by Id")
|
|
|
|
// Should not be able to start 2 receivers with the same id
|
|
constructed2, err := makeTestLogReceiver(testReceiverId)
|
|
require.Nil(err, "could not create second receiver with same id")
|
|
|
|
err = constructed2.Start(context.Background(), componenttest.NewNopHost())
|
|
require.NotNil(err, "should not be able to start another receiver with same id before shutting down the previous one")
|
|
|
|
// Should not be able to get a hold of an receiver after shutdown
|
|
testReceiver.Shutdown(context.Background())
|
|
require.Nil(GetReceiverInstance(testReceiverId), "should not be able to find inmemory receiver after shutdown")
|
|
|
|
// Should be able to start a new receiver with same id after shutting down
|
|
constructed3, err := makeTestLogReceiver(testReceiverId)
|
|
require.Nil(err, "could not make receiver with same Id after shutting down old one")
|
|
|
|
err = constructed3.Start(context.Background(), componenttest.NewNopHost())
|
|
require.Nil(err, "should be able to start another receiver with same id after shutting down the previous one")
|
|
|
|
testReceiver3 := GetReceiverInstance(testReceiverId)
|
|
require.NotNil(testReceiver3, "could not get receiver instance by Id")
|
|
|
|
testReceiver3.Shutdown(context.Background())
|
|
require.Nil(GetReceiverInstance(testReceiverId))
|
|
}
|
|
|
|
func makeTestLogReceiver(receiverId string) (receiver.Logs, error) {
|
|
factory := NewFactory()
|
|
|
|
cfg := factory.CreateDefaultConfig()
|
|
|
|
confmap.NewFromStringMap(map[string]any{"id": receiverId}).Unmarshal(&cfg)
|
|
|
|
return factory.CreateLogsReceiver(
|
|
context.Background(), receiver.CreateSettings{}, cfg, consumertest.NewNop(),
|
|
)
|
|
}
|