Refactor: Introduce an interface so weave client can be mocked
This commit is contained in:
1 parent
c3ce052cc5
commit
591be2d08f
1 file changed
+5
-1
@@ -58,9 +58,13 @@ func getKubePeers(c kubernetes.Interface, includeWithNoIPAddr bool) ([]nodeInfo,
|
||||
return addresses, nil
|
||||
}
|
||||
|
||||
type weaveClient interface {
|
||||
RmPeer(peerName string) (string, error)
|
||||
}
|
||||
|
||||
// For each of those peers that is no longer listed as a node by
|
||||
// Kubernetes, remove it from Weave IPAM
|
||||
func reclaimRemovedPeers(weave *weaveapi.Client, cml *configMapAnnotations, nodes []nodeInfo, myPeerName, myNodeName string) error {
|
||||
func reclaimRemovedPeers(weave weaveClient, cml *configMapAnnotations, nodes []nodeInfo, myPeerName, myNodeName string) error {
|
||||
for loopsWhenNothingChanged := 0; loopsWhenNothingChanged < 3; loopsWhenNothingChanged++ {
|
||||
if err := cml.Init(); err != nil {
|
||||
return err
|
||||
|
||||
Reference in new issue
Block a user