Function TestOpenTelemetry

pubsub/pubsub_test.go:575–627

TestOpenTelemetry tests that OpenTelemetry tracing is working correctly.

(t *testing.T)

// TestOpenTelemetry tests that OpenTelemetry tracing is working correctly.
func TestOpenTelemetry(t *testing.T) {
	ctx := context.Background()

	te := oteltest.NewTestExporter(t, OpenTelemetryViews)
	defer func() {
		err := te.Shutdown(ctx)
		if err != nil {
			t.Logf("Error shutting down test exporter: %v", err)
		}
	}()

	ds := NewDriverSub()
	dt := &driverTopic{
		subs: []*driverSub{ds},
	}
	topic := NewTopic(dt, nil)
	defer topic.Shutdown(ctx)
	sub := NewSubscription(ds, nil, nil)
	defer sub.Shutdown(ctx)
	if err := topic.Send(ctx, &Message{Body: []byte("x")}); err != nil {
		t.Fatal(err)
	}
	if err := topic.Shutdown(ctx); err != nil {
		t.Fatal(err)
	}
	msg, err := sub.Receive(ctx)
	if err != nil {
		t.Fatal(err)
	}
	msg.Ack()
	err = sub.Shutdown(ctx)
	if err != nil {
		t.Fatal(err)
	}
	_, _ = sub.Receive(ctx)

	spans := te.GetSpans()
	metrics := te.GetMetrics(ctx)

	diff := oteltest.Diff(spans.Snapshots(), metrics, pkgName, "gocloud.dev/pubsub", []oteltest.Call{
		{Method: "driver.Topic.SendBatch", Code: gcerrors.OK},
		{Method: "Topic.Send", Code: gcerrors.OK},
		{Method: "Topic.Shutdown", Code: gcerrors.OK},
		{Method: "driver.Subscription.ReceiveBatch", Code: gcerrors.OK},
		{Method: "Subscription.Receive", Code: gcerrors.OK},
		{Method: "driver.Subscription.SendAcks", Code: gcerrors.OK},
		{Method: "Subscription.Shutdown", Code: gcerrors.OK},
		{Method: "Subscription.Receive", Code: gcerrors.FailedPrecondition},
	})
	if diff != "" {
		t.Error(diff)
	}
}

func TestURLMux(t *testing.T) {
	ctx := context.Background()