From b6c8314b0489e95a1047ecc0bb5fe983b11730bf Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Mon, 11 May 2026 07:18:23 +0100
Subject: [PATCH] test: tighten lmstudio provider assertions

---
 extensions/lmstudio/index.test.ts      | 11 +++++++----
 extensions/lmstudio/src/models.test.ts |  6 ++++--
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/extensions/lmstudio/index.test.ts b/extensions/lmstudio/index.test.ts
index 936d259dd7f..dd43e3d6bef 100644
--- a/extensions/lmstudio/index.test.ts
+++ b/extensions/lmstudio/index.test.ts
@@ -35,16 +35,19 @@ function createRemoteProviderConfig(overrides?: Partial<ModelProviderConfig>): M
 describe("lmstudio plugin", () => {
   it("canonicalizes base URLs during provider normalization", () => {
     const provider = registerProvider();
+    const providerConfig = createRemoteProviderConfig({
+      baseUrl: "http://localhost:1234/api/v1/",
+    });
 
     expect(
       provider?.normalizeConfig?.({
         provider: "lmstudio",
-        providerConfig: createRemoteProviderConfig({
-          baseUrl: "http://localhost:1234/api/v1/",
-        }),
+        providerConfig,
       }),
-    ).toMatchObject({
+    ).toEqual({
+      ...providerConfig,
       baseUrl: "http://localhost:1234/v1",
+      request: { allowPrivateNetwork: true },
     });
   });
 
diff --git a/extensions/lmstudio/src/models.test.ts b/extensions/lmstudio/src/models.test.ts
index 5c667bbf7fc..5e771467872 100644
--- a/extensions/lmstudio/src/models.test.ts
+++ b/extensions/lmstudio/src/models.test.ts
@@ -368,7 +368,10 @@ describe("lmstudio-models", () => {
     if (!loadCall) {
       throw new Error("expected LM Studio model load request");
     }
-    expect(loadCall[1]).toMatchObject({
+    const loadInit = loadCall[1] as RequestInit;
+    const { signal, ...stableLoadInit } = loadInit;
+    expect(signal).toBeInstanceOf(AbortSignal);
+    expect(stableLoadInit).toEqual({
       method: "POST",
       headers: {
         "X-Proxy-Auth": "required",
@@ -380,7 +383,6 @@ describe("lmstudio-models", () => {
         context_length: 32768,
       }),
     });
-    const loadInit = loadCall[1] as RequestInit;
     const loadBody = parseJsonRequestBody(loadInit) as { context_length: number };
     expect(loadBody.context_length).not.toBe(LMSTUDIO_DEFAULT_LOAD_CONTEXT_LENGTH);
   });