Function TestHATracker_MetricsCleanup

pkg/ha/ha_tracker_test.go:776–839

(t *testing.T)

}

func TestHATracker_MetricsCleanup(t *testing.T) {
	t.Parallel()
	reg := prometheus.NewPedanticRegistry()
	tr, err := NewHATracker(HATrackerConfig{EnableHATracker: false}, nil, haTrackerStatusConfig, prometheus.WrapRegistererWithPrefix("cortex_", reg), "test-ha-tracker", log.NewNopLogger())
	require.NoError(t, err)

	metrics := []string{
		"cortex_ha_tracker_elected_replica_changes_total",
		"cortex_ha_tracker_elected_replica_timestamp_seconds",
		"cortex_ha_tracker_kv_store_cas_total",
		"cortex_ha_tracker_user_replica_group_count",
	}

	tr.electedReplicaChanges.WithLabelValues("userA", "replicaGroup1").Add(5)
	tr.electedReplicaChanges.WithLabelValues("userA", "replicaGroup2").Add(8)
	tr.electedReplicaChanges.WithLabelValues("userB", "replicaGroup").Add(10)
	tr.electedReplicaTimestamp.WithLabelValues("userA", "replicaGroup1").Add(5)
	tr.electedReplicaTimestamp.WithLabelValues("userA", "replicaGroup2").Add(8)
	tr.electedReplicaTimestamp.WithLabelValues("userB", "replicaGroup").Add(10)
	tr.kvCASCalls.WithLabelValues("userA", "replicaGroup1").Add(5)
	tr.kvCASCalls.WithLabelValues("userA", "replicaGroup2").Add(8)
	tr.kvCASCalls.WithLabelValues("userB", "replicaGroup").Add(10)
	tr.userReplicaGroupCount.WithLabelValues("userA").Add(5)

	require.NoError(t, testutil.GatherAndCompare(reg, strings.NewReader(`
		# HELP cortex_ha_tracker_elected_replica_changes_total The total number of times the elected replica has changed for a user ID/cluster.
		# TYPE cortex_ha_tracker_elected_replica_changes_total counter
		cortex_ha_tracker_elected_replica_changes_total{cluster="replicaGroup",user="userB"} 10
		cortex_ha_tracker_elected_replica_changes_total{cluster="replicaGroup1",user="userA"} 5
		cortex_ha_tracker_elected_replica_changes_total{cluster="replicaGroup2",user="userA"} 8

		# HELP cortex_ha_tracker_elected_replica_timestamp_seconds The timestamp stored for the currently elected replica, from the KVStore.
		# TYPE cortex_ha_tracker_elected_replica_timestamp_seconds gauge
		cortex_ha_tracker_elected_replica_timestamp_seconds{cluster="replicaGroup",user="userB"} 10
		cortex_ha_tracker_elected_replica_timestamp_seconds{cluster="replicaGroup1",user="userA"} 5
		cortex_ha_tracker_elected_replica_timestamp_seconds{cluster="replicaGroup2",user="userA"} 8

		# HELP cortex_ha_tracker_kv_store_cas_total The total number of CAS calls to the KV store for a user ID/cluster.
		# TYPE cortex_ha_tracker_kv_store_cas_total counter
		cortex_ha_tracker_kv_store_cas_total{cluster="replicaGroup",user="userB"} 10
		cortex_ha_tracker_kv_store_cas_total{cluster="replicaGroup1",user="userA"} 5
		cortex_ha_tracker_kv_store_cas_total{cluster="replicaGroup2",user="userA"} 8
		
		# HELP cortex_ha_tracker_user_replica_group_count Number of HA replica groups tracked for each user.
		# TYPE cortex_ha_tracker_user_replica_group_count gauge
		cortex_ha_tracker_user_replica_group_count{user="userA"} 5
	`), metrics...))

	tr.CleanupHATrackerMetricsForUser("userA")

	require.NoError(t, testutil.GatherAndCompare(reg, strings.NewReader(`
		# HELP cortex_ha_tracker_elected_replica_changes_total The total number of times the elected replica has changed for a user ID/cluster.
		# TYPE cortex_ha_tracker_elected_replica_changes_total counter
		cortex_ha_tracker_elected_replica_changes_total{cluster="replicaGroup",user="userB"} 10

		# HELP cortex_ha_tracker_elected_replica_timestamp_seconds The timestamp stored for the currently elected replica, from the KVStore.
		# TYPE cortex_ha_tracker_elected_replica_timestamp_seconds gauge
		cortex_ha_tracker_elected_replica_timestamp_seconds{cluster="replicaGroup",user="userB"} 10