go.chromium.org/luci@v0.0.0-20240309015107-7cdc2e660f33/buildbucket/appengine/internal/clients/swarming.mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: swarming.go
     3  
     4  package clients
     5  
     6  import (
     7  	context "context"
     8  	reflect "reflect"
     9  
    10  	gomock "github.com/golang/mock/gomock"
    11  	apipb "go.chromium.org/luci/swarming/proto/api_v2"
    12  )
    13  
    14  // MockSwarmingClient is a mock of SwarmingClient interface.
    15  type MockSwarmingClient struct {
    16  	ctrl     *gomock.Controller
    17  	recorder *MockSwarmingClientMockRecorder
    18  }
    19  
    20  // MockSwarmingClientMockRecorder is the mock recorder for MockSwarmingClient.
    21  type MockSwarmingClientMockRecorder struct {
    22  	mock *MockSwarmingClient
    23  }
    24  
    25  // NewMockSwarmingClient creates a new mock instance.
    26  func NewMockSwarmingClient(ctrl *gomock.Controller) *MockSwarmingClient {
    27  	mock := &MockSwarmingClient{ctrl: ctrl}
    28  	mock.recorder = &MockSwarmingClientMockRecorder{mock}
    29  	return mock
    30  }
    31  
    32  // EXPECT returns an object that allows the caller to indicate expected use.
    33  func (m *MockSwarmingClient) EXPECT() *MockSwarmingClientMockRecorder {
    34  	return m.recorder
    35  }
    36  
    37  // CancelTask mocks base method.
    38  func (m *MockSwarmingClient) CancelTask(ctx context.Context, req *apipb.TaskCancelRequest) (*apipb.CancelResponse, error) {
    39  	m.ctrl.T.Helper()
    40  	ret := m.ctrl.Call(m, "CancelTask", ctx, req)
    41  	ret0, _ := ret[0].(*apipb.CancelResponse)
    42  	ret1, _ := ret[1].(error)
    43  	return ret0, ret1
    44  }
    45  
    46  // CancelTask indicates an expected call of CancelTask.
    47  func (mr *MockSwarmingClientMockRecorder) CancelTask(ctx, req interface{}) *gomock.Call {
    48  	mr.mock.ctrl.T.Helper()
    49  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelTask", reflect.TypeOf((*MockSwarmingClient)(nil).CancelTask), ctx, req)
    50  }
    51  
    52  // CreateTask mocks base method.
    53  func (m *MockSwarmingClient) CreateTask(ctx context.Context, createTaskReq *apipb.NewTaskRequest) (*apipb.TaskRequestMetadataResponse, error) {
    54  	m.ctrl.T.Helper()
    55  	ret := m.ctrl.Call(m, "CreateTask", ctx, createTaskReq)
    56  	ret0, _ := ret[0].(*apipb.TaskRequestMetadataResponse)
    57  	ret1, _ := ret[1].(error)
    58  	return ret0, ret1
    59  }
    60  
    61  // CreateTask indicates an expected call of CreateTask.
    62  func (mr *MockSwarmingClientMockRecorder) CreateTask(ctx, createTaskReq interface{}) *gomock.Call {
    63  	mr.mock.ctrl.T.Helper()
    64  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockSwarmingClient)(nil).CreateTask), ctx, createTaskReq)
    65  }
    66  
    67  // GetTaskResult mocks base method.
    68  func (m *MockSwarmingClient) GetTaskResult(ctx context.Context, taskID string) (*apipb.TaskResultResponse, error) {
    69  	m.ctrl.T.Helper()
    70  	ret := m.ctrl.Call(m, "GetTaskResult", ctx, taskID)
    71  	ret0, _ := ret[0].(*apipb.TaskResultResponse)
    72  	ret1, _ := ret[1].(error)
    73  	return ret0, ret1
    74  }
    75  
    76  // GetTaskResult indicates an expected call of GetTaskResult.
    77  func (mr *MockSwarmingClientMockRecorder) GetTaskResult(ctx, taskID interface{}) *gomock.Call {
    78  	mr.mock.ctrl.T.Helper()
    79  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskResult", reflect.TypeOf((*MockSwarmingClient)(nil).GetTaskResult), ctx, taskID)
    80  }