Function TestTokenizeContext_ErrorHandling

pkg/sql/tokenizer/context_test.go:189–228

TestTokenizeContext_ErrorHandling verifies proper error handling

(t *testing.T)

// TestTokenizeContext_ErrorHandling verifies proper error handling
func TestTokenizeContext_ErrorHandling(t *testing.T) {
	tests := []struct {
		name        string
		input       string
		shouldError bool
	}{
		{
			name:        "unterminated string",
			input:       "SELECT 'unterminated",
			shouldError: true,
		},
		{
			name:        "invalid character",
			input:       "SELECT \x00 FROM users",
			shouldError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			ctx := context.Background()
			tkz := GetTokenizer()
			defer PutTokenizer(tkz)

			tokens, err := tkz.TokenizeContext(ctx, []byte(tt.input))

			if tt.shouldError && err == nil {
				t.Error("Expected error but got none")
			}

			if err != nil && err == context.Canceled {
				t.Error("Should not return context.Canceled for SQL errors")
			}

			if tt.shouldError && tokens != nil {
				t.Error("Expected nil tokens on error")
			}
		})
	}
}

// TestTokenizeContext_CancellationResponseTime verifies fast cancellation (< 100ms requirement)
func TestTokenizeContext_CancellationResponseTime(t *testing.T) {