Skip to content

Commit 96731f4

Browse files
committed
Fix empty namespace and name
1 parent b61b28d commit 96731f4

File tree

4 files changed

+17
-13
lines changed

4 files changed

+17
-13
lines changed

ray-operator/controllers/ray/metrics/ray_cluster_metrics_test.go

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@ import (
1919
rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
2020
)
2121

22+
var defaultRayClusterMetricTTL = 10
23+
2224
func TestRayClusterInfo(t *testing.T) {
2325
tests := []struct {
2426
name string
@@ -62,7 +64,7 @@ func TestRayClusterInfo(t *testing.T) {
6264
objs[i] = &tc.clusters[i]
6365
}
6466
client := fake.NewClientBuilder().WithScheme(k8sScheme).WithObjects(objs...).Build()
65-
manager := NewRayClusterMetricsManager(context.Background(), client)
67+
manager := NewRayClusterMetricsManager(context.Background(), client, defaultRayClusterMetricTTL)
6668
reg := prometheus.NewRegistry()
6769
reg.MustRegister(manager)
6870

@@ -104,7 +106,7 @@ func TestScheduleRayClusterMetricForCleanup(t *testing.T) {
104106
require.NoError(t, rayv1.AddToScheme(k8sScheme))
105107

106108
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
107-
manager := NewRayClusterMetricsManager(ctx, client)
109+
manager := NewRayClusterMetricsManager(ctx, client, defaultRayClusterMetricTTL)
108110

109111
// Schedule a metric for cleanup
110112
manager.ScheduleRayClusterMetricForCleanup("test-cluster", "test-namespace")
@@ -123,7 +125,7 @@ func TestCleanupExpiredRayClusterMetrics(t *testing.T) {
123125
require.NoError(t, rayv1.AddToScheme(k8sScheme))
124126

125127
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
126-
manager := NewRayClusterMetricsManager(ctx, client)
128+
manager := NewRayClusterMetricsManager(ctx, client, defaultRayClusterMetricTTL)
127129

128130
// Set up a metric
129131
manager.ObserveRayClusterProvisionedDuration("expired-cluster", "test-namespace", 123.45)
@@ -166,7 +168,7 @@ func TestRayClusterCleanupLoop(t *testing.T) {
166168
require.NoError(t, rayv1.AddToScheme(k8sScheme))
167169

168170
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
169-
manager := NewRayClusterMetricsManager(ctx, client)
171+
manager := NewRayClusterMetricsManager(ctx, client, defaultRayClusterMetricTTL)
170172

171173
// Set up a metric
172174
manager.ObserveRayClusterProvisionedDuration("test-cluster", "test-namespace", 123.45)
@@ -252,7 +254,7 @@ func TestRayClusterConditionProvisioned(t *testing.T) {
252254
objs[i] = &tc.clusters[i]
253255
}
254256
client := fake.NewClientBuilder().WithScheme(k8sScheme).WithObjects(objs...).Build()
255-
manager := NewRayClusterMetricsManager(context.Background(), client)
257+
manager := NewRayClusterMetricsManager(context.Background(), client, defaultRayClusterMetricTTL)
256258
reg := prometheus.NewRegistry()
257259
reg.MustRegister(manager)
258260

ray-operator/controllers/ray/metrics/ray_job_metrics_test.go

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ import (
2020
rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
2121
)
2222

23+
var defaultRayJobMetricTTL = 10
24+
2325
func TestMetricRayJobInfo(t *testing.T) {
2426
tests := []struct {
2527
name string
@@ -59,7 +61,7 @@ func TestMetricRayJobInfo(t *testing.T) {
5961
objs[i] = &tc.rayJobs[i]
6062
}
6163
client := fake.NewClientBuilder().WithScheme(k8sScheme).WithObjects(objs...).Build()
62-
manager := NewRayJobMetricsManager(context.Background(), client)
64+
manager := NewRayJobMetricsManager(context.Background(), client, defaultRayJobMetricTTL)
6365
reg := prometheus.NewRegistry()
6466
reg.MustRegister(manager)
6567

@@ -139,7 +141,7 @@ func TestMetricRayJobDeploymentStatus(t *testing.T) {
139141
objs[i] = &tc.rayJobs[i]
140142
}
141143
client := fake.NewClientBuilder().WithScheme(k8sScheme).WithObjects(objs...).Build()
142-
manager := NewRayJobMetricsManager(context.Background(), client)
144+
manager := NewRayJobMetricsManager(context.Background(), client, defaultRayJobMetricTTL)
143145
reg := prometheus.NewRegistry()
144146
reg.MustRegister(manager)
145147

@@ -196,7 +198,7 @@ func TestScheduleRayJobMetricForCleanup(t *testing.T) {
196198
require.NoError(t, rayv1.AddToScheme(k8sScheme))
197199
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
198200
// Create a metrics manager
199-
manager := NewRayJobMetricsManager(context.Background(), client)
201+
manager := NewRayJobMetricsManager(context.Background(), client, defaultRayJobMetricTTL)
200202

201203
// Schedule a cleanup for a job
202204
manager.ScheduleRayJobMetricForCleanup("test-job", "default")
@@ -218,7 +220,7 @@ func TestCleanupExpiredRayJobMetrics(t *testing.T) {
218220
k8sScheme := runtime.NewScheme()
219221
require.NoError(t, rayv1.AddToScheme(k8sScheme))
220222
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
221-
manager := NewRayJobMetricsManager(context.Background(), client)
223+
manager := NewRayJobMetricsManager(context.Background(), client, defaultRayJobMetricTTL)
222224

223225
// Register the manager with the registry
224226
registry.MustRegister(manager)
@@ -258,7 +260,7 @@ func TestRayJobCleanupLoop(t *testing.T) {
258260
k8sScheme := runtime.NewScheme()
259261
require.NoError(t, rayv1.AddToScheme(k8sScheme))
260262
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
261-
manager := NewRayJobMetricsManager(context.Background(), client)
263+
manager := NewRayJobMetricsManager(context.Background(), client, defaultRayJobMetricTTL)
262264

263265
// Start the cleanup loop
264266
ctx, cancel := context.WithCancel(context.Background())
@@ -292,7 +294,7 @@ func TestRayJobConditionProvisioned(t *testing.T) {
292294
k8sScheme := runtime.NewScheme()
293295
require.NoError(t, rayv1.AddToScheme(k8sScheme))
294296
client := fake.NewClientBuilder().WithScheme(k8sScheme).Build()
295-
manager := NewRayJobMetricsManager(context.Background(), client)
297+
manager := NewRayJobMetricsManager(context.Background(), client, defaultRayJobMetricTTL)
296298
registry.MustRegister(manager)
297299

298300
// Simulate a job becoming provisioned

ray-operator/controllers/ray/raycluster_controller.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ func (r *RayClusterReconciler) Reconcile(ctx context.Context, request ctrl.Reque
136136
// Clear all related expectations
137137
r.rayClusterScaleExpectation.Delete(instance.Name, instance.Namespace)
138138
if r.options.RayClusterMetricsManager != nil {
139-
r.options.RayClusterMetricsManager.ScheduleRayClusterMetricForCleanup(instance.Name, instance.Namespace)
139+
r.options.RayClusterMetricsManager.ScheduleRayClusterMetricForCleanup(request.Name, request.Namespace)
140140
}
141141
} else {
142142
logger.Error(err, "Read request instance error!")

ray-operator/controllers/ray/rayjob_controller.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ func (r *RayJobReconciler) Reconcile(ctx context.Context, request ctrl.Request)
9494
logger.Info("RayJob resource not found. Ignoring since object must be deleted")
9595
// Schedule metrics cleanup when job reaches terminal status
9696
if r.options.RayJobMetricsManager != nil {
97-
r.options.RayJobMetricsManager.ScheduleRayJobMetricForCleanup(rayJobInstance.Name, rayJobInstance.Namespace)
97+
r.options.RayJobMetricsManager.ScheduleRayJobMetricForCleanup(request.Name, request.Namespace)
9898
}
9999
return ctrl.Result{}, nil
100100
}

0 commit comments

Comments
 (0)