Function TestCollector_Set_Add

coderd/prometheusmetrics/collector_test.go:82–117

(t *testing.T)

}

func TestCollector_Set_Add(t *testing.T) {
	t.Parallel()

	// given
	agentsGauge := prometheusmetrics.NewCachedGaugeVec(prometheus.NewGaugeVec(prometheus.GaugeOpts{
		Namespace: "coderd",
		Subsystem: "agents",
		Name:      "up",
		Help:      "The number of active agents per workspace.",
	}, []string{"username", "workspace_name"}))

	// when
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationAdd, 9, "first user", "my workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationAdd, 8, "second user", "your workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationAdd, 7, "first user", "my workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationAdd, 6, "second user", "your workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationSet, 5, "first user", "my workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationSet, 4, "second user", "your workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationAdd, 3, "first user", "my workspace")
	agentsGauge.WithLabelValues(prometheusmetrics.VectorOperationAdd, 2, "second user", "your workspace")
	agentsGauge.Commit()

	// then
	ch := make(chan prometheus.Metric, 2)
	agentsGauge.Collect(ch)

	metrics := collectAndSortMetrics(t, agentsGauge, 2)

	assert.Equal(t, "first user", metrics[0].Label[0].GetValue())   // Username
	assert.Equal(t, "my workspace", metrics[0].Label[1].GetValue()) // Workspace name
	assert.Equal(t, 8, int(metrics[0].Gauge.GetValue()))            // Metric value

	assert.Equal(t, "second user", metrics[1].Label[0].GetValue())    // Username
	assert.Equal(t, "your workspace", metrics[1].Label[1].GetValue()) // Workspace name
	assert.Equal(t, 6, int(metrics[1].Gauge.GetValue()))              // Metric value
}

func collectAndSortMetrics(t *testing.T, collector prometheus.Collector, count int) []*dto.Metric {
	ch := make(chan prometheus.Metric, count)