cmd/ragflow_server_test.go
Rama main · réplica de lectura del repositorio original.
//
// 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 main
import (
"context"
"os"
"os/exec"
"path/filepath"
"ragflow/internal/server/bootstrap"
"strings"
"testing"
"time"
)
func parseArgsForTest(t *testing.T, argv ...string) (*bootstrap.ServerArgs, error) {
t.Helper()
orig := os.Args
os.Args = append([]string{"ragflow_server"}, argv...)
defer func() { os.Args = orig }()
return parseArgs()
}
func TestParseArgsMigrateSelectsMode(t *testing.T) {
args, err := parseArgsForTest(t, "--migrate")
if err != nil {
t.Fatalf("parseArgs(--migrate) error = %v", err)
}
if args.Mode == nil || *args.Mode != "migrate" {
t.Fatalf("mode = %v, want migrate", args.Mode)
}
}
func TestParseArgsLogLevel(t *testing.T) {
for _, level := range []string{"debug", "info", "warn", "error"} {
for _, argv := range [][]string{{"--api", "--log-level", level}, {"--migrate", "--log-level=" + level}} {
args, err := parseArgsForTest(t, argv...)
if err != nil {
t.Fatalf("parseArgs(%v) error = %v", argv, err)
}
if got := bootstrap.SelectedLogLevel(args, "info"); got != level {
t.Errorf("selectedLogLevel(%v) = %q, want %q", argv, got, level)
}
}
}
for _, argv := range [][]string{{"--api", "--log-level"}, {"--api", "--log-level=trace"}, {"--api", "--log-level", "fatal"}} {
if _, err := parseArgsForTest(t, argv...); err == nil {
t.Errorf("parseArgs(%v) error = nil, want error", argv)
}
}
}
func TestSelectedLogLevelPrecedence(t *testing.T) {
args, err := parseArgsForTest(t, "--api")
if err != nil {
t.Fatalf("parseArgs(--api) error = %v", err)
}
if got := bootstrap.SelectedLogLevel(args, ""); got != "info" {
t.Errorf("default log level = %q, want info", got)
}
if got := bootstrap.SelectedLogLevel(args, "info"); got != "info" {
t.Errorf("configured log level = %q, want info", got)
}
args, err = parseArgsForTest(t, "--api", "--log-level", "error")
if err != nil {
t.Fatalf("parseArgs with --log-level error = %v", err)
}
if got := bootstrap.SelectedLogLevel(args, "info"); got != "error" {
t.Errorf("--log-level log level = %q, want error", got)
}
if _, err = parseArgsForTest(t, "--api", "--debug"); err == nil {
t.Error("parseArgs(--api --debug) error = nil, want rejection")
} else if !strings.Contains(err.Error(), "flag provided but not defined") {
// The stdlib flag package owns unknown-flag errors now; the old
// "unknown parameter" string no longer applies.
t.Errorf("parseArgs(--api --debug) error = %q, want \"flag provided but not defined\"", err)
}
}
// Mode flags share a single slot, so the last one passed wins. Putting
// --migrate next to a server mode is not a runtime conflict, so it must not
// be rejected.
func TestParseArgsMigrateFollowsLastModeWins(t *testing.T) {
for _, mode := range []string{"--api", "--admin", "--ingestor", "--syncer"} {
want := strings.TrimPrefix(mode, "--")
args, err := parseArgsForTest(t, mode, "--migrate")
if err != nil {
t.Fatalf("parseArgs(%s --migrate) error = %v", mode, err)
}
if args.Mode == nil || *args.Mode != "migrate" {
t.Errorf("parseArgs(%s --migrate) mode = %v, want migrate", mode, args.Mode)
}
args, err = parseArgsForTest(t, "--migrate", mode)
if err != nil {
t.Fatalf("parseArgs(--migrate %s) error = %v", mode, err)
}
if args.Mode == nil || *args.Mode != want {
t.Errorf("parseArgs(--migrate %s) mode = %v, want %s", mode, args.Mode, want)
}
}
}
func TestParseArgsKeepsMCPOutOfCLI(t *testing.T) {
t.Setenv("RAGFLOW_MCP_PORT", "invalid")
for _, mode := range []string{"--api", "--admin", "--ingestor", "--syncer", "--migrate", "--help", "--version"} {
if _, err := parseArgsForTest(t, mode); err != nil {
t.Fatalf("%s depends on MCP environment: %v", mode, err)
}
}
for _, flag := range []string{"--enable-mcpserver", "--mcp-host=localhost", "--mcp-port=9382", "--mcp-mode=host", "--mcp-host-api-key=unused", "--transport-sse-enabled", "--no-transport-sse-enabled", "--transport-streamable-http-enabled", "--no-transport-streamable-http-enabled", "--json-response", "--no-json-response"} {
if _, err := parseArgsForTest(t, "--api", flag); err == nil {
t.Fatalf("obsolete MCP flag accepted: %s", flag)
}
}
}
func TestMainMigrateReachesConfiguration(t *testing.T) {
if os.Getenv("RAGFLOW_TEST_MIGRATE_MAIN") == "1" {
os.Args = append([]string{os.Args[0]}, os.Args[len(os.Args)-3:]...)
main()
return
}
ctx, cancel := context.WithTimeout(t.Context(), 15*time.Second)
defer cancel()
dir := t.TempDir()
cmd := exec.CommandContext(ctx, os.Args[0], "-test.run=^TestMainMigrateReachesConfiguration$", "--", "--migrate", "--config", filepath.Join(dir, "missing.yaml"))
cmd.Dir = dir
cmd.Env = append(os.Environ(), "RAGFLOW_TEST_MIGRATE_MAIN=1")
output, err := cmd.CombinedOutput()
if err == nil || !strings.Contains(string(output), "initialize configuration") || strings.Contains(string(output), "Usage:") {
t.Fatalf("migration did not reach config: %v\n%s", err, output)
}
}
// TestPrintHelpDeepdoc guards the DeepDoc help text. It must describe the
// DeepDoc server (not the Sync service) and show the real --name default
// (deepdoc_server). The hand-rolled help text previously copied the syncer
// case verbatim, printing "Sync service server name (default: "default_syncer")"
// under --deepdoc.
func TestPrintHelpDeepdoc(t *testing.T) {
if os.Getenv("RAGFLOW_TEST_DEEPDOC_HELP") == "1" {
os.Args = append([]string{os.Args[0]}, os.Args[len(os.Args)-2:]...)
main()
return
}
ctx, cancel := context.WithTimeout(t.Context(), 15*time.Second)
defer cancel()
dir := t.TempDir()
cmd := exec.CommandContext(ctx, os.Args[0], "-test.run=^TestPrintHelpDeepdoc$", "--", "--deepdoc", "--help")
cmd.Dir = dir
cmd.Env = append(os.Environ(), "RAGFLOW_TEST_DEEPDOC_HELP=1")
output, err := cmd.CombinedOutput()
if err == nil {
t.Fatalf("deepdoc --help did not exit non-zero: %s", output)
}
if !strings.Contains(string(output), "DeepDoc server name") {
t.Errorf("deepdoc help missing the DeepDoc server name line:\n%s", output)
}
if !strings.Contains(string(output), `default: "deepdoc_server"`) {
t.Errorf("deepdoc help missing the correct --name default (want deepdoc_server):\n%s", output)
}
if strings.Contains(string(output), "Sync service server name") {
t.Errorf("deepdoc help still shows the syncer's text:\n%s", output)
}
}