mock_flux_meter.go

// Code generated by MockGen. DO NOT EDIT.
// Source: flux-meter.go

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

import (
	reflect "reflect"

	languagev1 "github.com/fluxninja/aperture/v2/api/gen/proto/go/aperture/policy/language/v1"
	iface "github.com/fluxninja/aperture/v2/pkg/policies/flowcontrol/iface"
	gomock "github.com/golang/mock/gomock"
	prometheus "github.com/prometheus/client_golang/prometheus"
)

// MockFluxMeter is a mock of FluxMeter interface.
type MockFluxMeter struct {
	ctrl     *gomock.Controller
	recorder *MockFluxMeterMockRecorder
}

// MockFluxMeterMockRecorder is the mock recorder for MockFluxMeter.
type MockFluxMeterMockRecorder struct {
	mock *MockFluxMeter
}

// NewMockFluxMeter creates a new mock instance.
func NewMockFluxMeter(ctrl *gomock.Controller) *MockFluxMeter {
	mock := &MockFluxMeter{ctrl: ctrl}
	mock.recorder = &MockFluxMeterMockRecorder{mock}
	return mock
}

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

// GetAttributeKey mocks base method.
func (m *MockFluxMeter) GetAttributeKey() string {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetAttributeKey")
	ret0, _ := ret[0].(string)
	return ret0
}

// GetAttributeKey indicates an expected call of GetAttributeKey.
func (mr *MockFluxMeterMockRecorder) GetAttributeKey() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAttributeKey", reflect.TypeOf((*MockFluxMeter)(nil).GetAttributeKey))
}

// GetFluxMeterID mocks base method.
func (m *MockFluxMeter) GetFluxMeterID() iface.FluxMeterID {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetFluxMeterID")
	ret0, _ := ret[0].(iface.FluxMeterID)
	return ret0
}

// GetFluxMeterID indicates an expected call of GetFluxMeterID.
func (mr *MockFluxMeterMockRecorder) GetFluxMeterID() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFluxMeterID", reflect.TypeOf((*MockFluxMeter)(nil).GetFluxMeterID))
}

// GetFluxMeterName mocks base method.
func (m *MockFluxMeter) GetFluxMeterName() string {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetFluxMeterName")
	ret0, _ := ret[0].(string)
	return ret0
}

// GetFluxMeterName indicates an expected call of GetFluxMeterName.
func (mr *MockFluxMeterMockRecorder) GetFluxMeterName() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFluxMeterName", reflect.TypeOf((*MockFluxMeter)(nil).GetFluxMeterName))
}

// GetHistogram mocks base method.
func (m *MockFluxMeter) GetHistogram(labels map[string]string) prometheus.Observer {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetHistogram", labels)
	ret0, _ := ret[0].(prometheus.Observer)
	return ret0
}

// GetHistogram indicates an expected call of GetHistogram.
func (mr *MockFluxMeterMockRecorder) GetHistogram(labels interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHistogram", reflect.TypeOf((*MockFluxMeter)(nil).GetHistogram), labels)
}

// GetInvalidFluxMeterTotal mocks base method.
func (m *MockFluxMeter) GetInvalidFluxMeterTotal(labels map[string]string) (prometheus.Counter, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetInvalidFluxMeterTotal", labels)
	ret0, _ := ret[0].(prometheus.Counter)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetInvalidFluxMeterTotal indicates an expected call of GetInvalidFluxMeterTotal.
func (mr *MockFluxMeterMockRecorder) GetInvalidFluxMeterTotal(labels interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInvalidFluxMeterTotal", reflect.TypeOf((*MockFluxMeter)(nil).GetInvalidFluxMeterTotal), labels)
}

// GetSelectors mocks base method.
func (m *MockFluxMeter) GetSelectors() []*languagev1.Selector {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetSelectors")
	ret0, _ := ret[0].([]*languagev1.Selector)
	return ret0
}

// GetSelectors indicates an expected call of GetSelectors.
func (mr *MockFluxMeterMockRecorder) GetSelectors() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSelectors", reflect.TypeOf((*MockFluxMeter)(nil).GetSelectors))
}