From 51d6271ed4f6b2efe174cfee2a58dbeb5ae1fd17 Mon Sep 17 00:00:00 2001
From: "molecule-ai[bot]" <276602405+molecule-ai[bot]@users.noreply.github.com>
Date: Tue, 21 Apr 2026 12:15:36 +0000
Subject: [PATCH] =?UTF-8?q?fix(tests):=20update=20orgTokenValidateQuery=20?=
 =?UTF-8?q?mock=20=E2=80=94=20Validate=20reads=203=20columns=20(#1366)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../internal/middleware/wsauth_middleware_test.go         | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/workspace-server/internal/middleware/wsauth_middleware_test.go b/workspace-server/internal/middleware/wsauth_middleware_test.go
index db58ab59..6ed046ee 100644
--- a/workspace-server/internal/middleware/wsauth_middleware_test.go
+++ b/workspace-server/internal/middleware/wsauth_middleware_test.go
@@ -473,8 +473,8 @@ func TestAdminAuth_InvalidBearer_Returns401(t *testing.T) {
 // token (org_id="ws-org-1").
 // ────────────────────────────────────────────────────────────────────────────
 
-// orgTokenValidateQuery is declared in wsauth_middleware_org_id_test.go
-// and reused here — same package, shared const, matched by sqlmock regex.
+// orgTokenValidateQuery is matched for orgtoken.Validate().
+const orgTokenValidateQuery = "SELECT id, prefix, org_id::text FROM org_api_tokens"
 
 // orgTokenOrgIDQuery is matched for the org_id lookup added in the F1097 fix.
 const orgTokenOrgIDQuery = "SELECT org_id::text FROM org_api_tokens"
@@ -525,8 +525,8 @@ func TestAdminAuth_OrgToken_SetsOrgID(t *testing.T) {
 			// (ValidateAnyToken), so ValidateAnyToken is NOT called here.
 			mock.ExpectQuery(orgTokenValidateQuery).
 				WithArgs(orgTokenHash[:]).
-				WillReturnRows(sqlmock.NewRows([]string{"id", "prefix"}).
-					AddRow("tok-org-1", "tok-org-1"))
+				WillReturnRows(sqlmock.NewRows([]string{"id", "prefix", "org_id"}).
+					AddRow("tok-org-1", "tok-org-1", nil))
 
 			// Best-effort last_used_at UPDATE (after Validate).
 			mock.ExpectExec(orgTokenLastUsedQuery).