Function TestCluster_AddRemovePeer

store/engine/raft/node_test.go:220–260

(t *testing.T)

}

func TestCluster_AddRemovePeer(t *testing.T) {
	cluster := NewTestCluster(3)
	defer cluster.Close()

	ctx := context.Background()
	require.Eventually(t, func() bool {
		return cluster.IsReady(ctx)
	}, 10*time.Second, 100*time.Millisecond)

	n1 := cluster.GetNode(0)
	require.NoError(t, n1.Set(ctx, "foo", []byte("bar")))
	require.Eventually(t, func() bool {
		got, _ := n1.Get(ctx, "foo")
		return string(got) == "bar"
	}, 1*time.Second, 100*time.Millisecond)

	t.Run("add a new peer node", func(t *testing.T) {
		n4, err := cluster.createNode(n1.config.Peers)
		require.NoError(t, err)
		require.NotNil(t, n4)

		require.NoError(t, cluster.AddNode(ctx, n4.config.ID, n4.Addr()))
		require.Eventually(t, func() bool {
			return n4.IsReady(ctx)
		}, 10*time.Second, 100*time.Millisecond)

		require.NoError(t, n4.Set(ctx, "foo", []byte("bar-1")))
		require.Eventually(t, func() bool {
			got, _ := n1.Get(ctx, "foo")
			return string(got) == "bar-1"
		}, 1*time.Second, 100*time.Millisecond)
		require.Len(t, n1.ListPeers(), 4)
	})

	t.Run("remove a peer node", func(t *testing.T) {
		cluster.RemoveNode(ctx, 4)
		require.Eventually(t, func() bool {
			return len(n1.ListPeers()) == 3
		}, 10*time.Second, 100*time.Millisecond)
	})
}

func TestTriggerSnapshot(t *testing.T) {
	cluster := NewTestCluster(3)