wehub-resource-sync
117 lines
3.7 KiB
Go
117 lines
3.7 KiB
Go
/*
|
|
* Copyright 2024 CloudWeGo Authors
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: interface.go
|
|
|
|
// Package document is a generated GoMock package.
|
|
package document
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
document "github.com/cloudwego/eino/components/document"
|
|
schema "github.com/cloudwego/eino/schema"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockLoader is a mock of Loader interface.
|
|
type MockLoader struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockLoaderMockRecorder
|
|
}
|
|
|
|
// MockLoaderMockRecorder is the mock recorder for MockLoader.
|
|
type MockLoaderMockRecorder struct {
|
|
mock *MockLoader
|
|
}
|
|
|
|
// NewMockLoader creates a new mock instance.
|
|
func NewMockLoader(ctrl *gomock.Controller) *MockLoader {
|
|
mock := &MockLoader{ctrl: ctrl}
|
|
mock.recorder = &MockLoaderMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockLoader) EXPECT() *MockLoaderMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Load mocks base method.
|
|
func (m *MockLoader) Load(ctx context.Context, src document.Source, opts ...document.LoaderOption) ([]*schema.Document, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []interface{}{ctx, src}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "Load", varargs...)
|
|
ret0, _ := ret[0].([]*schema.Document)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Load indicates an expected call of Load.
|
|
func (mr *MockLoaderMockRecorder) Load(ctx, src interface{}, opts ...interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]interface{}{ctx, src}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Load", reflect.TypeOf((*MockLoader)(nil).Load), varargs...)
|
|
}
|
|
|
|
// MockTransformer is a mock of Transformer interface.
|
|
type MockTransformer struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockTransformerMockRecorder
|
|
}
|
|
|
|
// MockTransformerMockRecorder is the mock recorder for MockTransformer.
|
|
type MockTransformerMockRecorder struct {
|
|
mock *MockTransformer
|
|
}
|
|
|
|
// NewMockTransformer creates a new mock instance.
|
|
func NewMockTransformer(ctrl *gomock.Controller) *MockTransformer {
|
|
mock := &MockTransformer{ctrl: ctrl}
|
|
mock.recorder = &MockTransformerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockTransformer) EXPECT() *MockTransformerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Transform mocks base method.
|
|
func (m *MockTransformer) Transform(ctx context.Context, src []*schema.Document, opts ...document.TransformerOption) ([]*schema.Document, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []interface{}{ctx, src}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "Transform", varargs...)
|
|
ret0, _ := ret[0].([]*schema.Document)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Transform indicates an expected call of Transform.
|
|
func (mr *MockTransformerMockRecorder) Transform(ctx, src interface{}, opts ...interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]interface{}{ctx, src}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Transform", reflect.TypeOf((*MockTransformer)(nil).Transform), varargs...)
|
|
}
|