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

package mock

import (
	context "context"

	bizerr "github.com/tidbcloud/pkgv2/bizerr"

	grpc "google.golang.org/grpc"

	grpcli "github.com/tidbcloud/dataflow-service-ng/pkg/grpc/grpcli"

	mock "github.com/stretchr/testify/mock"

	nextgenv1 "github.com/tidbcloud/tidb-management-service/api/spec/global/nextgen/v1"

	serverlessv1 "github.com/tidbcloud/tidb-management-service/api/spec/regional/serverless/v1"

	tidb_mgmt_servicev1 "github.com/tidbcloud/tidb-management-service/api/spec/tidb_mgmt_service/v1"
)

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

// DeleteClusterCustomStatus provides a mock function with given fields: ctx, in, opts
func (_m *MgmtSvcGrpcClient) DeleteClusterCustomStatus(ctx context.Context, in *tidb_mgmt_servicev1.DeleteClusterCustomStatusRequest, opts ...grpc.CallOption) bizerr.BizError {
	_va := make([]interface{}, len(opts))
	for _i := range opts {
		_va[_i] = opts[_i]
	}
	var _ca []interface{}
	_ca = append(_ca, ctx, in)
	_ca = append(_ca, _va...)
	ret := _m.Called(_ca...)

	if len(ret) == 0 {
		panic("no return value specified for DeleteClusterCustomStatus")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.DeleteClusterCustomStatusRequest, ...grpc.CallOption) bizerr.BizError); ok {
		r0 = rf(ctx, in, opts...)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// ExportClient provides a mock function with no fields
func (_m *MgmtSvcGrpcClient) ExportClient() grpcli.ExportGrpcClient {
	ret := _m.Called()

	if len(ret) == 0 {
		panic("no return value specified for ExportClient")
	}

	var r0 grpcli.ExportGrpcClient
	if rf, ok := ret.Get(0).(func() grpcli.ExportGrpcClient); ok {
		r0 = rf()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(grpcli.ExportGrpcClient)
		}
	}

	return r0
}

// GetClusterByID provides a mock function with given fields: ctx, clusterID
func (_m *MgmtSvcGrpcClient) GetClusterByID(ctx context.Context, clusterID uint64) (*tidb_mgmt_servicev1.Cluster, bizerr.BizError) {
	ret := _m.Called(ctx, clusterID)

	if len(ret) == 0 {
		panic("no return value specified for GetClusterByID")
	}

	var r0 *tidb_mgmt_servicev1.Cluster
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64) (*tidb_mgmt_servicev1.Cluster, bizerr.BizError)); ok {
		return rf(ctx, clusterID)
	}
	if rf, ok := ret.Get(0).(func(context.Context, uint64) *tidb_mgmt_servicev1.Cluster); ok {
		r0 = rf(ctx, clusterID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*tidb_mgmt_servicev1.Cluster)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, uint64) bizerr.BizError); ok {
		r1 = rf(ctx, clusterID)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// GetResourcePoolByID provides a mock function with given fields: ctx, resourcePoolID
func (_m *MgmtSvcGrpcClient) GetResourcePoolByID(ctx context.Context, resourcePoolID uint64) (*nextgenv1.ResourcePool, bizerr.BizError) {
	ret := _m.Called(ctx, resourcePoolID)

	if len(ret) == 0 {
		panic("no return value specified for GetResourcePoolByID")
	}

	var r0 *nextgenv1.ResourcePool
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64) (*nextgenv1.ResourcePool, bizerr.BizError)); ok {
		return rf(ctx, resourcePoolID)
	}
	if rf, ok := ret.Get(0).(func(context.Context, uint64) *nextgenv1.ResourcePool); ok {
		r0 = rf(ctx, resourcePoolID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*nextgenv1.ResourcePool)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, uint64) bizerr.BizError); ok {
		r1 = rf(ctx, resourcePoolID)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// GetResourcePoolIncludingDeleted provides a mock function with given fields: ctx, resourcePoolID
func (_m *MgmtSvcGrpcClient) GetResourcePoolIncludingDeleted(ctx context.Context, resourcePoolID uint64) (*nextgenv1.ResourcePool, bizerr.BizError) {
	ret := _m.Called(ctx, resourcePoolID)

	if len(ret) == 0 {
		panic("no return value specified for GetResourcePoolIncludingDeleted")
	}

	var r0 *nextgenv1.ResourcePool
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64) (*nextgenv1.ResourcePool, bizerr.BizError)); ok {
		return rf(ctx, resourcePoolID)
	}
	if rf, ok := ret.Get(0).(func(context.Context, uint64) *nextgenv1.ResourcePool); ok {
		r0 = rf(ctx, resourcePoolID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*nextgenv1.ResourcePool)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, uint64) bizerr.BizError); ok {
		r1 = rf(ctx, resourcePoolID)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// GetServerlessRegional provides a mock function with given fields: ctx, clusterId
func (_m *MgmtSvcGrpcClient) GetServerlessRegional(ctx context.Context, clusterId uint64) (*serverlessv1.ServerlessRegionalServiceClusterInfo, bizerr.BizError) {
	ret := _m.Called(ctx, clusterId)

	if len(ret) == 0 {
		panic("no return value specified for GetServerlessRegional")
	}

	var r0 *serverlessv1.ServerlessRegionalServiceClusterInfo
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64) (*serverlessv1.ServerlessRegionalServiceClusterInfo, bizerr.BizError)); ok {
		return rf(ctx, clusterId)
	}
	if rf, ok := ret.Get(0).(func(context.Context, uint64) *serverlessv1.ServerlessRegionalServiceClusterInfo); ok {
		r0 = rf(ctx, clusterId)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*serverlessv1.ServerlessRegionalServiceClusterInfo)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, uint64) bizerr.BizError); ok {
		r1 = rf(ctx, clusterId)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// IsClusterCustomStatusExisted provides a mock function with given fields: ctx, in, opts
func (_m *MgmtSvcGrpcClient) IsClusterCustomStatusExisted(ctx context.Context, in *tidb_mgmt_servicev1.IsClusterCustomStatusExistedRequest, opts ...grpc.CallOption) (*tidb_mgmt_servicev1.IsClusterCustomStatusExistedResponse, bizerr.BizError) {
	_va := make([]interface{}, len(opts))
	for _i := range opts {
		_va[_i] = opts[_i]
	}
	var _ca []interface{}
	_ca = append(_ca, ctx, in)
	_ca = append(_ca, _va...)
	ret := _m.Called(_ca...)

	if len(ret) == 0 {
		panic("no return value specified for IsClusterCustomStatusExisted")
	}

	var r0 *tidb_mgmt_servicev1.IsClusterCustomStatusExistedResponse
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.IsClusterCustomStatusExistedRequest, ...grpc.CallOption) (*tidb_mgmt_servicev1.IsClusterCustomStatusExistedResponse, bizerr.BizError)); ok {
		return rf(ctx, in, opts...)
	}
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.IsClusterCustomStatusExistedRequest, ...grpc.CallOption) *tidb_mgmt_servicev1.IsClusterCustomStatusExistedResponse); ok {
		r0 = rf(ctx, in, opts...)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*tidb_mgmt_servicev1.IsClusterCustomStatusExistedResponse)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, *tidb_mgmt_servicev1.IsClusterCustomStatusExistedRequest, ...grpc.CallOption) bizerr.BizError); ok {
		r1 = rf(ctx, in, opts...)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// ListClusters provides a mock function with given fields: ctx, in, opts
func (_m *MgmtSvcGrpcClient) ListClusters(ctx context.Context, in *tidb_mgmt_servicev1.ListClustersRequest, opts ...grpc.CallOption) (*tidb_mgmt_servicev1.ListClustersResponse, bizerr.BizError) {
	_va := make([]interface{}, len(opts))
	for _i := range opts {
		_va[_i] = opts[_i]
	}
	var _ca []interface{}
	_ca = append(_ca, ctx, in)
	_ca = append(_ca, _va...)
	ret := _m.Called(_ca...)

	if len(ret) == 0 {
		panic("no return value specified for ListClusters")
	}

	var r0 *tidb_mgmt_servicev1.ListClustersResponse
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.ListClustersRequest, ...grpc.CallOption) (*tidb_mgmt_servicev1.ListClustersResponse, bizerr.BizError)); ok {
		return rf(ctx, in, opts...)
	}
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.ListClustersRequest, ...grpc.CallOption) *tidb_mgmt_servicev1.ListClustersResponse); ok {
		r0 = rf(ctx, in, opts...)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*tidb_mgmt_servicev1.ListClustersResponse)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, *tidb_mgmt_servicev1.ListClustersRequest, ...grpc.CallOption) bizerr.BizError); ok {
		r1 = rf(ctx, in, opts...)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// RestoreDedicatedCluster provides a mock function with given fields: ctx, req
func (_m *MgmtSvcGrpcClient) RestoreDedicatedCluster(ctx context.Context, req *tidb_mgmt_servicev1.RestoreDedicatedClusterRequest) (*tidb_mgmt_servicev1.Cluster, bizerr.BizError) {
	ret := _m.Called(ctx, req)

	if len(ret) == 0 {
		panic("no return value specified for RestoreDedicatedCluster")
	}

	var r0 *tidb_mgmt_servicev1.Cluster
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.RestoreDedicatedClusterRequest) (*tidb_mgmt_servicev1.Cluster, bizerr.BizError)); ok {
		return rf(ctx, req)
	}
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.RestoreDedicatedClusterRequest) *tidb_mgmt_servicev1.Cluster); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*tidb_mgmt_servicev1.Cluster)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, *tidb_mgmt_servicev1.RestoreDedicatedClusterRequest) bizerr.BizError); ok {
		r1 = rf(ctx, req)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// SetClusterCustomStatus provides a mock function with given fields: ctx, in, opts
func (_m *MgmtSvcGrpcClient) SetClusterCustomStatus(ctx context.Context, in *tidb_mgmt_servicev1.SetClusterCustomStatusRequest, opts ...grpc.CallOption) (*tidb_mgmt_servicev1.ClusterCustomStatus, bizerr.BizError) {
	_va := make([]interface{}, len(opts))
	for _i := range opts {
		_va[_i] = opts[_i]
	}
	var _ca []interface{}
	_ca = append(_ca, ctx, in)
	_ca = append(_ca, _va...)
	ret := _m.Called(_ca...)

	if len(ret) == 0 {
		panic("no return value specified for SetClusterCustomStatus")
	}

	var r0 *tidb_mgmt_servicev1.ClusterCustomStatus
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.SetClusterCustomStatusRequest, ...grpc.CallOption) (*tidb_mgmt_servicev1.ClusterCustomStatus, bizerr.BizError)); ok {
		return rf(ctx, in, opts...)
	}
	if rf, ok := ret.Get(0).(func(context.Context, *tidb_mgmt_servicev1.SetClusterCustomStatusRequest, ...grpc.CallOption) *tidb_mgmt_servicev1.ClusterCustomStatus); ok {
		r0 = rf(ctx, in, opts...)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*tidb_mgmt_servicev1.ClusterCustomStatus)
		}
	}

	if rf, ok := ret.Get(1).(func(context.Context, *tidb_mgmt_servicev1.SetClusterCustomStatusRequest, ...grpc.CallOption) bizerr.BizError); ok {
		r1 = rf(ctx, in, opts...)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// UpdateServerlessRegionalAnnotations provides a mock function with given fields: ctx, clusterId, annotations
func (_m *MgmtSvcGrpcClient) UpdateServerlessRegionalAnnotations(ctx context.Context, clusterId uint64, annotations map[string]string) bizerr.BizError {
	ret := _m.Called(ctx, clusterId, annotations)

	if len(ret) == 0 {
		panic("no return value specified for UpdateServerlessRegionalAnnotations")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, map[string]string) bizerr.BizError); ok {
		r0 = rf(ctx, clusterId, annotations)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// NewMgmtSvcGrpcClient creates a new instance of MgmtSvcGrpcClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func NewMgmtSvcGrpcClient(t interface {
	mock.TestingT
	Cleanup(func())
}) *MgmtSvcGrpcClient {
	mock := &MgmtSvcGrpcClient{}
	mock.Mock.Test(t)

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

	return mock
}