-
Notifications
You must be signed in to change notification settings - Fork 262
/
client_v1_mock.go
115 lines (101 loc) · 3.62 KB
/
client_v1_mock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// Code generated by MockGen. DO NOT EDIT.
// Source: client_v1.go
//
// Generated by this command:
//
// mockgen -destination mocks/client_v1_mock.go -source client_v1.go -package mocks
//
// Package mocks is a generated GoMock package.
package mocks
import (
context "context"
reflect "reflect"
inference "d7y.io/api/v2/pkg/apis/inference"
gomock "go.uber.org/mock/gomock"
grpc "google.golang.org/grpc"
)
// MockV1 is a mock of V1 interface.
type MockV1 struct {
ctrl *gomock.Controller
recorder *MockV1MockRecorder
}
// MockV1MockRecorder is the mock recorder for MockV1.
type MockV1MockRecorder struct {
mock *MockV1
}
// NewMockV1 creates a new mock instance.
func NewMockV1(ctrl *gomock.Controller) *MockV1 {
mock := &MockV1{ctrl: ctrl}
mock.recorder = &MockV1MockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockV1) EXPECT() *MockV1MockRecorder {
return m.recorder
}
// Close mocks base method.
func (m *MockV1) Close() error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Close")
ret0, _ := ret[0].(error)
return ret0
}
// Close indicates an expected call of Close.
func (mr *MockV1MockRecorder) Close() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockV1)(nil).Close))
}
// ModelInfer mocks base method.
func (m *MockV1) ModelInfer(arg0 context.Context, arg1 *inference.ModelInferRequest, arg2 ...grpc.CallOption) (*inference.ModelInferResponse, error) {
m.ctrl.T.Helper()
varargs := []any{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "ModelInfer", varargs...)
ret0, _ := ret[0].(*inference.ModelInferResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ModelInfer indicates an expected call of ModelInfer.
func (mr *MockV1MockRecorder) ModelInfer(arg0, arg1 any, arg2 ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelInfer", reflect.TypeOf((*MockV1)(nil).ModelInfer), varargs...)
}
// ModelReady mocks base method.
func (m *MockV1) ModelReady(arg0 context.Context, arg1 *inference.ModelReadyRequest, arg2 ...grpc.CallOption) (*inference.ModelReadyResponse, error) {
m.ctrl.T.Helper()
varargs := []any{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "ModelReady", varargs...)
ret0, _ := ret[0].(*inference.ModelReadyResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ModelReady indicates an expected call of ModelReady.
func (mr *MockV1MockRecorder) ModelReady(arg0, arg1 any, arg2 ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelReady", reflect.TypeOf((*MockV1)(nil).ModelReady), varargs...)
}
// ServerReady mocks base method.
func (m *MockV1) ServerReady(arg0 context.Context, arg1 *inference.ServerReadyRequest, arg2 ...grpc.CallOption) (*inference.ServerReadyResponse, error) {
m.ctrl.T.Helper()
varargs := []any{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "ServerReady", varargs...)
ret0, _ := ret[0].(*inference.ServerReadyResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ServerReady indicates an expected call of ServerReady.
func (mr *MockV1MockRecorder) ServerReady(arg0, arg1 any, arg2 ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerReady", reflect.TypeOf((*MockV1)(nil).ServerReady), varargs...)
}