From ea3ee51ba08dd9a25ec3daff468b80f609767246 Mon Sep 17 00:00:00 2001
From: Michael Johnston <lastobelus@mac.com>
Date: Sat, 22 Aug 2026 02:16:39 -0700
Subject: [PATCH] fix(lastcode): restore main test expectations

---
 apps/desktop/src/updates/DesktopUpdates.test.ts   | 15 +++++++++++----
 .../chat/MessagesTimeline.logic.test.ts           |  1 +
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/apps/desktop/src/updates/DesktopUpdates.test.ts b/apps/desktop/src/updates/DesktopUpdates.test.ts
index 537226891bbf..af765b798002 100644
--- a/apps/desktop/src/updates/DesktopUpdates.test.ts
+++ b/apps/desktop/src/updates/DesktopUpdates.test.ts
@@ -213,7 +213,9 @@ function makeHarness(options: UpdatesHarnessOptions = {}) {
   };
   const setUpdateChannelError = options.setUpdateChannelError;
   const settingsLayer =
-    setUpdateChannelError || options.beforeSetUpdateChannel
+    setUpdateChannelError ||
+    options.beforeSetUpdateChannel ||
+    options.localNightliesEnabled !== undefined
       ? Layer.succeed(DesktopAppSettings.DesktopAppSettings, {
           get: Effect.sync(() => testSettings),
           load: Effect.sync(() => testSettings),
@@ -1161,9 +1163,14 @@ describe("DesktopUpdates", () => {
       const installStarted = yield* Deferred.make<void>();
       const releaseInstall = yield* Deferred.make<void>();
       const harness = makeHarness({
-        stopBackend: Deferred.succeed(installStarted, undefined).pipe(
-          Effect.andThen(Deferred.await(releaseInstall)),
-        ),
+        backends: [
+          {
+            desiredRunning: true,
+            stop: Deferred.succeed(installStarted, undefined).pipe(
+              Effect.andThen(Deferred.await(releaseInstall)),
+            ),
+          },
+        ],
       });
 
       yield* Effect.scoped(
diff --git a/apps/web/src/components/chat/MessagesTimeline.logic.test.ts b/apps/web/src/components/chat/MessagesTimeline.logic.test.ts
index c7017406cde7..ab4c31c69ce0 100644
--- a/apps/web/src/components/chat/MessagesTimeline.logic.test.ts
+++ b/apps/web/src/components/chat/MessagesTimeline.logic.test.ts
@@ -294,6 +294,7 @@ describe("deriveMessagesTimelineRows", () => {
         kind: "working",
         id: "working-indicator-row",
         createdAt: "2026-01-01T00:01:00Z",
+        showThinking: true,
       },
     ]);
   });