// Code generated by MockGen. DO NOT EDIT.
// Source: ./internal/domain/facade/o11y.go
//
// Generated by this command:
//
//	mockgen -source=./internal/domain/facade/o11y.go -destination ./internal/domain/facade/mock_facade/mock_o11y.go -package mock_facade
//

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

import (
	context "context"
	reflect "reflect"
	time "time"

	constant "github.com/tidbcloud/serverless-service/internal/domain/constant"
	facade "github.com/tidbcloud/serverless-service/internal/domain/facade"
	tidb_mgmt_servicev1 "github.com/tidbcloud/tidb-management-service/api/spec/tidb_mgmt_service/v1"
	gomock "go.uber.org/mock/gomock"
)

// MockO11yFacade is a mock of O11yFacade interface.
type MockO11yFacade struct {
	ctrl     *gomock.Controller
	recorder *MockO11yFacadeMockRecorder
}

// MockO11yFacadeMockRecorder is the mock recorder for MockO11yFacade.
type MockO11yFacadeMockRecorder struct {
	mock *MockO11yFacade
}

// NewMockO11yFacade creates a new mock instance.
func NewMockO11yFacade(ctrl *gomock.Controller) *MockO11yFacade {
	mock := &MockO11yFacade{ctrl: ctrl}
	mock.recorder = &MockO11yFacadeMockRecorder{mock}
	return mock
}

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

// GetBranchesSkuUsagesByRegion mocks base method.
func (m *MockO11yFacade) GetBranchesSkuUsagesByRegion(ctx context.Context, keyspaceIds []uint32, metricType constant.MetricType, o11yAddr string) (map[string]float64, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetBranchesSkuUsagesByRegion", ctx, keyspaceIds, metricType, o11yAddr)
	ret0, _ := ret[0].(map[string]float64)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetBranchesSkuUsagesByRegion indicates an expected call of GetBranchesSkuUsagesByRegion.
func (mr *MockO11yFacadeMockRecorder) GetBranchesSkuUsagesByRegion(ctx, keyspaceIds, metricType, o11yAddr any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBranchesSkuUsagesByRegion", reflect.TypeOf((*MockO11yFacade)(nil).GetBranchesSkuUsagesByRegion), ctx, keyspaceIds, metricType, o11yAddr)
}

// GetClusterRowStorage mocks base method.
func (m *MockO11yFacade) GetClusterRowStorage(ctx context.Context, cluster *tidb_mgmt_servicev1.Cluster) (int64, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetClusterRowStorage", ctx, cluster)
	ret0, _ := ret[0].(int64)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetClusterRowStorage indicates an expected call of GetClusterRowStorage.
func (mr *MockO11yFacadeMockRecorder) GetClusterRowStorage(ctx, cluster any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClusterRowStorage", reflect.TypeOf((*MockO11yFacade)(nil).GetClusterRowStorage), ctx, cluster)
}

// GetClusterSkuUsageByRegion mocks base method.
func (m *MockO11yFacade) GetClusterSkuUsageByRegion(ctx context.Context, clusterId string, keyspaceId uint32, o11yAddr string) (map[string]float64, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetClusterSkuUsageByRegion", ctx, clusterId, keyspaceId, o11yAddr)
	ret0, _ := ret[0].(map[string]float64)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetClusterSkuUsageByRegion indicates an expected call of GetClusterSkuUsageByRegion.
func (mr *MockO11yFacadeMockRecorder) GetClusterSkuUsageByRegion(ctx, clusterId, keyspaceId, o11yAddr any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClusterSkuUsageByRegion", reflect.TypeOf((*MockO11yFacade)(nil).GetClusterSkuUsageByRegion), ctx, clusterId, keyspaceId, o11yAddr)
}

// GetDowntimes mocks base method.
func (m *MockO11yFacade) GetDowntimes(ctx context.Context, o11yAddr string) (map[string]facade.DowntimeRes, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetDowntimes", ctx, o11yAddr)
	ret0, _ := ret[0].(map[string]facade.DowntimeRes)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetDowntimes indicates an expected call of GetDowntimes.
func (mr *MockO11yFacadeMockRecorder) GetDowntimes(ctx, o11yAddr any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDowntimes", reflect.TypeOf((*MockO11yFacade)(nil).GetDowntimes), ctx, o11yAddr)
}

// GetExportUsage mocks base method.
func (m *MockO11yFacade) GetExportUsage(ctx context.Context, exportID, o11yAddr string, completedAt time.Time, metricType constant.MetricType) (int64, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetExportUsage", ctx, exportID, o11yAddr, completedAt, metricType)
	ret0, _ := ret[0].(int64)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetExportUsage indicates an expected call of GetExportUsage.
func (mr *MockO11yFacadeMockRecorder) GetExportUsage(ctx, exportID, o11yAddr, completedAt, metricType any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExportUsage", reflect.TypeOf((*MockO11yFacade)(nil).GetExportUsage), ctx, exportID, o11yAddr, completedAt, metricType)
}

// GetProvisionedRCU mocks base method.
func (m *MockO11yFacade) GetProvisionedRCU(ctx context.Context, o11yAddr string) (map[uint64]uint64, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetProvisionedRCU", ctx, o11yAddr)
	ret0, _ := ret[0].(map[uint64]uint64)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetProvisionedRCU indicates an expected call of GetProvisionedRCU.
func (mr *MockO11yFacadeMockRecorder) GetProvisionedRCU(ctx, o11yAddr any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionedRCU", reflect.TypeOf((*MockO11yFacade)(nil).GetProvisionedRCU), ctx, o11yAddr)
}

// GetSQLiteDBFileFromS3 mocks base method.
func (m *MockO11yFacade) GetSQLiteDBFileFromS3(ctx context.Context, region, keySuffix string) (string, string, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetSQLiteDBFileFromS3", ctx, region, keySuffix)
	ret0, _ := ret[0].(string)
	ret1, _ := ret[1].(string)
	ret2, _ := ret[2].(error)
	return ret0, ret1, ret2
}

// GetSQLiteDBFileFromS3 indicates an expected call of GetSQLiteDBFileFromS3.
func (mr *MockO11yFacadeMockRecorder) GetSQLiteDBFileFromS3(ctx, region, keySuffix any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSQLiteDBFileFromS3", reflect.TypeOf((*MockO11yFacade)(nil).GetSQLiteDBFileFromS3), ctx, region, keySuffix)
}