Files
ragflow/internal/ingestion/service/test_helpers.go
Jack 8ac80284c4 Feat: add embedding support (#16769)
### Summary

Feat: add embedding support
2026-07-09 16:24:39 +08:00

48 lines
1.6 KiB
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
// ──────────────────────────────────────────────────────────
// 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)
// 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)
}