github.com/lingyao2333/mo-zero@v1.4.1/core/discov/internal/statewatcher_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: statewatcher.go
     3  
     4  // Package internal is a generated GoMock package.
     5  package internal
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	connectivity "google.golang.org/grpc/connectivity"
    13  )
    14  
    15  // MocketcdConn is a mock of etcdConn interface
    16  type MocketcdConn struct {
    17  	ctrl     *gomock.Controller
    18  	recorder *MocketcdConnMockRecorder
    19  }
    20  
    21  // MocketcdConnMockRecorder is the mock recorder for MocketcdConn
    22  type MocketcdConnMockRecorder struct {
    23  	mock *MocketcdConn
    24  }
    25  
    26  // NewMocketcdConn creates a new mock instance
    27  func NewMocketcdConn(ctrl *gomock.Controller) *MocketcdConn {
    28  	mock := &MocketcdConn{ctrl: ctrl}
    29  	mock.recorder = &MocketcdConnMockRecorder{mock}
    30  	return mock
    31  }
    32  
    33  // EXPECT returns an object that allows the caller to indicate expected use
    34  func (m *MocketcdConn) EXPECT() *MocketcdConnMockRecorder {
    35  	return m.recorder
    36  }
    37  
    38  // GetState mocks base method
    39  func (m *MocketcdConn) GetState() connectivity.State {
    40  	m.ctrl.T.Helper()
    41  	ret := m.ctrl.Call(m, "GetState")
    42  	ret0, _ := ret[0].(connectivity.State)
    43  	return ret0
    44  }
    45  
    46  // GetState indicates an expected call of GetState
    47  func (mr *MocketcdConnMockRecorder) GetState() *gomock.Call {
    48  	mr.mock.ctrl.T.Helper()
    49  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetState", reflect.TypeOf((*MocketcdConn)(nil).GetState))
    50  }
    51  
    52  // WaitForStateChange mocks base method
    53  func (m *MocketcdConn) WaitForStateChange(ctx context.Context, sourceState connectivity.State) bool {
    54  	m.ctrl.T.Helper()
    55  	ret := m.ctrl.Call(m, "WaitForStateChange", ctx, sourceState)
    56  	ret0, _ := ret[0].(bool)
    57  	return ret0
    58  }
    59  
    60  // WaitForStateChange indicates an expected call of WaitForStateChange
    61  func (mr *MocketcdConnMockRecorder) WaitForStateChange(ctx, sourceState interface{}) *gomock.Call {
    62  	mr.mock.ctrl.T.Helper()
    63  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WaitForStateChange", reflect.TypeOf((*MocketcdConn)(nil).WaitForStateChange), ctx, sourceState)
    64  }