Function verifyCollectedMetrics

coderd/prometheusmetrics/aggregator_test.go:173–224

(t *testing.T, expected []*agentproto.Stats_Metric, actual []prometheus.Metric)

}

func verifyCollectedMetrics(t *testing.T, expected []*agentproto.Stats_Metric, actual []prometheus.Metric) bool {
	if len(expected) != len(actual) {
		t.Logf("expected %d metrics, got %d", len(expected), len(actual))
		return false
	}

	// ensure stable iteration order
	sort.Slice(expected, func(i, j int) bool {
		return expected[i].Name < expected[j].Name
	})

	sort.Slice(actual, func(i, j int) bool {
		m1 := prometheusMetricToString(t, actual[i])
		m2 := prometheusMetricToString(t, actual[j])
		return m1 < m2
	})

	for i, e := range expected {
		desc := actual[i].Desc()
		assert.Contains(t, desc.String(), e.Name)

		var d dto.Metric
		err := actual[i].Write(&d)
		assert.NoError(t, err)

		switch e.Type {
		case agentproto.Stats_Metric_COUNTER:
			if e.Value != d.Counter.GetValue() {
				return false
			}
		case agentproto.Stats_Metric_GAUGE:
			if e.Value != d.Gauge.GetValue() {
				return false
			}
		default:
			assert.Failf(t, "unsupported type: %s", string(e.Type))
		}

		expectedLabels := make([]*agentproto.Stats_Metric_Label, len(e.Labels))
		copy(expectedLabels, e.Labels)

		dtoLabels := asMetricAgentLabels(d.GetLabel())
		// dto labels are sorted in alphabetical order.
		sortFn := func(i, j int) bool {
			return expectedLabels[i].Name < expectedLabels[j].Name
		}
		sort.Slice(expectedLabels, sortFn)
		sort.Slice(dtoLabels, sortFn)
		assert.Equal(t, expectedLabels, dtoLabels, d.String())
	}
	return true
}

func prometheusMetricToString(t *testing.T, m prometheus.Metric) string {
	var sb strings.Builder