Function TestMovedAndAskErrorAddressExtraction

internal/proto/redis_errors_test.go:266–326

TestMovedAndAskErrorAddressExtraction tests address extraction from MOVED/ASK errors

(t *testing.T)

// TestMovedAndAskErrorAddressExtraction tests address extraction from MOVED/ASK errors
func TestMovedAndAskErrorAddressExtraction(t *testing.T) {
	tests := []struct {
		name         string
		errorMsg     string
		expectedAddr string
	}{
		{
			name:         "MOVED with IP address",
			errorMsg:     "MOVED 3999 127.0.0.1:6381",
			expectedAddr: "127.0.0.1:6381",
		},
		{
			name:         "MOVED with hostname",
			errorMsg:     "MOVED 3999 redis-node-1:6379",
			expectedAddr: "redis-node-1:6379",
		},
		{
			name:         "ASK with IP address",
			errorMsg:     "ASK 3999 192.168.1.100:6380",
			expectedAddr: "192.168.1.100:6380",
		},
		{
			name:         "ASK with hostname",
			errorMsg:     "ASK 3999 redis-node-2:6379",
			expectedAddr: "redis-node-2:6379",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := parseTypedRedisError(tt.errorMsg)

			var addr string
			if movedErr, ok := IsMovedError(err); ok {
				addr = movedErr.Addr()
			} else if askErr, ok := IsAskError(err); ok {
				addr = askErr.Addr()
			} else {
				t.Fatalf("Error is neither MOVED nor ASK: %v", err)
			}

			if addr != tt.expectedAddr {
				t.Errorf("Address mismatch: got %q, want %q", addr, tt.expectedAddr)
			}

			// Test with wrapped error
			wrappedErr := fmt.Errorf("wrapped: %w", err)
			if movedErr, ok := IsMovedError(wrappedErr); ok {
				addr = movedErr.Addr()
			} else if askErr, ok := IsAskError(wrappedErr); ok {
				addr = askErr.Addr()
			} else {
				t.Fatalf("Wrapped error is neither MOVED nor ASK: %v", wrappedErr)
			}

			if addr != tt.expectedAddr {
				t.Errorf("Address mismatch in wrapped error: got %q, want %q", addr, tt.expectedAddr)
			}