// Code generated by MockGen. DO NOT EDIT.
// Source: gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v18/internal/module/job_router/server (interfaces: admissionControllerClient)
//
// Generated by this command:
//
//	mockgen -typed -self_package gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v18/internal/module/job_router/server -destination mock_for_test.go -package server gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v18/internal/module/job_router/server admissionControllerClient
//

// Package server is a generated GoMock package.
package server

import (
	context "context"
	reflect "reflect"

	api "gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v18/internal/api"
	rpc "gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v18/internal/module/runner_controller/rpc"
	gomock "go.uber.org/mock/gomock"
)

// MockadmissionControllerClient is a mock of admissionControllerClient interface.
type MockadmissionControllerClient struct {
	ctrl     *gomock.Controller
	recorder *MockadmissionControllerClientMockRecorder
	isgomock struct{}
}

// MockadmissionControllerClientMockRecorder is the mock recorder for MockadmissionControllerClient.
type MockadmissionControllerClientMockRecorder struct {
	mock *MockadmissionControllerClient
}

// NewMockadmissionControllerClient creates a new mock instance.
func NewMockadmissionControllerClient(ctrl *gomock.Controller) *MockadmissionControllerClient {
	mock := &MockadmissionControllerClient{ctrl: ctrl}
	mock.recorder = &MockadmissionControllerClientMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockadmissionControllerClient) EXPECT() *MockadmissionControllerClientMockRecorder {
	return m.recorder
}

// AdmitJob mocks base method.
func (m *MockadmissionControllerClient) AdmitJob(ctx context.Context, agentKey api.AgentKey, req *rpc.AdmitJobRequest) (*rpc.AdmitJobResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "AdmitJob", ctx, agentKey, req)
	ret0, _ := ret[0].(*rpc.AdmitJobResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// AdmitJob indicates an expected call of AdmitJob.
func (mr *MockadmissionControllerClientMockRecorder) AdmitJob(ctx, agentKey, req any) *MockadmissionControllerClientAdmitJobCall {
	mr.mock.ctrl.T.Helper()
	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AdmitJob", reflect.TypeOf((*MockadmissionControllerClient)(nil).AdmitJob), ctx, agentKey, req)
	return &MockadmissionControllerClientAdmitJobCall{Call: call}
}

// MockadmissionControllerClientAdmitJobCall wrap *gomock.Call
type MockadmissionControllerClientAdmitJobCall struct {
	*gomock.Call
}

// Return rewrite *gomock.Call.Return
func (c *MockadmissionControllerClientAdmitJobCall) Return(arg0 *rpc.AdmitJobResponse, arg1 error) *MockadmissionControllerClientAdmitJobCall {
	c.Call = c.Call.Return(arg0, arg1)
	return c
}

// Do rewrite *gomock.Call.Do
func (c *MockadmissionControllerClientAdmitJobCall) Do(f func(context.Context, api.AgentKey, *rpc.AdmitJobRequest) (*rpc.AdmitJobResponse, error)) *MockadmissionControllerClientAdmitJobCall {
	c.Call = c.Call.Do(f)
	return c
}

// DoAndReturn rewrite *gomock.Call.DoAndReturn
func (c *MockadmissionControllerClientAdmitJobCall) DoAndReturn(f func(context.Context, api.AgentKey, *rpc.AdmitJobRequest) (*rpc.AdmitJobResponse, error)) *MockadmissionControllerClientAdmitJobCall {
	c.Call = c.Call.DoAndReturn(f)
	return c
}
