Function TestReplicateManagerNodes

db/sg_replicate_cfg_test.go:89–142

Test node operations on SGReplicateManager

(t *testing.T)

// Test node operations on SGReplicateManager
func TestReplicateManagerNodes(t *testing.T) {

	ctx := base.TestCtx(t)
	testBucket := base.GetTestBucket(t)
	defer testBucket.Close(ctx)

	testCfg, err := base.NewCfgSG(ctx, testBucket.GetSingleDataStore(), "")
	require.NoError(t, err)

	manager, err := NewSGReplicateManager(ctx, &DatabaseContext{Name: "test"}, testCfg)
	require.NoError(t, err)
	defer manager.Stop()

	err = manager.registerNodeForHost("node1", "host1")
	require.NoError(t, err)

	nodes, err := manager.getNodes()
	require.NoError(t, err)
	assert.Len(t, nodes, 1)

	err = manager.registerNodeForHost("node2", "host2")
	require.NoError(t, err)

	nodes, err = manager.getNodes()
	require.NoError(t, err)
	assert.Len(t, nodes, 2)

	// re-adding an existing node is a no-op
	err = manager.registerNodeForHost("node1", "host1")
	require.NoError(t, err)

	nodes, err = manager.getNodes()
	require.NoError(t, err)
	assert.Len(t, nodes, 2)

	// Remove node
	err = manager.RemoveNode("node1")
	require.NoError(t, err)

	nodes, err = manager.getNodes()
	require.NoError(t, err)
	require.Len(t, nodes, 1)
	node2, ok := nodes["node2"]
	require.True(t, ok)
	require.Equal(t, node2.UUID, "node2")

	// Removing an already removed node is a no-op
	err = manager.RemoveNode("node1")
	require.NoError(t, err)

	replications, err := manager.GetReplications()
	require.NoError(t, err)
	assert.Len(t, replications, 0)
}

// Test concurrent node operations on SGReplicateManager
func TestReplicateManagerConcurrentNodeOperations(t *testing.T) {