@@ -20,6 +20,8 @@ import (
20
20
rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
21
21
)
22
22
23
+ var defaultRayJobMetricTTL = 10
24
+
23
25
func TestMetricRayJobInfo (t * testing.T ) {
24
26
tests := []struct {
25
27
name string
@@ -59,7 +61,7 @@ func TestMetricRayJobInfo(t *testing.T) {
59
61
objs [i ] = & tc .rayJobs [i ]
60
62
}
61
63
client := fake .NewClientBuilder ().WithScheme (k8sScheme ).WithObjects (objs ... ).Build ()
62
- manager := NewRayJobMetricsManager (context .Background (), client )
64
+ manager := NewRayJobMetricsManager (context .Background (), client , defaultRayJobMetricTTL )
63
65
reg := prometheus .NewRegistry ()
64
66
reg .MustRegister (manager )
65
67
@@ -139,7 +141,7 @@ func TestMetricRayJobDeploymentStatus(t *testing.T) {
139
141
objs [i ] = & tc .rayJobs [i ]
140
142
}
141
143
client := fake .NewClientBuilder ().WithScheme (k8sScheme ).WithObjects (objs ... ).Build ()
142
- manager := NewRayJobMetricsManager (context .Background (), client )
144
+ manager := NewRayJobMetricsManager (context .Background (), client , defaultRayJobMetricTTL )
143
145
reg := prometheus .NewRegistry ()
144
146
reg .MustRegister (manager )
145
147
@@ -196,7 +198,7 @@ func TestScheduleRayJobMetricForCleanup(t *testing.T) {
196
198
require .NoError (t , rayv1 .AddToScheme (k8sScheme ))
197
199
client := fake .NewClientBuilder ().WithScheme (k8sScheme ).Build ()
198
200
// Create a metrics manager
199
- manager := NewRayJobMetricsManager (context .Background (), client )
201
+ manager := NewRayJobMetricsManager (context .Background (), client , defaultRayJobMetricTTL )
200
202
201
203
// Schedule a cleanup for a job
202
204
manager .ScheduleRayJobMetricForCleanup ("test-job" , "default" )
@@ -218,7 +220,7 @@ func TestCleanupExpiredRayJobMetrics(t *testing.T) {
218
220
k8sScheme := runtime .NewScheme ()
219
221
require .NoError (t , rayv1 .AddToScheme (k8sScheme ))
220
222
client := fake .NewClientBuilder ().WithScheme (k8sScheme ).Build ()
221
- manager := NewRayJobMetricsManager (context .Background (), client )
223
+ manager := NewRayJobMetricsManager (context .Background (), client , defaultRayJobMetricTTL )
222
224
223
225
// Register the manager with the registry
224
226
registry .MustRegister (manager )
@@ -258,7 +260,7 @@ func TestRayJobCleanupLoop(t *testing.T) {
258
260
k8sScheme := runtime .NewScheme ()
259
261
require .NoError (t , rayv1 .AddToScheme (k8sScheme ))
260
262
client := fake .NewClientBuilder ().WithScheme (k8sScheme ).Build ()
261
- manager := NewRayJobMetricsManager (context .Background (), client )
263
+ manager := NewRayJobMetricsManager (context .Background (), client , defaultRayJobMetricTTL )
262
264
263
265
// Start the cleanup loop
264
266
ctx , cancel := context .WithCancel (context .Background ())
@@ -292,7 +294,7 @@ func TestRayJobConditionProvisioned(t *testing.T) {
292
294
k8sScheme := runtime .NewScheme ()
293
295
require .NoError (t , rayv1 .AddToScheme (k8sScheme ))
294
296
client := fake .NewClientBuilder ().WithScheme (k8sScheme ).Build ()
295
- manager := NewRayJobMetricsManager (context .Background (), client )
297
+ manager := NewRayJobMetricsManager (context .Background (), client , defaultRayJobMetricTTL )
296
298
registry .MustRegister (manager )
297
299
298
300
// Simulate a job becoming provisioned
0 commit comments