ragflow/internal/cli/cli_test.go

//
//  Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

package cli

import (
	"errors"
	"fmt"
	"testing"
)

func TestSanitizeCLIError_StripsSingleQuotedUserInput(t *testing.T) {
	t.Parallel()

	cases := []struct {
		name string
		in   error
		want string
	}{
		{
			name: "dataset name in single quotes",
			in:   fmt.Errorf("dataset '%s' not found", "secret-project-name"),
			want: "dataset not found",
		},
		{
			name: "file path in single quotes",
			in:   fmt.Errorf("file '%s' has bad content", "/home/user/.ssh/id_rsa"),
			want: "file has bad content",
		},
		{
			name: "no quoted content passes through",
			in:   errors.New("connection refused"),
			want: "connection refused",
		},
		{
			name: "nil error returns empty string",
			in:   nil,
			want: "",
		},
		{
			name: "empty string after stripping",
			in:   errors.New("'everything-stripped'"),
			want: "command failed",
		},
		{
			name: "two quoted paths in one error",
			in:   fmt.Errorf("copy '%s' to '%s' failed", "/secret/a", "/secret/b"),
			want: "copy to failed",
		},
		{
			name: "three quoted values mixed with text — only the sensitive spans are stripped",
			in:   fmt.Errorf("'%s' is not a valid %s in %s", "secret-name", "kind", "scope"),
			want: "is not a valid kind in scope",
		},
		{
			name: "unmatched single quote is preserved",
			in:   errors.New("oops 'unterminated"),
			want: "oops 'unterminated",
		},
	}

	for _, tc := range cases {
		tc := tc
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			if got := sanitizeCLIError(tc.in); got != tc.want {
				t.Errorf("sanitizeCLIError(%v) = %q, want %q", tc.in, got, tc.want)
			}
		})
	}
}