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

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/ydb-platform/ydb-go-genproto/Ydb_Query_V1 (interfaces: QueryServiceClient,QueryService_AttachSessionClient,QueryService_ExecuteQueryClient)
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -destination grpc_client_mock_test.go --typed -package query -write_package_comment=false github.com/ydb-platform/ydb-go-genproto/Ydb_Query_V1 QueryServiceClient,QueryService_AttachSessionClient,QueryService_ExecuteQueryClient
     7  package query
     8  
     9  import (
    10  	context "context"
    11  	reflect "reflect"
    12  
    13  	Ydb_Query_V1 "github.com/ydb-platform/ydb-go-genproto/Ydb_Query_V1"
    14  	Ydb_Operations "github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Operations"
    15  	Ydb_Query "github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Query"
    16  	gomock "go.uber.org/mock/gomock"
    17  	grpc "google.golang.org/grpc"
    18  	metadata "google.golang.org/grpc/metadata"
    19  )
    20  
    21  // MockQueryServiceClient is a mock of QueryServiceClient interface.
    22  type MockQueryServiceClient struct {
    23  	ctrl     *gomock.Controller
    24  	recorder *MockQueryServiceClientMockRecorder
    25  }
    26  
    27  // MockQueryServiceClientMockRecorder is the mock recorder for MockQueryServiceClient.
    28  type MockQueryServiceClientMockRecorder struct {
    29  	mock *MockQueryServiceClient
    30  }
    31  
    32  // NewMockQueryServiceClient creates a new mock instance.
    33  func NewMockQueryServiceClient(ctrl *gomock.Controller) *MockQueryServiceClient {
    34  	mock := &MockQueryServiceClient{ctrl: ctrl}
    35  	mock.recorder = &MockQueryServiceClientMockRecorder{mock}
    36  	return mock
    37  }
    38  
    39  // EXPECT returns an object that allows the caller to indicate expected use.
    40  func (m *MockQueryServiceClient) EXPECT() *MockQueryServiceClientMockRecorder {
    41  	return m.recorder
    42  }
    43  
    44  // AttachSession mocks base method.
    45  func (m *MockQueryServiceClient) AttachSession(arg0 context.Context, arg1 *Ydb_Query.AttachSessionRequest, arg2 ...grpc.CallOption) (Ydb_Query_V1.QueryService_AttachSessionClient, error) {
    46  	m.ctrl.T.Helper()
    47  	varargs := []any{arg0, arg1}
    48  	for _, a := range arg2 {
    49  		varargs = append(varargs, a)
    50  	}
    51  	ret := m.ctrl.Call(m, "AttachSession", varargs...)
    52  	ret0, _ := ret[0].(Ydb_Query_V1.QueryService_AttachSessionClient)
    53  	ret1, _ := ret[1].(error)
    54  	return ret0, ret1
    55  }
    56  
    57  // AttachSession indicates an expected call of AttachSession.
    58  func (mr *MockQueryServiceClientMockRecorder) AttachSession(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientAttachSessionCall {
    59  	mr.mock.ctrl.T.Helper()
    60  	varargs := append([]any{arg0, arg1}, arg2...)
    61  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AttachSession", reflect.TypeOf((*MockQueryServiceClient)(nil).AttachSession), varargs...)
    62  	return &MockQueryServiceClientAttachSessionCall{Call: call}
    63  }
    64  
    65  // MockQueryServiceClientAttachSessionCall wrap *gomock.Call
    66  type MockQueryServiceClientAttachSessionCall struct {
    67  	*gomock.Call
    68  }
    69  
    70  // Return rewrite *gomock.Call.Return
    71  func (c *MockQueryServiceClientAttachSessionCall) Return(arg0 Ydb_Query_V1.QueryService_AttachSessionClient, arg1 error) *MockQueryServiceClientAttachSessionCall {
    72  	c.Call = c.Call.Return(arg0, arg1)
    73  	return c
    74  }
    75  
    76  // Do rewrite *gomock.Call.Do
    77  func (c *MockQueryServiceClientAttachSessionCall) Do(f func(context.Context, *Ydb_Query.AttachSessionRequest, ...grpc.CallOption) (Ydb_Query_V1.QueryService_AttachSessionClient, error)) *MockQueryServiceClientAttachSessionCall {
    78  	c.Call = c.Call.Do(f)
    79  	return c
    80  }
    81  
    82  // DoAndReturn rewrite *gomock.Call.DoAndReturn
    83  func (c *MockQueryServiceClientAttachSessionCall) DoAndReturn(f func(context.Context, *Ydb_Query.AttachSessionRequest, ...grpc.CallOption) (Ydb_Query_V1.QueryService_AttachSessionClient, error)) *MockQueryServiceClientAttachSessionCall {
    84  	c.Call = c.Call.DoAndReturn(f)
    85  	return c
    86  }
    87  
    88  // BeginTransaction mocks base method.
    89  func (m *MockQueryServiceClient) BeginTransaction(arg0 context.Context, arg1 *Ydb_Query.BeginTransactionRequest, arg2 ...grpc.CallOption) (*Ydb_Query.BeginTransactionResponse, error) {
    90  	m.ctrl.T.Helper()
    91  	varargs := []any{arg0, arg1}
    92  	for _, a := range arg2 {
    93  		varargs = append(varargs, a)
    94  	}
    95  	ret := m.ctrl.Call(m, "BeginTransaction", varargs...)
    96  	ret0, _ := ret[0].(*Ydb_Query.BeginTransactionResponse)
    97  	ret1, _ := ret[1].(error)
    98  	return ret0, ret1
    99  }
   100  
   101  // BeginTransaction indicates an expected call of BeginTransaction.
   102  func (mr *MockQueryServiceClientMockRecorder) BeginTransaction(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientBeginTransactionCall {
   103  	mr.mock.ctrl.T.Helper()
   104  	varargs := append([]any{arg0, arg1}, arg2...)
   105  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BeginTransaction", reflect.TypeOf((*MockQueryServiceClient)(nil).BeginTransaction), varargs...)
   106  	return &MockQueryServiceClientBeginTransactionCall{Call: call}
   107  }
   108  
   109  // MockQueryServiceClientBeginTransactionCall wrap *gomock.Call
   110  type MockQueryServiceClientBeginTransactionCall struct {
   111  	*gomock.Call
   112  }
   113  
   114  // Return rewrite *gomock.Call.Return
   115  func (c *MockQueryServiceClientBeginTransactionCall) Return(arg0 *Ydb_Query.BeginTransactionResponse, arg1 error) *MockQueryServiceClientBeginTransactionCall {
   116  	c.Call = c.Call.Return(arg0, arg1)
   117  	return c
   118  }
   119  
   120  // Do rewrite *gomock.Call.Do
   121  func (c *MockQueryServiceClientBeginTransactionCall) Do(f func(context.Context, *Ydb_Query.BeginTransactionRequest, ...grpc.CallOption) (*Ydb_Query.BeginTransactionResponse, error)) *MockQueryServiceClientBeginTransactionCall {
   122  	c.Call = c.Call.Do(f)
   123  	return c
   124  }
   125  
   126  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   127  func (c *MockQueryServiceClientBeginTransactionCall) DoAndReturn(f func(context.Context, *Ydb_Query.BeginTransactionRequest, ...grpc.CallOption) (*Ydb_Query.BeginTransactionResponse, error)) *MockQueryServiceClientBeginTransactionCall {
   128  	c.Call = c.Call.DoAndReturn(f)
   129  	return c
   130  }
   131  
   132  // CommitTransaction mocks base method.
   133  func (m *MockQueryServiceClient) CommitTransaction(arg0 context.Context, arg1 *Ydb_Query.CommitTransactionRequest, arg2 ...grpc.CallOption) (*Ydb_Query.CommitTransactionResponse, error) {
   134  	m.ctrl.T.Helper()
   135  	varargs := []any{arg0, arg1}
   136  	for _, a := range arg2 {
   137  		varargs = append(varargs, a)
   138  	}
   139  	ret := m.ctrl.Call(m, "CommitTransaction", varargs...)
   140  	ret0, _ := ret[0].(*Ydb_Query.CommitTransactionResponse)
   141  	ret1, _ := ret[1].(error)
   142  	return ret0, ret1
   143  }
   144  
   145  // CommitTransaction indicates an expected call of CommitTransaction.
   146  func (mr *MockQueryServiceClientMockRecorder) CommitTransaction(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientCommitTransactionCall {
   147  	mr.mock.ctrl.T.Helper()
   148  	varargs := append([]any{arg0, arg1}, arg2...)
   149  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommitTransaction", reflect.TypeOf((*MockQueryServiceClient)(nil).CommitTransaction), varargs...)
   150  	return &MockQueryServiceClientCommitTransactionCall{Call: call}
   151  }
   152  
   153  // MockQueryServiceClientCommitTransactionCall wrap *gomock.Call
   154  type MockQueryServiceClientCommitTransactionCall struct {
   155  	*gomock.Call
   156  }
   157  
   158  // Return rewrite *gomock.Call.Return
   159  func (c *MockQueryServiceClientCommitTransactionCall) Return(arg0 *Ydb_Query.CommitTransactionResponse, arg1 error) *MockQueryServiceClientCommitTransactionCall {
   160  	c.Call = c.Call.Return(arg0, arg1)
   161  	return c
   162  }
   163  
   164  // Do rewrite *gomock.Call.Do
   165  func (c *MockQueryServiceClientCommitTransactionCall) Do(f func(context.Context, *Ydb_Query.CommitTransactionRequest, ...grpc.CallOption) (*Ydb_Query.CommitTransactionResponse, error)) *MockQueryServiceClientCommitTransactionCall {
   166  	c.Call = c.Call.Do(f)
   167  	return c
   168  }
   169  
   170  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   171  func (c *MockQueryServiceClientCommitTransactionCall) DoAndReturn(f func(context.Context, *Ydb_Query.CommitTransactionRequest, ...grpc.CallOption) (*Ydb_Query.CommitTransactionResponse, error)) *MockQueryServiceClientCommitTransactionCall {
   172  	c.Call = c.Call.DoAndReturn(f)
   173  	return c
   174  }
   175  
   176  // CreateSession mocks base method.
   177  func (m *MockQueryServiceClient) CreateSession(arg0 context.Context, arg1 *Ydb_Query.CreateSessionRequest, arg2 ...grpc.CallOption) (*Ydb_Query.CreateSessionResponse, error) {
   178  	m.ctrl.T.Helper()
   179  	varargs := []any{arg0, arg1}
   180  	for _, a := range arg2 {
   181  		varargs = append(varargs, a)
   182  	}
   183  	ret := m.ctrl.Call(m, "CreateSession", varargs...)
   184  	ret0, _ := ret[0].(*Ydb_Query.CreateSessionResponse)
   185  	ret1, _ := ret[1].(error)
   186  	return ret0, ret1
   187  }
   188  
   189  // CreateSession indicates an expected call of CreateSession.
   190  func (mr *MockQueryServiceClientMockRecorder) CreateSession(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientCreateSessionCall {
   191  	mr.mock.ctrl.T.Helper()
   192  	varargs := append([]any{arg0, arg1}, arg2...)
   193  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSession", reflect.TypeOf((*MockQueryServiceClient)(nil).CreateSession), varargs...)
   194  	return &MockQueryServiceClientCreateSessionCall{Call: call}
   195  }
   196  
   197  // MockQueryServiceClientCreateSessionCall wrap *gomock.Call
   198  type MockQueryServiceClientCreateSessionCall struct {
   199  	*gomock.Call
   200  }
   201  
   202  // Return rewrite *gomock.Call.Return
   203  func (c *MockQueryServiceClientCreateSessionCall) Return(arg0 *Ydb_Query.CreateSessionResponse, arg1 error) *MockQueryServiceClientCreateSessionCall {
   204  	c.Call = c.Call.Return(arg0, arg1)
   205  	return c
   206  }
   207  
   208  // Do rewrite *gomock.Call.Do
   209  func (c *MockQueryServiceClientCreateSessionCall) Do(f func(context.Context, *Ydb_Query.CreateSessionRequest, ...grpc.CallOption) (*Ydb_Query.CreateSessionResponse, error)) *MockQueryServiceClientCreateSessionCall {
   210  	c.Call = c.Call.Do(f)
   211  	return c
   212  }
   213  
   214  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   215  func (c *MockQueryServiceClientCreateSessionCall) DoAndReturn(f func(context.Context, *Ydb_Query.CreateSessionRequest, ...grpc.CallOption) (*Ydb_Query.CreateSessionResponse, error)) *MockQueryServiceClientCreateSessionCall {
   216  	c.Call = c.Call.DoAndReturn(f)
   217  	return c
   218  }
   219  
   220  // DeleteSession mocks base method.
   221  func (m *MockQueryServiceClient) DeleteSession(arg0 context.Context, arg1 *Ydb_Query.DeleteSessionRequest, arg2 ...grpc.CallOption) (*Ydb_Query.DeleteSessionResponse, error) {
   222  	m.ctrl.T.Helper()
   223  	varargs := []any{arg0, arg1}
   224  	for _, a := range arg2 {
   225  		varargs = append(varargs, a)
   226  	}
   227  	ret := m.ctrl.Call(m, "DeleteSession", varargs...)
   228  	ret0, _ := ret[0].(*Ydb_Query.DeleteSessionResponse)
   229  	ret1, _ := ret[1].(error)
   230  	return ret0, ret1
   231  }
   232  
   233  // DeleteSession indicates an expected call of DeleteSession.
   234  func (mr *MockQueryServiceClientMockRecorder) DeleteSession(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientDeleteSessionCall {
   235  	mr.mock.ctrl.T.Helper()
   236  	varargs := append([]any{arg0, arg1}, arg2...)
   237  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteSession", reflect.TypeOf((*MockQueryServiceClient)(nil).DeleteSession), varargs...)
   238  	return &MockQueryServiceClientDeleteSessionCall{Call: call}
   239  }
   240  
   241  // MockQueryServiceClientDeleteSessionCall wrap *gomock.Call
   242  type MockQueryServiceClientDeleteSessionCall struct {
   243  	*gomock.Call
   244  }
   245  
   246  // Return rewrite *gomock.Call.Return
   247  func (c *MockQueryServiceClientDeleteSessionCall) Return(arg0 *Ydb_Query.DeleteSessionResponse, arg1 error) *MockQueryServiceClientDeleteSessionCall {
   248  	c.Call = c.Call.Return(arg0, arg1)
   249  	return c
   250  }
   251  
   252  // Do rewrite *gomock.Call.Do
   253  func (c *MockQueryServiceClientDeleteSessionCall) Do(f func(context.Context, *Ydb_Query.DeleteSessionRequest, ...grpc.CallOption) (*Ydb_Query.DeleteSessionResponse, error)) *MockQueryServiceClientDeleteSessionCall {
   254  	c.Call = c.Call.Do(f)
   255  	return c
   256  }
   257  
   258  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   259  func (c *MockQueryServiceClientDeleteSessionCall) DoAndReturn(f func(context.Context, *Ydb_Query.DeleteSessionRequest, ...grpc.CallOption) (*Ydb_Query.DeleteSessionResponse, error)) *MockQueryServiceClientDeleteSessionCall {
   260  	c.Call = c.Call.DoAndReturn(f)
   261  	return c
   262  }
   263  
   264  // ExecuteQuery mocks base method.
   265  func (m *MockQueryServiceClient) ExecuteQuery(arg0 context.Context, arg1 *Ydb_Query.ExecuteQueryRequest, arg2 ...grpc.CallOption) (Ydb_Query_V1.QueryService_ExecuteQueryClient, error) {
   266  	m.ctrl.T.Helper()
   267  	varargs := []any{arg0, arg1}
   268  	for _, a := range arg2 {
   269  		varargs = append(varargs, a)
   270  	}
   271  	ret := m.ctrl.Call(m, "ExecuteQuery", varargs...)
   272  	ret0, _ := ret[0].(Ydb_Query_V1.QueryService_ExecuteQueryClient)
   273  	ret1, _ := ret[1].(error)
   274  	return ret0, ret1
   275  }
   276  
   277  // ExecuteQuery indicates an expected call of ExecuteQuery.
   278  func (mr *MockQueryServiceClientMockRecorder) ExecuteQuery(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientExecuteQueryCall {
   279  	mr.mock.ctrl.T.Helper()
   280  	varargs := append([]any{arg0, arg1}, arg2...)
   281  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecuteQuery", reflect.TypeOf((*MockQueryServiceClient)(nil).ExecuteQuery), varargs...)
   282  	return &MockQueryServiceClientExecuteQueryCall{Call: call}
   283  }
   284  
   285  // MockQueryServiceClientExecuteQueryCall wrap *gomock.Call
   286  type MockQueryServiceClientExecuteQueryCall struct {
   287  	*gomock.Call
   288  }
   289  
   290  // Return rewrite *gomock.Call.Return
   291  func (c *MockQueryServiceClientExecuteQueryCall) Return(arg0 Ydb_Query_V1.QueryService_ExecuteQueryClient, arg1 error) *MockQueryServiceClientExecuteQueryCall {
   292  	c.Call = c.Call.Return(arg0, arg1)
   293  	return c
   294  }
   295  
   296  // Do rewrite *gomock.Call.Do
   297  func (c *MockQueryServiceClientExecuteQueryCall) Do(f func(context.Context, *Ydb_Query.ExecuteQueryRequest, ...grpc.CallOption) (Ydb_Query_V1.QueryService_ExecuteQueryClient, error)) *MockQueryServiceClientExecuteQueryCall {
   298  	c.Call = c.Call.Do(f)
   299  	return c
   300  }
   301  
   302  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   303  func (c *MockQueryServiceClientExecuteQueryCall) DoAndReturn(f func(context.Context, *Ydb_Query.ExecuteQueryRequest, ...grpc.CallOption) (Ydb_Query_V1.QueryService_ExecuteQueryClient, error)) *MockQueryServiceClientExecuteQueryCall {
   304  	c.Call = c.Call.DoAndReturn(f)
   305  	return c
   306  }
   307  
   308  // ExecuteScript mocks base method.
   309  func (m *MockQueryServiceClient) ExecuteScript(arg0 context.Context, arg1 *Ydb_Query.ExecuteScriptRequest, arg2 ...grpc.CallOption) (*Ydb_Operations.Operation, error) {
   310  	m.ctrl.T.Helper()
   311  	varargs := []any{arg0, arg1}
   312  	for _, a := range arg2 {
   313  		varargs = append(varargs, a)
   314  	}
   315  	ret := m.ctrl.Call(m, "ExecuteScript", varargs...)
   316  	ret0, _ := ret[0].(*Ydb_Operations.Operation)
   317  	ret1, _ := ret[1].(error)
   318  	return ret0, ret1
   319  }
   320  
   321  // ExecuteScript indicates an expected call of ExecuteScript.
   322  func (mr *MockQueryServiceClientMockRecorder) ExecuteScript(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientExecuteScriptCall {
   323  	mr.mock.ctrl.T.Helper()
   324  	varargs := append([]any{arg0, arg1}, arg2...)
   325  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecuteScript", reflect.TypeOf((*MockQueryServiceClient)(nil).ExecuteScript), varargs...)
   326  	return &MockQueryServiceClientExecuteScriptCall{Call: call}
   327  }
   328  
   329  // MockQueryServiceClientExecuteScriptCall wrap *gomock.Call
   330  type MockQueryServiceClientExecuteScriptCall struct {
   331  	*gomock.Call
   332  }
   333  
   334  // Return rewrite *gomock.Call.Return
   335  func (c *MockQueryServiceClientExecuteScriptCall) Return(arg0 *Ydb_Operations.Operation, arg1 error) *MockQueryServiceClientExecuteScriptCall {
   336  	c.Call = c.Call.Return(arg0, arg1)
   337  	return c
   338  }
   339  
   340  // Do rewrite *gomock.Call.Do
   341  func (c *MockQueryServiceClientExecuteScriptCall) Do(f func(context.Context, *Ydb_Query.ExecuteScriptRequest, ...grpc.CallOption) (*Ydb_Operations.Operation, error)) *MockQueryServiceClientExecuteScriptCall {
   342  	c.Call = c.Call.Do(f)
   343  	return c
   344  }
   345  
   346  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   347  func (c *MockQueryServiceClientExecuteScriptCall) DoAndReturn(f func(context.Context, *Ydb_Query.ExecuteScriptRequest, ...grpc.CallOption) (*Ydb_Operations.Operation, error)) *MockQueryServiceClientExecuteScriptCall {
   348  	c.Call = c.Call.DoAndReturn(f)
   349  	return c
   350  }
   351  
   352  // FetchScriptResults mocks base method.
   353  func (m *MockQueryServiceClient) FetchScriptResults(arg0 context.Context, arg1 *Ydb_Query.FetchScriptResultsRequest, arg2 ...grpc.CallOption) (*Ydb_Query.FetchScriptResultsResponse, error) {
   354  	m.ctrl.T.Helper()
   355  	varargs := []any{arg0, arg1}
   356  	for _, a := range arg2 {
   357  		varargs = append(varargs, a)
   358  	}
   359  	ret := m.ctrl.Call(m, "FetchScriptResults", varargs...)
   360  	ret0, _ := ret[0].(*Ydb_Query.FetchScriptResultsResponse)
   361  	ret1, _ := ret[1].(error)
   362  	return ret0, ret1
   363  }
   364  
   365  // FetchScriptResults indicates an expected call of FetchScriptResults.
   366  func (mr *MockQueryServiceClientMockRecorder) FetchScriptResults(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientFetchScriptResultsCall {
   367  	mr.mock.ctrl.T.Helper()
   368  	varargs := append([]any{arg0, arg1}, arg2...)
   369  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchScriptResults", reflect.TypeOf((*MockQueryServiceClient)(nil).FetchScriptResults), varargs...)
   370  	return &MockQueryServiceClientFetchScriptResultsCall{Call: call}
   371  }
   372  
   373  // MockQueryServiceClientFetchScriptResultsCall wrap *gomock.Call
   374  type MockQueryServiceClientFetchScriptResultsCall struct {
   375  	*gomock.Call
   376  }
   377  
   378  // Return rewrite *gomock.Call.Return
   379  func (c *MockQueryServiceClientFetchScriptResultsCall) Return(arg0 *Ydb_Query.FetchScriptResultsResponse, arg1 error) *MockQueryServiceClientFetchScriptResultsCall {
   380  	c.Call = c.Call.Return(arg0, arg1)
   381  	return c
   382  }
   383  
   384  // Do rewrite *gomock.Call.Do
   385  func (c *MockQueryServiceClientFetchScriptResultsCall) Do(f func(context.Context, *Ydb_Query.FetchScriptResultsRequest, ...grpc.CallOption) (*Ydb_Query.FetchScriptResultsResponse, error)) *MockQueryServiceClientFetchScriptResultsCall {
   386  	c.Call = c.Call.Do(f)
   387  	return c
   388  }
   389  
   390  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   391  func (c *MockQueryServiceClientFetchScriptResultsCall) DoAndReturn(f func(context.Context, *Ydb_Query.FetchScriptResultsRequest, ...grpc.CallOption) (*Ydb_Query.FetchScriptResultsResponse, error)) *MockQueryServiceClientFetchScriptResultsCall {
   392  	c.Call = c.Call.DoAndReturn(f)
   393  	return c
   394  }
   395  
   396  // RollbackTransaction mocks base method.
   397  func (m *MockQueryServiceClient) RollbackTransaction(arg0 context.Context, arg1 *Ydb_Query.RollbackTransactionRequest, arg2 ...grpc.CallOption) (*Ydb_Query.RollbackTransactionResponse, error) {
   398  	m.ctrl.T.Helper()
   399  	varargs := []any{arg0, arg1}
   400  	for _, a := range arg2 {
   401  		varargs = append(varargs, a)
   402  	}
   403  	ret := m.ctrl.Call(m, "RollbackTransaction", varargs...)
   404  	ret0, _ := ret[0].(*Ydb_Query.RollbackTransactionResponse)
   405  	ret1, _ := ret[1].(error)
   406  	return ret0, ret1
   407  }
   408  
   409  // RollbackTransaction indicates an expected call of RollbackTransaction.
   410  func (mr *MockQueryServiceClientMockRecorder) RollbackTransaction(arg0, arg1 any, arg2 ...any) *MockQueryServiceClientRollbackTransactionCall {
   411  	mr.mock.ctrl.T.Helper()
   412  	varargs := append([]any{arg0, arg1}, arg2...)
   413  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RollbackTransaction", reflect.TypeOf((*MockQueryServiceClient)(nil).RollbackTransaction), varargs...)
   414  	return &MockQueryServiceClientRollbackTransactionCall{Call: call}
   415  }
   416  
   417  // MockQueryServiceClientRollbackTransactionCall wrap *gomock.Call
   418  type MockQueryServiceClientRollbackTransactionCall struct {
   419  	*gomock.Call
   420  }
   421  
   422  // Return rewrite *gomock.Call.Return
   423  func (c *MockQueryServiceClientRollbackTransactionCall) Return(arg0 *Ydb_Query.RollbackTransactionResponse, arg1 error) *MockQueryServiceClientRollbackTransactionCall {
   424  	c.Call = c.Call.Return(arg0, arg1)
   425  	return c
   426  }
   427  
   428  // Do rewrite *gomock.Call.Do
   429  func (c *MockQueryServiceClientRollbackTransactionCall) Do(f func(context.Context, *Ydb_Query.RollbackTransactionRequest, ...grpc.CallOption) (*Ydb_Query.RollbackTransactionResponse, error)) *MockQueryServiceClientRollbackTransactionCall {
   430  	c.Call = c.Call.Do(f)
   431  	return c
   432  }
   433  
   434  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   435  func (c *MockQueryServiceClientRollbackTransactionCall) DoAndReturn(f func(context.Context, *Ydb_Query.RollbackTransactionRequest, ...grpc.CallOption) (*Ydb_Query.RollbackTransactionResponse, error)) *MockQueryServiceClientRollbackTransactionCall {
   436  	c.Call = c.Call.DoAndReturn(f)
   437  	return c
   438  }
   439  
   440  // MockQueryService_AttachSessionClient is a mock of QueryService_AttachSessionClient interface.
   441  type MockQueryService_AttachSessionClient struct {
   442  	ctrl     *gomock.Controller
   443  	recorder *MockQueryService_AttachSessionClientMockRecorder
   444  }
   445  
   446  // MockQueryService_AttachSessionClientMockRecorder is the mock recorder for MockQueryService_AttachSessionClient.
   447  type MockQueryService_AttachSessionClientMockRecorder struct {
   448  	mock *MockQueryService_AttachSessionClient
   449  }
   450  
   451  // NewMockQueryService_AttachSessionClient creates a new mock instance.
   452  func NewMockQueryService_AttachSessionClient(ctrl *gomock.Controller) *MockQueryService_AttachSessionClient {
   453  	mock := &MockQueryService_AttachSessionClient{ctrl: ctrl}
   454  	mock.recorder = &MockQueryService_AttachSessionClientMockRecorder{mock}
   455  	return mock
   456  }
   457  
   458  // EXPECT returns an object that allows the caller to indicate expected use.
   459  func (m *MockQueryService_AttachSessionClient) EXPECT() *MockQueryService_AttachSessionClientMockRecorder {
   460  	return m.recorder
   461  }
   462  
   463  // CloseSend mocks base method.
   464  func (m *MockQueryService_AttachSessionClient) CloseSend() error {
   465  	m.ctrl.T.Helper()
   466  	ret := m.ctrl.Call(m, "CloseSend")
   467  	ret0, _ := ret[0].(error)
   468  	return ret0
   469  }
   470  
   471  // CloseSend indicates an expected call of CloseSend.
   472  func (mr *MockQueryService_AttachSessionClientMockRecorder) CloseSend() *MockQueryService_AttachSessionClientCloseSendCall {
   473  	mr.mock.ctrl.T.Helper()
   474  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).CloseSend))
   475  	return &MockQueryService_AttachSessionClientCloseSendCall{Call: call}
   476  }
   477  
   478  // MockQueryService_AttachSessionClientCloseSendCall wrap *gomock.Call
   479  type MockQueryService_AttachSessionClientCloseSendCall struct {
   480  	*gomock.Call
   481  }
   482  
   483  // Return rewrite *gomock.Call.Return
   484  func (c *MockQueryService_AttachSessionClientCloseSendCall) Return(arg0 error) *MockQueryService_AttachSessionClientCloseSendCall {
   485  	c.Call = c.Call.Return(arg0)
   486  	return c
   487  }
   488  
   489  // Do rewrite *gomock.Call.Do
   490  func (c *MockQueryService_AttachSessionClientCloseSendCall) Do(f func() error) *MockQueryService_AttachSessionClientCloseSendCall {
   491  	c.Call = c.Call.Do(f)
   492  	return c
   493  }
   494  
   495  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   496  func (c *MockQueryService_AttachSessionClientCloseSendCall) DoAndReturn(f func() error) *MockQueryService_AttachSessionClientCloseSendCall {
   497  	c.Call = c.Call.DoAndReturn(f)
   498  	return c
   499  }
   500  
   501  // Context mocks base method.
   502  func (m *MockQueryService_AttachSessionClient) Context() context.Context {
   503  	m.ctrl.T.Helper()
   504  	ret := m.ctrl.Call(m, "Context")
   505  	ret0, _ := ret[0].(context.Context)
   506  	return ret0
   507  }
   508  
   509  // Context indicates an expected call of Context.
   510  func (mr *MockQueryService_AttachSessionClientMockRecorder) Context() *MockQueryService_AttachSessionClientContextCall {
   511  	mr.mock.ctrl.T.Helper()
   512  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).Context))
   513  	return &MockQueryService_AttachSessionClientContextCall{Call: call}
   514  }
   515  
   516  // MockQueryService_AttachSessionClientContextCall wrap *gomock.Call
   517  type MockQueryService_AttachSessionClientContextCall struct {
   518  	*gomock.Call
   519  }
   520  
   521  // Return rewrite *gomock.Call.Return
   522  func (c *MockQueryService_AttachSessionClientContextCall) Return(arg0 context.Context) *MockQueryService_AttachSessionClientContextCall {
   523  	c.Call = c.Call.Return(arg0)
   524  	return c
   525  }
   526  
   527  // Do rewrite *gomock.Call.Do
   528  func (c *MockQueryService_AttachSessionClientContextCall) Do(f func() context.Context) *MockQueryService_AttachSessionClientContextCall {
   529  	c.Call = c.Call.Do(f)
   530  	return c
   531  }
   532  
   533  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   534  func (c *MockQueryService_AttachSessionClientContextCall) DoAndReturn(f func() context.Context) *MockQueryService_AttachSessionClientContextCall {
   535  	c.Call = c.Call.DoAndReturn(f)
   536  	return c
   537  }
   538  
   539  // Header mocks base method.
   540  func (m *MockQueryService_AttachSessionClient) Header() (metadata.MD, error) {
   541  	m.ctrl.T.Helper()
   542  	ret := m.ctrl.Call(m, "Header")
   543  	ret0, _ := ret[0].(metadata.MD)
   544  	ret1, _ := ret[1].(error)
   545  	return ret0, ret1
   546  }
   547  
   548  // Header indicates an expected call of Header.
   549  func (mr *MockQueryService_AttachSessionClientMockRecorder) Header() *MockQueryService_AttachSessionClientHeaderCall {
   550  	mr.mock.ctrl.T.Helper()
   551  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).Header))
   552  	return &MockQueryService_AttachSessionClientHeaderCall{Call: call}
   553  }
   554  
   555  // MockQueryService_AttachSessionClientHeaderCall wrap *gomock.Call
   556  type MockQueryService_AttachSessionClientHeaderCall struct {
   557  	*gomock.Call
   558  }
   559  
   560  // Return rewrite *gomock.Call.Return
   561  func (c *MockQueryService_AttachSessionClientHeaderCall) Return(arg0 metadata.MD, arg1 error) *MockQueryService_AttachSessionClientHeaderCall {
   562  	c.Call = c.Call.Return(arg0, arg1)
   563  	return c
   564  }
   565  
   566  // Do rewrite *gomock.Call.Do
   567  func (c *MockQueryService_AttachSessionClientHeaderCall) Do(f func() (metadata.MD, error)) *MockQueryService_AttachSessionClientHeaderCall {
   568  	c.Call = c.Call.Do(f)
   569  	return c
   570  }
   571  
   572  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   573  func (c *MockQueryService_AttachSessionClientHeaderCall) DoAndReturn(f func() (metadata.MD, error)) *MockQueryService_AttachSessionClientHeaderCall {
   574  	c.Call = c.Call.DoAndReturn(f)
   575  	return c
   576  }
   577  
   578  // Recv mocks base method.
   579  func (m *MockQueryService_AttachSessionClient) Recv() (*Ydb_Query.SessionState, error) {
   580  	m.ctrl.T.Helper()
   581  	ret := m.ctrl.Call(m, "Recv")
   582  	ret0, _ := ret[0].(*Ydb_Query.SessionState)
   583  	ret1, _ := ret[1].(error)
   584  	return ret0, ret1
   585  }
   586  
   587  // Recv indicates an expected call of Recv.
   588  func (mr *MockQueryService_AttachSessionClientMockRecorder) Recv() *MockQueryService_AttachSessionClientRecvCall {
   589  	mr.mock.ctrl.T.Helper()
   590  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).Recv))
   591  	return &MockQueryService_AttachSessionClientRecvCall{Call: call}
   592  }
   593  
   594  // MockQueryService_AttachSessionClientRecvCall wrap *gomock.Call
   595  type MockQueryService_AttachSessionClientRecvCall struct {
   596  	*gomock.Call
   597  }
   598  
   599  // Return rewrite *gomock.Call.Return
   600  func (c *MockQueryService_AttachSessionClientRecvCall) Return(arg0 *Ydb_Query.SessionState, arg1 error) *MockQueryService_AttachSessionClientRecvCall {
   601  	c.Call = c.Call.Return(arg0, arg1)
   602  	return c
   603  }
   604  
   605  // Do rewrite *gomock.Call.Do
   606  func (c *MockQueryService_AttachSessionClientRecvCall) Do(f func() (*Ydb_Query.SessionState, error)) *MockQueryService_AttachSessionClientRecvCall {
   607  	c.Call = c.Call.Do(f)
   608  	return c
   609  }
   610  
   611  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   612  func (c *MockQueryService_AttachSessionClientRecvCall) DoAndReturn(f func() (*Ydb_Query.SessionState, error)) *MockQueryService_AttachSessionClientRecvCall {
   613  	c.Call = c.Call.DoAndReturn(f)
   614  	return c
   615  }
   616  
   617  // RecvMsg mocks base method.
   618  func (m *MockQueryService_AttachSessionClient) RecvMsg(arg0 any) error {
   619  	m.ctrl.T.Helper()
   620  	ret := m.ctrl.Call(m, "RecvMsg", arg0)
   621  	ret0, _ := ret[0].(error)
   622  	return ret0
   623  }
   624  
   625  // RecvMsg indicates an expected call of RecvMsg.
   626  func (mr *MockQueryService_AttachSessionClientMockRecorder) RecvMsg(arg0 any) *MockQueryService_AttachSessionClientRecvMsgCall {
   627  	mr.mock.ctrl.T.Helper()
   628  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).RecvMsg), arg0)
   629  	return &MockQueryService_AttachSessionClientRecvMsgCall{Call: call}
   630  }
   631  
   632  // MockQueryService_AttachSessionClientRecvMsgCall wrap *gomock.Call
   633  type MockQueryService_AttachSessionClientRecvMsgCall struct {
   634  	*gomock.Call
   635  }
   636  
   637  // Return rewrite *gomock.Call.Return
   638  func (c *MockQueryService_AttachSessionClientRecvMsgCall) Return(arg0 error) *MockQueryService_AttachSessionClientRecvMsgCall {
   639  	c.Call = c.Call.Return(arg0)
   640  	return c
   641  }
   642  
   643  // Do rewrite *gomock.Call.Do
   644  func (c *MockQueryService_AttachSessionClientRecvMsgCall) Do(f func(any) error) *MockQueryService_AttachSessionClientRecvMsgCall {
   645  	c.Call = c.Call.Do(f)
   646  	return c
   647  }
   648  
   649  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   650  func (c *MockQueryService_AttachSessionClientRecvMsgCall) DoAndReturn(f func(any) error) *MockQueryService_AttachSessionClientRecvMsgCall {
   651  	c.Call = c.Call.DoAndReturn(f)
   652  	return c
   653  }
   654  
   655  // SendMsg mocks base method.
   656  func (m *MockQueryService_AttachSessionClient) SendMsg(arg0 any) error {
   657  	m.ctrl.T.Helper()
   658  	ret := m.ctrl.Call(m, "SendMsg", arg0)
   659  	ret0, _ := ret[0].(error)
   660  	return ret0
   661  }
   662  
   663  // SendMsg indicates an expected call of SendMsg.
   664  func (mr *MockQueryService_AttachSessionClientMockRecorder) SendMsg(arg0 any) *MockQueryService_AttachSessionClientSendMsgCall {
   665  	mr.mock.ctrl.T.Helper()
   666  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).SendMsg), arg0)
   667  	return &MockQueryService_AttachSessionClientSendMsgCall{Call: call}
   668  }
   669  
   670  // MockQueryService_AttachSessionClientSendMsgCall wrap *gomock.Call
   671  type MockQueryService_AttachSessionClientSendMsgCall struct {
   672  	*gomock.Call
   673  }
   674  
   675  // Return rewrite *gomock.Call.Return
   676  func (c *MockQueryService_AttachSessionClientSendMsgCall) Return(arg0 error) *MockQueryService_AttachSessionClientSendMsgCall {
   677  	c.Call = c.Call.Return(arg0)
   678  	return c
   679  }
   680  
   681  // Do rewrite *gomock.Call.Do
   682  func (c *MockQueryService_AttachSessionClientSendMsgCall) Do(f func(any) error) *MockQueryService_AttachSessionClientSendMsgCall {
   683  	c.Call = c.Call.Do(f)
   684  	return c
   685  }
   686  
   687  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   688  func (c *MockQueryService_AttachSessionClientSendMsgCall) DoAndReturn(f func(any) error) *MockQueryService_AttachSessionClientSendMsgCall {
   689  	c.Call = c.Call.DoAndReturn(f)
   690  	return c
   691  }
   692  
   693  // Trailer mocks base method.
   694  func (m *MockQueryService_AttachSessionClient) Trailer() metadata.MD {
   695  	m.ctrl.T.Helper()
   696  	ret := m.ctrl.Call(m, "Trailer")
   697  	ret0, _ := ret[0].(metadata.MD)
   698  	return ret0
   699  }
   700  
   701  // Trailer indicates an expected call of Trailer.
   702  func (mr *MockQueryService_AttachSessionClientMockRecorder) Trailer() *MockQueryService_AttachSessionClientTrailerCall {
   703  	mr.mock.ctrl.T.Helper()
   704  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockQueryService_AttachSessionClient)(nil).Trailer))
   705  	return &MockQueryService_AttachSessionClientTrailerCall{Call: call}
   706  }
   707  
   708  // MockQueryService_AttachSessionClientTrailerCall wrap *gomock.Call
   709  type MockQueryService_AttachSessionClientTrailerCall struct {
   710  	*gomock.Call
   711  }
   712  
   713  // Return rewrite *gomock.Call.Return
   714  func (c *MockQueryService_AttachSessionClientTrailerCall) Return(arg0 metadata.MD) *MockQueryService_AttachSessionClientTrailerCall {
   715  	c.Call = c.Call.Return(arg0)
   716  	return c
   717  }
   718  
   719  // Do rewrite *gomock.Call.Do
   720  func (c *MockQueryService_AttachSessionClientTrailerCall) Do(f func() metadata.MD) *MockQueryService_AttachSessionClientTrailerCall {
   721  	c.Call = c.Call.Do(f)
   722  	return c
   723  }
   724  
   725  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   726  func (c *MockQueryService_AttachSessionClientTrailerCall) DoAndReturn(f func() metadata.MD) *MockQueryService_AttachSessionClientTrailerCall {
   727  	c.Call = c.Call.DoAndReturn(f)
   728  	return c
   729  }
   730  
   731  // MockQueryService_ExecuteQueryClient is a mock of QueryService_ExecuteQueryClient interface.
   732  type MockQueryService_ExecuteQueryClient struct {
   733  	ctrl     *gomock.Controller
   734  	recorder *MockQueryService_ExecuteQueryClientMockRecorder
   735  }
   736  
   737  // MockQueryService_ExecuteQueryClientMockRecorder is the mock recorder for MockQueryService_ExecuteQueryClient.
   738  type MockQueryService_ExecuteQueryClientMockRecorder struct {
   739  	mock *MockQueryService_ExecuteQueryClient
   740  }
   741  
   742  // NewMockQueryService_ExecuteQueryClient creates a new mock instance.
   743  func NewMockQueryService_ExecuteQueryClient(ctrl *gomock.Controller) *MockQueryService_ExecuteQueryClient {
   744  	mock := &MockQueryService_ExecuteQueryClient{ctrl: ctrl}
   745  	mock.recorder = &MockQueryService_ExecuteQueryClientMockRecorder{mock}
   746  	return mock
   747  }
   748  
   749  // EXPECT returns an object that allows the caller to indicate expected use.
   750  func (m *MockQueryService_ExecuteQueryClient) EXPECT() *MockQueryService_ExecuteQueryClientMockRecorder {
   751  	return m.recorder
   752  }
   753  
   754  // CloseSend mocks base method.
   755  func (m *MockQueryService_ExecuteQueryClient) CloseSend() error {
   756  	m.ctrl.T.Helper()
   757  	ret := m.ctrl.Call(m, "CloseSend")
   758  	ret0, _ := ret[0].(error)
   759  	return ret0
   760  }
   761  
   762  // CloseSend indicates an expected call of CloseSend.
   763  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) CloseSend() *MockQueryService_ExecuteQueryClientCloseSendCall {
   764  	mr.mock.ctrl.T.Helper()
   765  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).CloseSend))
   766  	return &MockQueryService_ExecuteQueryClientCloseSendCall{Call: call}
   767  }
   768  
   769  // MockQueryService_ExecuteQueryClientCloseSendCall wrap *gomock.Call
   770  type MockQueryService_ExecuteQueryClientCloseSendCall struct {
   771  	*gomock.Call
   772  }
   773  
   774  // Return rewrite *gomock.Call.Return
   775  func (c *MockQueryService_ExecuteQueryClientCloseSendCall) Return(arg0 error) *MockQueryService_ExecuteQueryClientCloseSendCall {
   776  	c.Call = c.Call.Return(arg0)
   777  	return c
   778  }
   779  
   780  // Do rewrite *gomock.Call.Do
   781  func (c *MockQueryService_ExecuteQueryClientCloseSendCall) Do(f func() error) *MockQueryService_ExecuteQueryClientCloseSendCall {
   782  	c.Call = c.Call.Do(f)
   783  	return c
   784  }
   785  
   786  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   787  func (c *MockQueryService_ExecuteQueryClientCloseSendCall) DoAndReturn(f func() error) *MockQueryService_ExecuteQueryClientCloseSendCall {
   788  	c.Call = c.Call.DoAndReturn(f)
   789  	return c
   790  }
   791  
   792  // Context mocks base method.
   793  func (m *MockQueryService_ExecuteQueryClient) Context() context.Context {
   794  	m.ctrl.T.Helper()
   795  	ret := m.ctrl.Call(m, "Context")
   796  	ret0, _ := ret[0].(context.Context)
   797  	return ret0
   798  }
   799  
   800  // Context indicates an expected call of Context.
   801  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) Context() *MockQueryService_ExecuteQueryClientContextCall {
   802  	mr.mock.ctrl.T.Helper()
   803  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).Context))
   804  	return &MockQueryService_ExecuteQueryClientContextCall{Call: call}
   805  }
   806  
   807  // MockQueryService_ExecuteQueryClientContextCall wrap *gomock.Call
   808  type MockQueryService_ExecuteQueryClientContextCall struct {
   809  	*gomock.Call
   810  }
   811  
   812  // Return rewrite *gomock.Call.Return
   813  func (c *MockQueryService_ExecuteQueryClientContextCall) Return(arg0 context.Context) *MockQueryService_ExecuteQueryClientContextCall {
   814  	c.Call = c.Call.Return(arg0)
   815  	return c
   816  }
   817  
   818  // Do rewrite *gomock.Call.Do
   819  func (c *MockQueryService_ExecuteQueryClientContextCall) Do(f func() context.Context) *MockQueryService_ExecuteQueryClientContextCall {
   820  	c.Call = c.Call.Do(f)
   821  	return c
   822  }
   823  
   824  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   825  func (c *MockQueryService_ExecuteQueryClientContextCall) DoAndReturn(f func() context.Context) *MockQueryService_ExecuteQueryClientContextCall {
   826  	c.Call = c.Call.DoAndReturn(f)
   827  	return c
   828  }
   829  
   830  // Header mocks base method.
   831  func (m *MockQueryService_ExecuteQueryClient) Header() (metadata.MD, error) {
   832  	m.ctrl.T.Helper()
   833  	ret := m.ctrl.Call(m, "Header")
   834  	ret0, _ := ret[0].(metadata.MD)
   835  	ret1, _ := ret[1].(error)
   836  	return ret0, ret1
   837  }
   838  
   839  // Header indicates an expected call of Header.
   840  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) Header() *MockQueryService_ExecuteQueryClientHeaderCall {
   841  	mr.mock.ctrl.T.Helper()
   842  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).Header))
   843  	return &MockQueryService_ExecuteQueryClientHeaderCall{Call: call}
   844  }
   845  
   846  // MockQueryService_ExecuteQueryClientHeaderCall wrap *gomock.Call
   847  type MockQueryService_ExecuteQueryClientHeaderCall struct {
   848  	*gomock.Call
   849  }
   850  
   851  // Return rewrite *gomock.Call.Return
   852  func (c *MockQueryService_ExecuteQueryClientHeaderCall) Return(arg0 metadata.MD, arg1 error) *MockQueryService_ExecuteQueryClientHeaderCall {
   853  	c.Call = c.Call.Return(arg0, arg1)
   854  	return c
   855  }
   856  
   857  // Do rewrite *gomock.Call.Do
   858  func (c *MockQueryService_ExecuteQueryClientHeaderCall) Do(f func() (metadata.MD, error)) *MockQueryService_ExecuteQueryClientHeaderCall {
   859  	c.Call = c.Call.Do(f)
   860  	return c
   861  }
   862  
   863  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   864  func (c *MockQueryService_ExecuteQueryClientHeaderCall) DoAndReturn(f func() (metadata.MD, error)) *MockQueryService_ExecuteQueryClientHeaderCall {
   865  	c.Call = c.Call.DoAndReturn(f)
   866  	return c
   867  }
   868  
   869  // Recv mocks base method.
   870  func (m *MockQueryService_ExecuteQueryClient) Recv() (*Ydb_Query.ExecuteQueryResponsePart, error) {
   871  	m.ctrl.T.Helper()
   872  	ret := m.ctrl.Call(m, "Recv")
   873  	ret0, _ := ret[0].(*Ydb_Query.ExecuteQueryResponsePart)
   874  	ret1, _ := ret[1].(error)
   875  	return ret0, ret1
   876  }
   877  
   878  // Recv indicates an expected call of Recv.
   879  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) Recv() *MockQueryService_ExecuteQueryClientRecvCall {
   880  	mr.mock.ctrl.T.Helper()
   881  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).Recv))
   882  	return &MockQueryService_ExecuteQueryClientRecvCall{Call: call}
   883  }
   884  
   885  // MockQueryService_ExecuteQueryClientRecvCall wrap *gomock.Call
   886  type MockQueryService_ExecuteQueryClientRecvCall struct {
   887  	*gomock.Call
   888  }
   889  
   890  // Return rewrite *gomock.Call.Return
   891  func (c *MockQueryService_ExecuteQueryClientRecvCall) Return(arg0 *Ydb_Query.ExecuteQueryResponsePart, arg1 error) *MockQueryService_ExecuteQueryClientRecvCall {
   892  	c.Call = c.Call.Return(arg0, arg1)
   893  	return c
   894  }
   895  
   896  // Do rewrite *gomock.Call.Do
   897  func (c *MockQueryService_ExecuteQueryClientRecvCall) Do(f func() (*Ydb_Query.ExecuteQueryResponsePart, error)) *MockQueryService_ExecuteQueryClientRecvCall {
   898  	c.Call = c.Call.Do(f)
   899  	return c
   900  }
   901  
   902  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   903  func (c *MockQueryService_ExecuteQueryClientRecvCall) DoAndReturn(f func() (*Ydb_Query.ExecuteQueryResponsePart, error)) *MockQueryService_ExecuteQueryClientRecvCall {
   904  	c.Call = c.Call.DoAndReturn(f)
   905  	return c
   906  }
   907  
   908  // RecvMsg mocks base method.
   909  func (m *MockQueryService_ExecuteQueryClient) RecvMsg(arg0 any) error {
   910  	m.ctrl.T.Helper()
   911  	ret := m.ctrl.Call(m, "RecvMsg", arg0)
   912  	ret0, _ := ret[0].(error)
   913  	return ret0
   914  }
   915  
   916  // RecvMsg indicates an expected call of RecvMsg.
   917  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) RecvMsg(arg0 any) *MockQueryService_ExecuteQueryClientRecvMsgCall {
   918  	mr.mock.ctrl.T.Helper()
   919  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).RecvMsg), arg0)
   920  	return &MockQueryService_ExecuteQueryClientRecvMsgCall{Call: call}
   921  }
   922  
   923  // MockQueryService_ExecuteQueryClientRecvMsgCall wrap *gomock.Call
   924  type MockQueryService_ExecuteQueryClientRecvMsgCall struct {
   925  	*gomock.Call
   926  }
   927  
   928  // Return rewrite *gomock.Call.Return
   929  func (c *MockQueryService_ExecuteQueryClientRecvMsgCall) Return(arg0 error) *MockQueryService_ExecuteQueryClientRecvMsgCall {
   930  	c.Call = c.Call.Return(arg0)
   931  	return c
   932  }
   933  
   934  // Do rewrite *gomock.Call.Do
   935  func (c *MockQueryService_ExecuteQueryClientRecvMsgCall) Do(f func(any) error) *MockQueryService_ExecuteQueryClientRecvMsgCall {
   936  	c.Call = c.Call.Do(f)
   937  	return c
   938  }
   939  
   940  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   941  func (c *MockQueryService_ExecuteQueryClientRecvMsgCall) DoAndReturn(f func(any) error) *MockQueryService_ExecuteQueryClientRecvMsgCall {
   942  	c.Call = c.Call.DoAndReturn(f)
   943  	return c
   944  }
   945  
   946  // SendMsg mocks base method.
   947  func (m *MockQueryService_ExecuteQueryClient) SendMsg(arg0 any) error {
   948  	m.ctrl.T.Helper()
   949  	ret := m.ctrl.Call(m, "SendMsg", arg0)
   950  	ret0, _ := ret[0].(error)
   951  	return ret0
   952  }
   953  
   954  // SendMsg indicates an expected call of SendMsg.
   955  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) SendMsg(arg0 any) *MockQueryService_ExecuteQueryClientSendMsgCall {
   956  	mr.mock.ctrl.T.Helper()
   957  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).SendMsg), arg0)
   958  	return &MockQueryService_ExecuteQueryClientSendMsgCall{Call: call}
   959  }
   960  
   961  // MockQueryService_ExecuteQueryClientSendMsgCall wrap *gomock.Call
   962  type MockQueryService_ExecuteQueryClientSendMsgCall struct {
   963  	*gomock.Call
   964  }
   965  
   966  // Return rewrite *gomock.Call.Return
   967  func (c *MockQueryService_ExecuteQueryClientSendMsgCall) Return(arg0 error) *MockQueryService_ExecuteQueryClientSendMsgCall {
   968  	c.Call = c.Call.Return(arg0)
   969  	return c
   970  }
   971  
   972  // Do rewrite *gomock.Call.Do
   973  func (c *MockQueryService_ExecuteQueryClientSendMsgCall) Do(f func(any) error) *MockQueryService_ExecuteQueryClientSendMsgCall {
   974  	c.Call = c.Call.Do(f)
   975  	return c
   976  }
   977  
   978  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   979  func (c *MockQueryService_ExecuteQueryClientSendMsgCall) DoAndReturn(f func(any) error) *MockQueryService_ExecuteQueryClientSendMsgCall {
   980  	c.Call = c.Call.DoAndReturn(f)
   981  	return c
   982  }
   983  
   984  // Trailer mocks base method.
   985  func (m *MockQueryService_ExecuteQueryClient) Trailer() metadata.MD {
   986  	m.ctrl.T.Helper()
   987  	ret := m.ctrl.Call(m, "Trailer")
   988  	ret0, _ := ret[0].(metadata.MD)
   989  	return ret0
   990  }
   991  
   992  // Trailer indicates an expected call of Trailer.
   993  func (mr *MockQueryService_ExecuteQueryClientMockRecorder) Trailer() *MockQueryService_ExecuteQueryClientTrailerCall {
   994  	mr.mock.ctrl.T.Helper()
   995  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockQueryService_ExecuteQueryClient)(nil).Trailer))
   996  	return &MockQueryService_ExecuteQueryClientTrailerCall{Call: call}
   997  }
   998  
   999  // MockQueryService_ExecuteQueryClientTrailerCall wrap *gomock.Call
  1000  type MockQueryService_ExecuteQueryClientTrailerCall struct {
  1001  	*gomock.Call
  1002  }
  1003  
  1004  // Return rewrite *gomock.Call.Return
  1005  func (c *MockQueryService_ExecuteQueryClientTrailerCall) Return(arg0 metadata.MD) *MockQueryService_ExecuteQueryClientTrailerCall {
  1006  	c.Call = c.Call.Return(arg0)
  1007  	return c
  1008  }
  1009  
  1010  // Do rewrite *gomock.Call.Do
  1011  func (c *MockQueryService_ExecuteQueryClientTrailerCall) Do(f func() metadata.MD) *MockQueryService_ExecuteQueryClientTrailerCall {
  1012  	c.Call = c.Call.Do(f)
  1013  	return c
  1014  }
  1015  
  1016  // DoAndReturn rewrite *gomock.Call.DoAndReturn
  1017  func (c *MockQueryService_ExecuteQueryClientTrailerCall) DoAndReturn(f func() metadata.MD) *MockQueryService_ExecuteQueryClientTrailerCall {
  1018  	c.Call = c.Call.DoAndReturn(f)
  1019  	return c
  1020  }