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 }