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

package mock

import (
	context "context"

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

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

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

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

// GetCloudAccountID provides a mock function with given fields: ctx, req
func (_m *TiDBClusterInfraClient) GetCloudAccountID(ctx context.Context, req *grpcli.RegionalNetworkReq) (string, bizerr.BizError) {
	ret := _m.Called(ctx, req)

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

	var r0 string
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *grpcli.RegionalNetworkReq) (string, bizerr.BizError)); ok {
		return rf(ctx, req)
	}
	if rf, ok := ret.Get(0).(func(context.Context, *grpcli.RegionalNetworkReq) string); ok {
		r0 = rf(ctx, req)
	} else {
		r0 = ret.Get(0).(string)
	}

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

	return r0, r1
}

// GetClusterNodeRole provides a mock function with given fields: ctx, clusterID
func (_m *TiDBClusterInfraClient) GetClusterNodeRole(ctx context.Context, clusterID uint64) (string, bizerr.BizError) {
	ret := _m.Called(ctx, clusterID)

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

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

	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
}

// GetNetwork provides a mock function with given fields: ctx, req
func (_m *TiDBClusterInfraClient) GetNetwork(ctx context.Context, req *grpcli.RegionalNetworkReq) (*grpcli.RegionalNetwork, bizerr.BizError) {
	ret := _m.Called(ctx, req)

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

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

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

	return r0, r1
}

// GetNetworkNodeRole provides a mock function with given fields: ctx, req
func (_m *TiDBClusterInfraClient) GetNetworkNodeRole(ctx context.Context, req *grpcli.RegionalNetworkReq) (string, bizerr.BizError) {
	ret := _m.Called(ctx, req)

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

	var r0 string
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, *grpcli.RegionalNetworkReq) (string, bizerr.BizError)); ok {
		return rf(ctx, req)
	}
	if rf, ok := ret.Get(0).(func(context.Context, *grpcli.RegionalNetworkReq) string); ok {
		r0 = rf(ctx, req)
	} else {
		r0 = ret.Get(0).(string)
	}

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

	return r0, r1
}

// GetProviderVPCInfo provides a mock function with given fields: ctx, req
func (_m *TiDBClusterInfraClient) GetProviderVPCInfo(ctx context.Context, req *grpcli.ProviderVPCInfoReq) (*grpcli.ProviderVPCInfo, bizerr.BizError) {
	ret := _m.Called(ctx, req)

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

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

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

	return r0, r1
}

// NewTiDBClusterInfraClient creates a new instance of TiDBClusterInfraClient. 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 NewTiDBClusterInfraClient(t interface {
	mock.TestingT
	Cleanup(func())
}) *TiDBClusterInfraClient {
	mock := &TiDBClusterInfraClient{}
	mock.Mock.Test(t)

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

	return mock
}