github.com/ydb-platform/ydb-go-sdk/v3@v3.89.2/internal/discovery/grpc_client_mock_test.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/ydb-platform/ydb-go-genproto/Ydb_Discovery_V1 (interfaces: DiscoveryServiceClient)
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -destination grpc_client_mock_test.go --typed -package discovery -write_package_comment=false github.com/ydb-platform/ydb-go-genproto/Ydb_Discovery_V1 DiscoveryServiceClient
     7  package discovery
     8  
     9  import (
    10  	context "context"
    11  	reflect "reflect"
    12  
    13  	Ydb_Discovery "github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Discovery"
    14  	gomock "go.uber.org/mock/gomock"
    15  	grpc "google.golang.org/grpc"
    16  )
    17  
    18  // MockDiscoveryServiceClient is a mock of DiscoveryServiceClient interface.
    19  type MockDiscoveryServiceClient struct {
    20  	ctrl     *gomock.Controller
    21  	recorder *MockDiscoveryServiceClientMockRecorder
    22  }
    23  
    24  // MockDiscoveryServiceClientMockRecorder is the mock recorder for MockDiscoveryServiceClient.
    25  type MockDiscoveryServiceClientMockRecorder struct {
    26  	mock *MockDiscoveryServiceClient
    27  }
    28  
    29  // NewMockDiscoveryServiceClient creates a new mock instance.
    30  func NewMockDiscoveryServiceClient(ctrl *gomock.Controller) *MockDiscoveryServiceClient {
    31  	mock := &MockDiscoveryServiceClient{ctrl: ctrl}
    32  	mock.recorder = &MockDiscoveryServiceClientMockRecorder{mock}
    33  	return mock
    34  }
    35  
    36  // EXPECT returns an object that allows the caller to indicate expected use.
    37  func (m *MockDiscoveryServiceClient) EXPECT() *MockDiscoveryServiceClientMockRecorder {
    38  	return m.recorder
    39  }
    40  
    41  // ListEndpoints mocks base method.
    42  func (m *MockDiscoveryServiceClient) ListEndpoints(arg0 context.Context, arg1 *Ydb_Discovery.ListEndpointsRequest, arg2 ...grpc.CallOption) (*Ydb_Discovery.ListEndpointsResponse, error) {
    43  	m.ctrl.T.Helper()
    44  	varargs := []any{arg0, arg1}
    45  	for _, a := range arg2 {
    46  		varargs = append(varargs, a)
    47  	}
    48  	ret := m.ctrl.Call(m, "ListEndpoints", varargs...)
    49  	ret0, _ := ret[0].(*Ydb_Discovery.ListEndpointsResponse)
    50  	ret1, _ := ret[1].(error)
    51  	return ret0, ret1
    52  }
    53  
    54  // ListEndpoints indicates an expected call of ListEndpoints.
    55  func (mr *MockDiscoveryServiceClientMockRecorder) ListEndpoints(arg0, arg1 any, arg2 ...any) *MockDiscoveryServiceClientListEndpointsCall {
    56  	mr.mock.ctrl.T.Helper()
    57  	varargs := append([]any{arg0, arg1}, arg2...)
    58  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEndpoints", reflect.TypeOf((*MockDiscoveryServiceClient)(nil).ListEndpoints), varargs...)
    59  	return &MockDiscoveryServiceClientListEndpointsCall{Call: call}
    60  }
    61  
    62  // MockDiscoveryServiceClientListEndpointsCall wrap *gomock.Call
    63  type MockDiscoveryServiceClientListEndpointsCall struct {
    64  	*gomock.Call
    65  }
    66  
    67  // Return rewrite *gomock.Call.Return
    68  func (c *MockDiscoveryServiceClientListEndpointsCall) Return(arg0 *Ydb_Discovery.ListEndpointsResponse, arg1 error) *MockDiscoveryServiceClientListEndpointsCall {
    69  	c.Call = c.Call.Return(arg0, arg1)
    70  	return c
    71  }
    72  
    73  // Do rewrite *gomock.Call.Do
    74  func (c *MockDiscoveryServiceClientListEndpointsCall) Do(f func(context.Context, *Ydb_Discovery.ListEndpointsRequest, ...grpc.CallOption) (*Ydb_Discovery.ListEndpointsResponse, error)) *MockDiscoveryServiceClientListEndpointsCall {
    75  	c.Call = c.Call.Do(f)
    76  	return c
    77  }
    78  
    79  // DoAndReturn rewrite *gomock.Call.DoAndReturn
    80  func (c *MockDiscoveryServiceClientListEndpointsCall) DoAndReturn(f func(context.Context, *Ydb_Discovery.ListEndpointsRequest, ...grpc.CallOption) (*Ydb_Discovery.ListEndpointsResponse, error)) *MockDiscoveryServiceClientListEndpointsCall {
    81  	c.Call = c.Call.DoAndReturn(f)
    82  	return c
    83  }
    84  
    85  // WhoAmI mocks base method.
    86  func (m *MockDiscoveryServiceClient) WhoAmI(arg0 context.Context, arg1 *Ydb_Discovery.WhoAmIRequest, arg2 ...grpc.CallOption) (*Ydb_Discovery.WhoAmIResponse, error) {
    87  	m.ctrl.T.Helper()
    88  	varargs := []any{arg0, arg1}
    89  	for _, a := range arg2 {
    90  		varargs = append(varargs, a)
    91  	}
    92  	ret := m.ctrl.Call(m, "WhoAmI", varargs...)
    93  	ret0, _ := ret[0].(*Ydb_Discovery.WhoAmIResponse)
    94  	ret1, _ := ret[1].(error)
    95  	return ret0, ret1
    96  }
    97  
    98  // WhoAmI indicates an expected call of WhoAmI.
    99  func (mr *MockDiscoveryServiceClientMockRecorder) WhoAmI(arg0, arg1 any, arg2 ...any) *MockDiscoveryServiceClientWhoAmICall {
   100  	mr.mock.ctrl.T.Helper()
   101  	varargs := append([]any{arg0, arg1}, arg2...)
   102  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WhoAmI", reflect.TypeOf((*MockDiscoveryServiceClient)(nil).WhoAmI), varargs...)
   103  	return &MockDiscoveryServiceClientWhoAmICall{Call: call}
   104  }
   105  
   106  // MockDiscoveryServiceClientWhoAmICall wrap *gomock.Call
   107  type MockDiscoveryServiceClientWhoAmICall struct {
   108  	*gomock.Call
   109  }
   110  
   111  // Return rewrite *gomock.Call.Return
   112  func (c *MockDiscoveryServiceClientWhoAmICall) Return(arg0 *Ydb_Discovery.WhoAmIResponse, arg1 error) *MockDiscoveryServiceClientWhoAmICall {
   113  	c.Call = c.Call.Return(arg0, arg1)
   114  	return c
   115  }
   116  
   117  // Do rewrite *gomock.Call.Do
   118  func (c *MockDiscoveryServiceClientWhoAmICall) Do(f func(context.Context, *Ydb_Discovery.WhoAmIRequest, ...grpc.CallOption) (*Ydb_Discovery.WhoAmIResponse, error)) *MockDiscoveryServiceClientWhoAmICall {
   119  	c.Call = c.Call.Do(f)
   120  	return c
   121  }
   122  
   123  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   124  func (c *MockDiscoveryServiceClientWhoAmICall) DoAndReturn(f func(context.Context, *Ydb_Discovery.WhoAmIRequest, ...grpc.CallOption) (*Ydb_Discovery.WhoAmIResponse, error)) *MockDiscoveryServiceClientWhoAmICall {
   125  	c.Call = c.Call.DoAndReturn(f)
   126  	return c
   127  }