Function TestTracingHook_ProcessHook_LongCommand

extra/redisotel/tracing_test.go:501–576

(t *testing.T)

}

func TestTracingHook_ProcessHook_LongCommand(t *testing.T) {
	imsb := tracetest.NewInMemoryExporter()
	provider := sdktrace.NewTracerProvider(sdktrace.WithSyncer(imsb))
	hook := newTracingHook(
		"redis://localhost:6379",
		WithTracerProvider(provider),
	)
	longValue := strings.Repeat("a", 102400)

	tests := []struct {
		name     string
		cmd      redis.Cmder
		expected string
	}{
		{
			name:     "short command",
			cmd:      redis.NewCmd(context.Background(), "SET", "key", "value"),
			expected: "SET key value",
		},
		{
			name:     "set command with long key",
			cmd:      redis.NewCmd(context.Background(), "SET", longValue, "value"),
			expected: "SET " + longValue + " value",
		},
		{
			name:     "set command with long value",
			cmd:      redis.NewCmd(context.Background(), "SET", "key", longValue),
			expected: "SET key " + longValue,
		},
		{
			name:     "set command with long key and value",
			cmd:      redis.NewCmd(context.Background(), "SET", longValue, longValue),
			expected: "SET " + longValue + " " + longValue,
		},
		{
			name:     "short command with many arguments",
			cmd:      redis.NewCmd(context.Background(), "MSET", "key1", "value1", "key2", "value2", "key3", "value3", "key4", "value4", "key5", "value5"),
			expected: "MSET key1 value1 key2 value2 key3 value3 key4 value4 key5 value5",
		},
		{
			name:     "long command",
			cmd:      redis.NewCmd(context.Background(), longValue, "key", "value"),
			expected: longValue + " key value",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			defer imsb.Reset()

			processHook := hook.ProcessHook(func(ctx context.Context, cmd redis.Cmder) error {
				return nil
			})

			if err := processHook(context.Background(), tt.cmd); err != nil {
				t.Fatal(err)
			}