mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-05 06:20:45 +01:00
28 lines
715 B
Go
28 lines
715 B
Go
|
package routemanager
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"github.com/netbirdio/netbird/route"
|
||
|
)
|
||
|
|
||
|
// MockManager is the mock instance of a route manager
|
||
|
type MockManager struct {
|
||
|
UpdateRoutesFunc func(updateSerial uint64, newRoutes []*route.Route) error
|
||
|
StopFunc func()
|
||
|
}
|
||
|
|
||
|
// UpdateRoutes mock implementation of UpdateRoutes from Manager interface
|
||
|
func (m *MockManager) UpdateRoutes(updateSerial uint64, newRoutes []*route.Route) error {
|
||
|
if m.UpdateRoutesFunc != nil {
|
||
|
return m.UpdateRoutesFunc(updateSerial, newRoutes)
|
||
|
}
|
||
|
return fmt.Errorf("method UpdateRoutes is not implemented")
|
||
|
}
|
||
|
|
||
|
// Stop mock implementation of Stop from Manager interface
|
||
|
func (m *MockManager) Stop() {
|
||
|
if m.StopFunc != nil {
|
||
|
m.StopFunc()
|
||
|
}
|
||
|
}
|