ragflow/internal/ingestion/service/test_helpers.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 service

import "context"

// ──────────────────────────────────────────────────────────
// Ingestor Test Helpers
// ──────────────────────────────────────────────────────────

// NewTestIngestor creates a new Ingestor for testing with default values.
func NewTestIngestor() *Ingestor {
	return &Ingestor{
		id: "test-ingestor",
	}
}

// IngestorOption configures a test Ingestor.
type IngestorOption func(*Ingestor)

func newUnitIngestor(name string, maxConcurrency int32, supportedTypes []string) *Ingestor {
	ingestor := NewIngestor(name, maxConcurrency, supportedTypes)
	ingestor.checkpointExists = func(context.Context, string) (bool, error) {
		return false, nil
	}
	return ingestor
}

// SetupTestIngestor creates a new test Ingestor with the given options.
func SetupTestIngestor(t testingT, opts ...IngestorOption) *Ingestor {
	t.Helper()
	i := NewTestIngestor()
	for _, opt := range opts {
		opt(i)
	}
	return i
}

// testingT is a subset of testing.TB used by test helpers (no *testing.T import needed).
type testingT interface {
	Helper()
	Fatalf(format string, args ...any)
}