54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
// Code generated by mockery v2.20.0. DO NOT EDIT.
|
||
|
|
||
|
package rate
|
||
|
|
||
|
import mock "github.com/stretchr/testify/mock"
|
||
|
|
||
|
// MockServersStatusProvider is an autogenerated mock type for the ServersStatusProvider type
|
||
|
type MockServersStatusProvider struct {
|
||
|
mock.Mock
|
||
|
}
|
||
|
|
||
|
// IsLeader provides a mock function with given fields:
|
||
|
func (_m *MockServersStatusProvider) IsLeader() bool {
|
||
|
ret := _m.Called()
|
||
|
|
||
|
var r0 bool
|
||
|
if rf, ok := ret.Get(0).(func() bool); ok {
|
||
|
r0 = rf()
|
||
|
} else {
|
||
|
r0 = ret.Get(0).(bool)
|
||
|
}
|
||
|
|
||
|
return r0
|
||
|
}
|
||
|
|
||
|
// IsServer provides a mock function with given fields: addr
|
||
|
func (_m *MockServersStatusProvider) IsServer(addr string) bool {
|
||
|
ret := _m.Called(addr)
|
||
|
|
||
|
var r0 bool
|
||
|
if rf, ok := ret.Get(0).(func(string) bool); ok {
|
||
|
r0 = rf(addr)
|
||
|
} else {
|
||
|
r0 = ret.Get(0).(bool)
|
||
|
}
|
||
|
|
||
|
return r0
|
||
|
}
|
||
|
|
||
|
type mockConstructorTestingTNewMockServersStatusProvider interface {
|
||
|
mock.TestingT
|
||
|
Cleanup(func())
|
||
|
}
|
||
|
|
||
|
// NewMockServersStatusProvider creates a new instance of MockServersStatusProvider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
||
|
func NewMockServersStatusProvider(t mockConstructorTestingTNewMockServersStatusProvider) *MockServersStatusProvider {
|
||
|
mock := &MockServersStatusProvider{}
|
||
|
mock.Mock.Test(t)
|
||
|
|
||
|
t.Cleanup(func() { mock.AssertExpectations(t) })
|
||
|
|
||
|
return mock
|
||
|
}
|