From f5058803b625b6a922ec10563bc2412db3b2847e Mon Sep 17 00:00:00 2001
From: Dmytro Stanchiev <git@dmytros.dev>
Date: Wed, 22 Apr 2026 17:37:54 -0400
Subject: [PATCH] test(ai-toolbar): harden stacked preview regression

---
 tests/ai-toolbar.test.ts | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/tests/ai-toolbar.test.ts b/tests/ai-toolbar.test.ts
index 583ac54..0aebabe 100644
--- a/tests/ai-toolbar.test.ts
+++ b/tests/ai-toolbar.test.ts
@@ -187,9 +187,18 @@ describe("AI capture redesign", () => {
 
 	test("attachment previews stack one per row instead of using a two-column desktop grid", () => {
 		const source = readToolbarSource();
+		const previewGridClassName = source.match(
+			/className="([^"]*\bmt-3\b[^"]*\bgrid\b[^"]*\bgap-2\b[^"]*)"/,
+		)?.[1];
 
-		expect(source).not.toContain('"mt-3 grid gap-2 grid-cols-2"');
 		expect(source).toContain('"mt-3 grid gap-2"');
+		expect(previewGridClassName).toBeDefined();
+		expect(previewGridClassName).toMatch(/\bmt-3\b/);
+		expect(previewGridClassName).toMatch(/\bgrid\b/);
+		expect(previewGridClassName).toMatch(/\bgap-2\b/);
+		expect(previewGridClassName).not.toMatch(
+			/(?:^|\s)(?:[a-z]+:)*grid-cols-(?:\[[^\]]+\]|\S+)/,
+		);
 	});
 });