Function TestHostInfo

modules/generator/processor/hostinfo/processor_test.go:21–57

(t *testing.T)

)

func TestHostInfo(t *testing.T) {
	testRegistry := registry.NewTestRegistry()
	cfg := Config{}

	cfg.RegisterFlagsAndApplyDefaults("", nil)
	p, err := New(cfg, testRegistry, nil, prometheus.NewCounter(prometheus.CounterOpts{}))
	require.NoError(t, err)
	require.Equal(t, p.Name(), processor.HostInfoName)
	defer p.Shutdown(context.TODO())

	req := &tempopb.PushSpansRequest{
		Batches: []*trace_v1.ResourceSpans{
			test.MakeBatch(10, nil),
			test.MakeBatch(10, nil),
		},
	}

	for i, b := range req.Batches {
		b.Resource.Attributes = append(b.Resource.Attributes, []*common_v1.KeyValue{
			{Key: "host.id", Value: &common_v1.AnyValue{Value: &common_v1.AnyValue_StringValue{StringValue: "test" + strconv.Itoa(i)}}},
		}...)
	}

	p.PushSpans(context.Background(), req)

	lbls0 := labels.FromMap(map[string]string{
		hostIdentifierAttr: "test0",
		hostSourceAttr:     "host.id",
	})
	assert.Equal(t, 1.0, testRegistry.Query(hostInfoMetric, lbls0))

	lbls1 := labels.FromMap(map[string]string{
		hostIdentifierAttr: "test1",
		hostSourceAttr:     "host.id",
	})
	assert.Equal(t, 1.0, testRegistry.Query(hostInfoMetric, lbls1))
}

func TestHostInfoHostSource(t *testing.T) {
	testRegistry := registry.NewTestRegistry()