mirror of
https://github.com/kubernetes-sigs/node-feature-discovery.git
synced 2025-03-31 04:04:51 +00:00
81378a3235
Implement new registration infrastructure under the "source" package. This change loosens the coupling between label sources and the nfd-worker, making it easier to refactor and move the code around. Also, create a separate interface (ConfigurableSource) for configurable feature sources in order to eliminate boilerplate code. Add safety checks to the sources that they actually implement the interfaces they should. In sake of consistency and predictability (of behavior) change all methods of the sources to use pointer receivers. Add simple unit tests for the new functionality and include source/... into make test target.
63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
// Code generated by mockery v1.0.0. DO NOT EDIT.
|
|
|
|
// Re-generate by running 'make mock'
|
|
|
|
package source
|
|
|
|
import mock "github.com/stretchr/testify/mock"
|
|
|
|
// MockLabelSource is an autogenerated mock type for the LabelSource type
|
|
type MockLabelSource struct {
|
|
mock.Mock
|
|
}
|
|
|
|
// Discover provides a mock function with given fields:
|
|
func (_m *MockLabelSource) Discover() (FeatureLabels, error) {
|
|
ret := _m.Called()
|
|
|
|
var r0 FeatureLabels
|
|
if rf, ok := ret.Get(0).(func() FeatureLabels); ok {
|
|
r0 = rf()
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(FeatureLabels)
|
|
}
|
|
}
|
|
|
|
var r1 error
|
|
if rf, ok := ret.Get(1).(func() error); ok {
|
|
r1 = rf()
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// Name provides a mock function with given fields:
|
|
func (_m *MockLabelSource) Name() string {
|
|
ret := _m.Called()
|
|
|
|
var r0 string
|
|
if rf, ok := ret.Get(0).(func() string); ok {
|
|
r0 = rf()
|
|
} else {
|
|
r0 = ret.Get(0).(string)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// Priority provides a mock function with given fields:
|
|
func (_m *MockLabelSource) Priority() int {
|
|
ret := _m.Called()
|
|
|
|
var r0 int
|
|
if rf, ok := ret.Get(0).(func() int); ok {
|
|
r0 = rf()
|
|
} else {
|
|
r0 = ret.Get(0).(int)
|
|
}
|
|
|
|
return r0
|
|
}
|