From 31d439ee40c32f41f73c103bc1b643966acc9895 Mon Sep 17 00:00:00 2001
From: James Long <longster@gmail.com>
Date: Tue, 10 Mar 2026 16:01:05 -0400
Subject: [PATCH] Fix test

---
 ...leware.test.ts => workspace-router-middleware.test.ts} | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)
 rename packages/opencode/test/control-plane/{session-proxy-middleware.test.ts => workspace-router-middleware.test.ts} (95%)

diff --git a/packages/opencode/test/control-plane/session-proxy-middleware.test.ts b/packages/opencode/test/control-plane/workspace-router-middleware.test.ts
similarity index 95%
rename from packages/opencode/test/control-plane/session-proxy-middleware.test.ts
rename to packages/opencode/test/control-plane/workspace-router-middleware.test.ts
index c674d95ec7..67c7c2d1fb 100644
--- a/packages/opencode/test/control-plane/session-proxy-middleware.test.ts
+++ b/packages/opencode/test/control-plane/workspace-router-middleware.test.ts
@@ -98,12 +98,14 @@ async function setup(state: State) {
     id2,
     app,
     async request(input: RequestInfo | URL, init?: RequestInit) {
+      const headers = new Headers(init?.headers)
+      headers.set("x-opencode-workspace", state.workspace === "first" ? id1 : id2)
       return Instance.provide({
         directory: tmp.path,
         fn: async () =>
           WorkspaceContext.provide({
             workspaceID: state.workspace === "first" ? id1 : id2,
-            fn: () => app.request(input, init),
+            fn: () => app.request(input, { ...init, headers }),
           }),
       })
     },
@@ -120,7 +122,7 @@ describe("control-plane/session-proxy-middleware", () => {
     const ctx = await setup(state)
 
     ctx.app.post("/session/foo", (c) => c.text("local", 200))
-    const response = await ctx.request("http://workspace.test/session/foo?x=1", {
+    const response = await ctx.request("http://workspace.test/session", {
       method: "POST",
       body: JSON.stringify({ hello: "world" }),
       headers: {
@@ -133,7 +135,7 @@ describe("control-plane/session-proxy-middleware", () => {
     expect(state.calls).toEqual([
       {
         method: "POST",
-        url: "/session/foo?x=1",
+        url: "/session",
         body: '{"hello":"world"}',
       },
     ])