Update test to call reclaimPeer()

This commit is contained in:
Bryan Boreham committed 2018-09-12 14:11:09 +00:00
1 parent 09be35dee0
commit 41c48dc678
1 file changed
+12 -8
+12 -8
View File
@@ -131,19 +131,17 @@ func TestPeerListFuzz(t *testing.T) {
// remove from Kubernetes first, then from test state
cml := newConfigMapAnnotations(configMapNamespace, configMapName, c)
err := cml.Init()
require.NoError(t, err)
require.NoError(t, cml.Init())
storedPeerList, err := cml.GetPeerList()
require.NoError(t, err)
found := storedPeerList.contains(peerName(i))
require.Equal(t, true, found, "peer %d not found in stored list", i)
if !found {
return
}
storedPeerList.remove(peerName(i))
err = cml.UpdatePeerList(*storedPeerList)
require.True(t, found, "peer %d not found in stored list", i)
_, err = reclaimPeer(mockWeave{}, cml, peerName(i), fmt.Sprintf("deleter-%d", i))
require.NoError(t, err)
storedPeerList, err = cml.GetPeerList()
require.NoError(t, err)
stateLock.Lock()
delete(nodeMap, i)
for pos = 0; pos < len(nodes); pos++ {
@@ -175,3 +173,9 @@ func TestPeerListFuzz(t *testing.T) {
require.NoError(t, err)
require.Equal(t, len(storedPeerList.Peers), len(nodes))
}
type mockWeave struct{}
func (mockWeave) RmPeer(peerName string) (string, error) {
return "", nil
}