pkg/discovery/mock/mock_etcd_client.go (142 lines of code) (raw):

/* * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You 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: test_etcd_client.go // Package mock is a generated GoMock package. package mock import ( context "context" reflect "reflect" gomock "github.com/golang/mock/gomock" clientv3 "go.etcd.io/etcd/client/v3" ) // MockEtcdClient is a mock of EtcdClient interface. type MockEtcdClient struct { ctrl *gomock.Controller recorder *MockEtcdClientMockRecorder } // MockEtcdClientMockRecorder is the mock recorder for MockEtcdClient. type MockEtcdClientMockRecorder struct { mock *MockEtcdClient } // NewMockEtcdClient creates a new mock instance. func NewMockEtcdClient(ctrl *gomock.Controller) *MockEtcdClient { mock := &MockEtcdClient{ctrl: ctrl} mock.recorder = &MockEtcdClientMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockEtcdClient) EXPECT() *MockEtcdClientMockRecorder { return m.recorder } // Close mocks base method. func (m *MockEtcdClient) 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 *MockEtcdClientMockRecorder) Close() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockEtcdClient)(nil).Close)) } // Compact mocks base method. func (m *MockEtcdClient) Compact(ctx context.Context, rev int64, opts ...clientv3.CompactOption) (*clientv3.CompactResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, rev} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Compact", varargs...) ret0, _ := ret[0].(*clientv3.CompactResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // Compact indicates an expected call of Compact. func (mr *MockEtcdClientMockRecorder) Compact(ctx, rev interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, rev}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Compact", reflect.TypeOf((*MockEtcdClient)(nil).Compact), varargs...) } // Delete mocks base method. func (m *MockEtcdClient) Delete(ctx context.Context, key string, opts ...clientv3.OpOption) (*clientv3.DeleteResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, key} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Delete", varargs...) ret0, _ := ret[0].(*clientv3.DeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // Delete indicates an expected call of Delete. func (mr *MockEtcdClientMockRecorder) Delete(ctx, key interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, key}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockEtcdClient)(nil).Delete), varargs...) } // Do mocks base method. func (m *MockEtcdClient) Do(ctx context.Context, op clientv3.Op) (clientv3.OpResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Do", ctx, op) ret0, _ := ret[0].(clientv3.OpResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // Do indicates an expected call of Do. func (mr *MockEtcdClientMockRecorder) Do(ctx, op interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Do", reflect.TypeOf((*MockEtcdClient)(nil).Do), ctx, op) } // Get mocks base method. func (m *MockEtcdClient) Get(ctx context.Context, key string, opts ...clientv3.OpOption) (*clientv3.GetResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, key} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Get", varargs...) ret0, _ := ret[0].(*clientv3.GetResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // Get indicates an expected call of Get. func (mr *MockEtcdClientMockRecorder) Get(ctx, key interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, key}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockEtcdClient)(nil).Get), varargs...) } // Put mocks base method. func (m *MockEtcdClient) Put(ctx context.Context, key, val string, opts ...clientv3.OpOption) (*clientv3.PutResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, key, val} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Put", varargs...) ret0, _ := ret[0].(*clientv3.PutResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // Put indicates an expected call of Put. func (mr *MockEtcdClientMockRecorder) Put(ctx, key, val interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, key, val}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockEtcdClient)(nil).Put), varargs...) } // RequestProgress mocks base method. func (m *MockEtcdClient) RequestProgress(ctx context.Context) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RequestProgress", ctx) ret0, _ := ret[0].(error) return ret0 } // RequestProgress indicates an expected call of RequestProgress. func (mr *MockEtcdClientMockRecorder) RequestProgress(ctx interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestProgress", reflect.TypeOf((*MockEtcdClient)(nil).RequestProgress), ctx) } // Txn mocks base method. func (m *MockEtcdClient) Txn(ctx context.Context) clientv3.Txn { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Txn", ctx) ret0, _ := ret[0].(clientv3.Txn) return ret0 } // Txn indicates an expected call of Txn. func (mr *MockEtcdClientMockRecorder) Txn(ctx interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Txn", reflect.TypeOf((*MockEtcdClient)(nil).Txn), ctx) } // Watch mocks base method. func (m *MockEtcdClient) Watch(ctx context.Context, key string, opts ...clientv3.OpOption) clientv3.WatchChan { m.ctrl.T.Helper() varargs := []interface{}{ctx, key} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Watch", varargs...) ret0, _ := ret[0].(clientv3.WatchChan) return ret0 } // Watch indicates an expected call of Watch. func (mr *MockEtcdClientMockRecorder) Watch(ctx, key interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, key}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Watch", reflect.TypeOf((*MockEtcdClient)(nil).Watch), varargs...) }