@@ -101,13 +101,13 @@ func TestRequestQueueDrainHandler(t *testing.T) {
101101 if w .Code != http .StatusOK {
102102 t .Errorf ("Expected status 200, got %d" , w .Code )
103103 }
104-
104+
105105 // Verify the drainer is in draining state by sending a probe
106106 probeReq := httptest .NewRequest (http .MethodGet , "/" , nil )
107107 probeReq .Header .Set ("User-Agent" , "kube-probe/1.0" )
108108 probeW := httptest .NewRecorder ()
109109 drainer .ServeHTTP (probeW , probeReq )
110-
110+
111111 // Should return 503 because drainer is draining
112112 if probeW .Code != http .StatusServiceUnavailable {
113113 t .Errorf ("Expected probe to return 503 during drain, got %d" , probeW .Code )
@@ -144,7 +144,7 @@ func TestRequestQueueDrainHandler(t *testing.T) {
144144 probeReq .Header .Set ("User-Agent" , "kube-probe/1.0" )
145145 probeW := httptest .NewRecorder ()
146146 drainer .ServeHTTP (probeW , probeReq )
147-
147+
148148 if probeW .Code != http .StatusServiceUnavailable {
149149 t .Errorf ("Expected probe to return 503 during drain, got %d" , probeW .Code )
150150 }
@@ -157,7 +157,7 @@ func TestRequestQueueDrainHandler(t *testing.T) {
157157 probeReq2 .Header .Set ("User-Agent" , "kube-probe/1.0" )
158158 probeW2 := httptest .NewRecorder ()
159159 drainer .ServeHTTP (probeW2 , probeReq2 )
160-
160+
161161 // Should return 200 because drainer was reset
162162 if probeW2 .Code != http .StatusOK {
163163 t .Errorf ("Expected probe to return 200 after reset, got %d" , probeW2 .Code )
0 commit comments