// Code generated by mockery v2.12.3. DO NOT EDIT.

package mocks

import (
	errors "github.com/pingcap/test-infra/common/errors"
	mock "github.com/stretchr/testify/mock"

	request "github.com/pingcap/test-infra/common/model/rms/request"

	response "github.com/pingcap/test-infra/common/model/rms/response"

	utils "github.com/pingcap/test-infra/services/rms/utils"
)

// StatisticsService is an autogenerated mock type for the StatisticsService type
type StatisticsService struct {
	mock.Mock
}

// GetResourcePoolStatus provides a mock function with given fields: ctx, req
func (_m *StatisticsService) GetResourcePoolStatus(ctx utils.Context, req request.GetResourcePoolStatusRequest) (*response.GetResourcePoolStatusResponse, *errors.StatusError) {
	ret := _m.Called(ctx, req)

	var r0 *response.GetResourcePoolStatusResponse
	if rf, ok := ret.Get(0).(func(utils.Context, request.GetResourcePoolStatusRequest) *response.GetResourcePoolStatusResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*response.GetResourcePoolStatusResponse)
		}
	}

	var r1 *errors.StatusError
	if rf, ok := ret.Get(1).(func(utils.Context, request.GetResourcePoolStatusRequest) *errors.StatusError); ok {
		r1 = rf(ctx, req)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(*errors.StatusError)
		}
	}

	return r0, r1
}

// GetTestbedResourceOvertime provides a mock function with given fields: ctx, req
func (_m *StatisticsService) GetTestbedResourceOvertime(ctx utils.Context, req request.GetTestbedResourceOvertimeRequest) (*response.GetTestbedResourceOvertimeResponse, *errors.StatusError) {
	ret := _m.Called(ctx, req)

	var r0 *response.GetTestbedResourceOvertimeResponse
	if rf, ok := ret.Get(0).(func(utils.Context, request.GetTestbedResourceOvertimeRequest) *response.GetTestbedResourceOvertimeResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*response.GetTestbedResourceOvertimeResponse)
		}
	}

	var r1 *errors.StatusError
	if rf, ok := ret.Get(1).(func(utils.Context, request.GetTestbedResourceOvertimeRequest) *errors.StatusError); ok {
		r1 = rf(ctx, req)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(*errors.StatusError)
		}
	}

	return r0, r1
}

// GetTestbedsResource provides a mock function with given fields: ctx, req
func (_m *StatisticsService) GetTestbedsResource(ctx utils.Context, req request.GetTestbedsResourceRequest) (*response.GetTestbedsResourceResponse, *errors.StatusError) {
	ret := _m.Called(ctx, req)

	var r0 *response.GetTestbedsResourceResponse
	if rf, ok := ret.Get(0).(func(utils.Context, request.GetTestbedsResourceRequest) *response.GetTestbedsResourceResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*response.GetTestbedsResourceResponse)
		}
	}

	var r1 *errors.StatusError
	if rf, ok := ret.Get(1).(func(utils.Context, request.GetTestbedsResourceRequest) *errors.StatusError); ok {
		r1 = rf(ctx, req)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(*errors.StatusError)
		}
	}

	return r0, r1
}

type NewStatisticsServiceT interface {
	mock.TestingT
	Cleanup(func())
}

// NewStatisticsService creates a new instance of StatisticsService. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
func NewStatisticsService(t NewStatisticsServiceT) *StatisticsService {
	mock := &StatisticsService{}
	mock.Mock.Test(t)

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}