fix(supervisor): allow native microVM restore subscriptions

Supervisor: allow compute restore subscriptions on the native Kubernetes microVM backend without a compute gateway URL.
Mono-RevId: e3ca49a622dd64b7aed290de0ea7528d423ace7b

Author: 1stvamp

apps/supervisor/src/env.test.ts

@@ -86,11 +86,11 @@ describe("worker queue selection", () => {
     {
       config: {
         KUBERNETES_FORCE_ENABLED: "true",
-        KUBERNETES_RUN_CRD_ENABLED: "true",
+        KUBERNETES_RUN_CRD_ENABLED: "false",
         KUBERNETES_RUNNER_RUNTIME: "microvm",
       },
       subscription: { ...restore, compat: "compute" },
-      error: "COMPUTE_GATEWAY_URL",
+      error: "compatibility",
     },
   ])(
     "rejects incompatible or unsupported subscriptions: $error",
@@ -105,6 +105,24 @@ describe("worker queue selection", () => {
     }
   );
 
+  it.each(["ondemand", "scheduled"])(
+    "accepts %s compute restores on the native microVM Runner backend without a gateway",
+    (queueClass) => {
+      expect(() =>
+        Env.parse({
+          ...base,
+          KUBERNETES_FORCE_ENABLED: "true",
+          KUBERNETES_RUN_CRD_ENABLED: "true",
+          KUBERNETES_RUNNER_RUNTIME: "microvm",
+          TRIGGER_WORKER_QUEUE_SUBSCRIPTIONS: JSON.stringify([
+            { ...ondemand, class: queueClass, compat: "compute" },
+            { ...restore, class: queueClass, compat: "compute" },
+          ]),
+        })
+      ).not.toThrow();
+    }
+  );
+
   it("accepts shared fresh work and compute restores on the compute backend", () => {
     expect(() =>
       Env.parse({

apps/supervisor/src/env.ts

@@ -378,13 +378,11 @@ export const Env = z
         });
       }
 
-      const compatibility =
-        data.COMPUTE_GATEWAY_URL ||
-        (data.KUBERNETES_FORCE_ENABLED &&
-          data.KUBERNETES_RUN_CRD_ENABLED &&
-          data.KUBERNETES_RUNNER_RUNTIME === "microvm")
-          ? "compute"
-          : "container";
+      const nativeMicrovm =
+        data.KUBERNETES_FORCE_ENABLED &&
+        data.KUBERNETES_RUN_CRD_ENABLED &&
+        data.KUBERNETES_RUNNER_RUNTIME === "microvm";
+      const compatibility = data.COMPUTE_GATEWAY_URL || nativeMicrovm ? "compute" : "container";
 
       for (const [index, subscription] of data.TRIGGER_WORKER_QUEUE_SUBSCRIPTIONS.entries()) {
         if (subscription.compat !== "any" && subscription.compat !== compatibility) {
@@ -395,7 +393,9 @@ export const Env = z
           });
         } else if (
           subscription.phase === "restore" &&
-          !(compatibility === "compute" ? data.COMPUTE_GATEWAY_URL : data.TRIGGER_CHECKPOINT_URL)
+          !(compatibility === "compute"
+            ? data.COMPUTE_GATEWAY_URL || nativeMicrovm
+            : data.TRIGGER_CHECKPOINT_URL)
         ) {
           ctx.addIssue({
             code: z.ZodIssueCode.custom,