fix: adjust restart disabled handling

Author: hosekpeter

internal/pkg/service/appsproxy/proxy/apphandler/upstream/upstream.go

@@ -197,6 +197,8 @@ func (u *AppUpstream) trace() chain.Middleware {
 				DNSDone: func(info httptrace.DNSDoneInfo) {
 					if info.Err != nil {
 						u.wakeup(ctx, info.Err)
+					} else {
+						u.restartDisabled.Store(false)
 					}
 				},
 			})

internal/pkg/service/appsproxy/proxy/pagewriter/restart_disabled.go

@@ -13,7 +13,7 @@ type RestartDisabledPageData struct {
 func (pw *Writer) WriteRestartDisabledPage(w http.ResponseWriter, req *http.Request, app api.AppConfig) {
 	w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate;")
 	w.Header().Set("pragma", "no-cache")
-	pw.writePage(w, req, "restart_disabled.gohtml", http.StatusNotFound, RestartDisabledPageData{
+	pw.writePage(w, req, "restart_disabled.gohtml", http.StatusServiceUnavailable, RestartDisabledPageData{
 		App: NewAppData(&app),
 	})
 }

internal/pkg/service/appsproxy/proxy/proxy_test.go

@@ -1842,7 +1842,7 @@ func TestAppProxyRouter(t *testing.T) {
 					require.NoError(t, err)
 					response, err := client.Do(request)
 					require.NoError(t, err)
-					if response.StatusCode == http.StatusNotFound {
+					if response.StatusCode == http.StatusServiceUnavailable {
 						body, err := io.ReadAll(response.Body)
 						require.NoError(t, err)
 						return strings.Contains(string(body), "Application Disabled")