Function TestOTelSpanLogger_CustomLogger

spanlogger/otel_spanlogger_test.go:115–158

(t *testing.T)

}

func TestOTelSpanLogger_CustomLogger(t *testing.T) {
	t.Cleanup(spanExporter.Reset)

	var expectedTraceID string
	var logged []map[string]string
	// logger will store all non-"trace_id" keys in `logged`
	// it will check that "trace_id" is equal to expectedTraceID.
	var logger funcLogger = func(keyvals ...interface{}) error {
		values := map[string]string{}
		var loggedTraceID string
		for i := 0; i < len(keyvals); i += 2 {
			k := keyvals[i].(string)
			v := keyvals[i+1].(string) // we only log strings.
			if k == "trace_id" {
				loggedTraceID = v
			} else {
				values[k] = v
			}
		}
		require.Equal(t, expectedTraceID, loggedTraceID)
		logged = append(logged, values)
		return nil
	}
	resolver := tenant.NewMultiResolver()

	span, ctx := NewOTel(context.Background(), logger, tracer, "test", resolver)
	expectedTraceID = trace.SpanFromContext(ctx).SpanContext().TraceID().String()
	_ = span.Log("msg", "original spanlogger")

	span = FromContext(ctx, log.NewNopLogger(), resolver)
	_ = span.Log("msg", "restored spanlogger")

	// No trace_id expected for the next one.
	expectedTraceID = ""
	span = FromContext(context.Background(), logger, resolver)
	_ = span.Log("msg", "fallback spanlogger")

	expect := []map[string]string{
		{"method": "test", "msg": "original spanlogger"},
		{"msg": "restored spanlogger"},
		{"msg": "fallback spanlogger"},
	}
	require.Equal(t, expect, logged)
}

func TestOTelSpanLogger_SetSpanAndLogTag(t *testing.T) {
	t.Cleanup(spanExporter.Reset)