replica_metrics_test.go

// Copyright 2020 The Cockroach Authors.
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0, included in the file
// licenses/APL.txt.

package kvserver

import (
	"testing"

	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/liveness"
	"github.com/cockroachdb/cockroach/pkg/roachpb"
	"github.com/cockroachdb/cockroach/pkg/util/leaktest"
	"github.com/cockroachdb/cockroach/pkg/util/log"
	"github.com/stretchr/testify/require"
)

func TestCalcRangeCounterIsLiveMap(t *testing.T) {
	defer leaktest.AfterTest(t)()
	defer log.Scope(t).Close(t)

	// Regression test for a bug, see:
	// https://github.com/cockroachdb/cockroach/pull/39936#pullrequestreview-359059629

	desc := roachpb.NewRangeDescriptor(123, roachpb.RKeyMin, roachpb.RKeyMax,
		roachpb.MakeReplicaSet([]roachpb.ReplicaDescriptor{
			{NodeID: 10, StoreID: 11, ReplicaID: 12, Type: roachpb.ReplicaTypeVoterFull()},
			{NodeID: 100, StoreID: 110, ReplicaID: 120, Type: roachpb.ReplicaTypeVoterFull()},
			{NodeID: 1000, StoreID: 1100, ReplicaID: 1200, Type: roachpb.ReplicaTypeVoterFull()},
			{NodeID: 2000, StoreID: 2100, ReplicaID: 2200, Type: roachpb.ReplicaTypeNonVoter()},
		}))

	{
		ctr, down, under, over := calcRangeCounter(1100, desc, liveness.IsLiveMap{
			1000: liveness.IsLiveMapEntry{IsLive: true}, // by NodeID
		}, 3 /* numVoters */, 4 /* numReplicas */, 4)

		require.True(t, ctr)
		require.True(t, down)
		require.True(t, under)
		require.False(t, over)
	}

	{
		ctr, down, under, over := calcRangeCounter(1000, desc, liveness.IsLiveMap{
			1000: liveness.IsLiveMapEntry{IsLive: false},
		}, 3 /* numVoters */, 4 /* numReplicas */, 4)

		// Does not confuse a non-live entry for a live one. In other words,
		// does not think that the liveness map has only entries for live nodes.
		require.False(t, ctr)
		require.False(t, down)
		require.False(t, under)
		require.False(t, over)
	}

	{
		ctr, down, under, over := calcRangeCounter(11, desc, liveness.IsLiveMap{
			10:   liveness.IsLiveMapEntry{IsLive: true},
			100:  liveness.IsLiveMapEntry{IsLive: true},
			1000: liveness.IsLiveMapEntry{IsLive: true},
			2000: liveness.IsLiveMapEntry{IsLive: true},
		}, 3 /* numVoters */, 4 /* numReplicas */, 4)

		require.True(t, ctr)
		require.False(t, down)
		require.False(t, under)
		require.False(t, over)
	}
}