Function TestReregisterMultipleTargets

ais/tests/regression_test.go:452–549

(t *testing.T)

}

func TestReregisterMultipleTargets(t *testing.T) {
	tutils.CheckSkip(t, tutils.SkipTestArgs{Long: true})

	var (
		filesSentOrig = make(map[string]int64)
		filesRecvOrig = make(map[string]int64)
		bytesSentOrig = make(map[string]int64)
		bytesRecvOrig = make(map[string]int64)
		filesSent     int64
		filesRecv     int64
		bytesSent     int64
		bytesRecv     int64

		m = ioContext{
			t:   t,
			num: 10000,
		}
	)

	m.initWithCleanupAndSaveState()
	m.expectTargets(2)
	targetsToUnregister := m.originalTargetCount - 1

	// Step 0: Collect rebalance stats
	clusterStats := tutils.GetClusterStats(t, m.proxyURL)
	for targetID, targetStats := range clusterStats.Target {
		filesSentOrig[targetID] = tutils.GetNamedStatsVal(targetStats, stats.StreamsOutObjCount)
		filesRecvOrig[targetID] = tutils.GetNamedStatsVal(targetStats, stats.StreamsInObjCount)
		bytesSentOrig[targetID] = tutils.GetNamedStatsVal(targetStats, stats.StreamsOutObjSize)
		bytesRecvOrig[targetID] = tutils.GetNamedStatsVal(targetStats, stats.StreamsInObjSize)
	}

	// Step 1: Unregister multiple targets
	removed := make(map[string]*cluster.Snode, m.smap.CountActiveTargets()-1)
	defer func() {
		var rebID string
		for _, tgt := range removed {
			rebID = m.stopMaintenance(tgt)
		}
		if len(removed) != 0 && rebID != "" {
			tutils.WaitForRebalanceByID(t, m.originalTargetCount, baseParams, rebID)
		}
	}()

	targets := m.smap.Tmap.ActiveNodes()
	for i := 0; i < targetsToUnregister; i++ {
		tlog.Logf("Put %s in maintenance (no rebalance)\n", targets[i].StringEx())
		args := &apc.ActValRmNode{DaemonID: targets[i].ID(), SkipRebalance: true}
		_, err := api.StartMaintenance(baseParams, args)
		tassert.CheckFatal(t, err)
		removed[targets[i].ID()] = targets[i]
	}

	smap, err := tutils.WaitForClusterState(proxyURL, "remove targets",
		m.smap.Version, m.originalProxyCount, m.originalTargetCount-targetsToUnregister)
	tassert.CheckFatal(t, err)
	tlog.Logf("The cluster now has %d target(s)\n", smap.CountActiveTargets())