/
serving_mock.go
77 lines (66 loc) · 3.07 KB
/
serving_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
// Code generated by MockGen. DO NOT EDIT.
// Source: github.com/feast-dev/feast/sdk/go/protos/feast/serving (interfaces: ServingServiceClient)
// Package mock_serving is a generated GoMock package.
package mock_serving
import (
context "context"
reflect "reflect"
serving "github.com/feast-dev/feast/sdk/go/protos/feast/serving"
gomock "github.com/golang/mock/gomock"
grpc "google.golang.org/grpc"
)
// MockServingServiceClient is a mock of ServingServiceClient interface
type MockServingServiceClient struct {
ctrl *gomock.Controller
recorder *MockServingServiceClientMockRecorder
}
// MockServingServiceClientMockRecorder is the mock recorder for MockServingServiceClient
type MockServingServiceClientMockRecorder struct {
mock *MockServingServiceClient
}
// NewMockServingServiceClient creates a new mock instance
func NewMockServingServiceClient(ctrl *gomock.Controller) *MockServingServiceClient {
mock := &MockServingServiceClient{ctrl: ctrl}
mock.recorder = &MockServingServiceClientMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockServingServiceClient) EXPECT() *MockServingServiceClientMockRecorder {
return m.recorder
}
// GetFeastServingInfo mocks base method
func (m *MockServingServiceClient) GetFeastServingInfo(arg0 context.Context, arg1 *serving.GetFeastServingInfoRequest, arg2 ...grpc.CallOption) (*serving.GetFeastServingInfoResponse, error) {
m.ctrl.T.Helper()
varargs := []interface{}{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "GetFeastServingInfo", varargs...)
ret0, _ := ret[0].(*serving.GetFeastServingInfoResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetFeastServingInfo indicates an expected call of GetFeastServingInfo
func (mr *MockServingServiceClientMockRecorder) GetFeastServingInfo(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]interface{}{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeastServingInfo", reflect.TypeOf((*MockServingServiceClient)(nil).GetFeastServingInfo), varargs...)
}
// GetOnlineFeaturesV2 mocks base method
func (m *MockServingServiceClient) GetOnlineFeaturesV2(arg0 context.Context, arg1 *serving.GetOnlineFeaturesRequestV2, arg2 ...grpc.CallOption) (*serving.GetOnlineFeaturesResponse, error) {
m.ctrl.T.Helper()
varargs := []interface{}{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "GetOnlineFeaturesV2", varargs...)
ret0, _ := ret[0].(*serving.GetOnlineFeaturesResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetOnlineFeaturesV2 indicates an expected call of GetOnlineFeaturesV2
func (mr *MockServingServiceClientMockRecorder) GetOnlineFeaturesV2(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]interface{}{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOnlineFeaturesV2", reflect.TypeOf((*MockServingServiceClient)(nil).GetOnlineFeaturesV2), varargs...)
}