Skip to content

Commit 74c25b4

Browse files
committed
Improve UT coverage
1 parent 17eb4eb commit 74c25b4

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

pkg/metrics/scheduler_test.go

+16
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,22 @@ func TestSchedulerApplicationsFailed(t *testing.T) {
168168
verifyMetric(t, 1, "failed", "yunikorn_scheduler_application_total", dto.MetricType_GAUGE, "state")
169169
}
170170

171+
func TestSchedulingCycle(t *testing.T) {
172+
sm = getSchedulerMetrics(t)
173+
defer unregisterMetrics()
174+
175+
sm.ObserveSchedulingCycle(time.Now().Add(-1 * time.Minute))
176+
verifyHistogram(t, "scheduling_cycle_milliseconds", 60, 1)
177+
}
178+
179+
func TestTryNodeEvaluation(t *testing.T) {
180+
sm = getSchedulerMetrics(t)
181+
defer unregisterMetrics()
182+
183+
sm.ObserveTryNodeEvaluation(time.Now().Add(-1 * time.Minute))
184+
verifyHistogram(t, "trynode_evaluation_milliseconds", 60, 1)
185+
}
186+
171187
func getSchedulerMetrics(t *testing.T) *SchedulerMetrics {
172188
unregisterMetrics()
173189
return InitSchedulerMetrics()

0 commit comments

Comments
 (0)