zgene
6bed393c12
Backend Tests / backend-unit-test (push) Has been cancelled
Backend Tests / benchmark-test (push) Has been cancelled
CI@main / Node.js v22 (ubuntu-latest) (push) Has been cancelled
Thrift Syntax Validation / validate-thrift (push) Has been cancelled
License Check / License Check (push) Has been cancelled
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: nl2sql.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -destination ../../../internal/mock/infra/nl2sql_mock/nl2sql_mock.go -package mock -source nl2sql.go Factory
|
|
//
|
|
|
|
// Package mock is a generated GoMock package.
|
|
package mock
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
schema "github.com/cloudwego/eino/schema"
|
|
document "github.com/coze-dev/coze-studio/backend/infra/document"
|
|
nl2sql "github.com/coze-dev/coze-studio/backend/infra/document/nl2sql"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockNL2SQL is a mock of NL2SQL interface.
|
|
type MockNL2SQL struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockNL2SQLMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockNL2SQLMockRecorder is the mock recorder for MockNL2SQL.
|
|
type MockNL2SQLMockRecorder struct {
|
|
mock *MockNL2SQL
|
|
}
|
|
|
|
// NewMockNL2SQL creates a new mock instance.
|
|
func NewMockNL2SQL(ctrl *gomock.Controller) *MockNL2SQL {
|
|
mock := &MockNL2SQL{ctrl: ctrl}
|
|
mock.recorder = &MockNL2SQLMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockNL2SQL) EXPECT() *MockNL2SQLMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// NL2SQL mocks base method.
|
|
func (m *MockNL2SQL) NL2SQL(ctx context.Context, messages []*schema.Message, tables []*document.TableSchema, opts ...nl2sql.Option) (string, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, messages, tables}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "NL2SQL", varargs...)
|
|
ret0, _ := ret[0].(string)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// NL2SQL indicates an expected call of NL2SQL.
|
|
func (mr *MockNL2SQLMockRecorder) NL2SQL(ctx, messages, tables any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, messages, tables}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NL2SQL", reflect.TypeOf((*MockNL2SQL)(nil).NL2SQL), varargs...)
|
|
}
|