From 7aef0ef6934cd0adc5552d9b97e60166e250a368 Mon Sep 17 00:00:00 2001
From: Adam Howard <91115+codeincontext@users.noreply.github.com>
Date: Wed, 8 Jan 2025 10:43:15 +0100
Subject: [PATCH] test: fix @/storybook imports from jest

---
 apps/nextjs/.storybook/chromatic.ts           |  2 +-
 apps/nextjs/jest.config.mjs                   |  1 +
 .../Chat/chat-message/index.stories.tsx       |  2 +-
 .../LessonPlanProgressDropdown.test.tsx       | 21 -------------------
 .../Onboarding/TermsContent.stories.tsx       |  2 +-
 5 files changed, 4 insertions(+), 24 deletions(-)

diff --git a/apps/nextjs/.storybook/chromatic.ts b/apps/nextjs/.storybook/chromatic.ts
index bc755b188..dc8bfec65 100644
--- a/apps/nextjs/.storybook/chromatic.ts
+++ b/apps/nextjs/.storybook/chromatic.ts
@@ -10,7 +10,7 @@ export function chromaticParams(modes: ChromaticModes[]) {
           mobile: { viewport: "mobile" },
         }),
         ...(modes.includes("mobile-wide") && {
-          mobile: { viewport: "mobile-wide" },
+          "mobile-wide": { viewport: "mobile-wide" },
         }),
         ...(modes.includes("desktop") && {
           desktop: { viewport: "desktop" },
diff --git a/apps/nextjs/jest.config.mjs b/apps/nextjs/jest.config.mjs
index 7898f8abb..a7fdb640d 100644
--- a/apps/nextjs/jest.config.mjs
+++ b/apps/nextjs/jest.config.mjs
@@ -27,6 +27,7 @@ const config = {
     "^(\\.{1,2}/.*)\\.js$": "$1",
     "^@/components/(.*)$": "<rootDir>/src/components/$1",
     "^@/assets/(.*)$": "<rootDir>/src/assets/$1",
+    "^@/storybook/(.*)$": "<rootDir>/.storybook/$1",
   },
   extensionsToTreatAsEsm: [".ts", ".tsx"],
   testMatch: ["**/*.test.ts", "**/*.test.tsx"],
diff --git a/apps/nextjs/src/components/AppComponents/Chat/chat-message/index.stories.tsx b/apps/nextjs/src/components/AppComponents/Chat/chat-message/index.stories.tsx
index 022e5f8bb..1341f5a59 100644
--- a/apps/nextjs/src/components/AppComponents/Chat/chat-message/index.stories.tsx
+++ b/apps/nextjs/src/components/AppComponents/Chat/chat-message/index.stories.tsx
@@ -22,7 +22,7 @@ const meta = {
     ailaStreamingStatus: "Idle",
   },
   parameters: {
-    ...chromaticParams(["mobile", "desktop"]),
+    ...chromaticParams(["desktop"]),
   },
 } satisfies Meta<typeof ChatMessage>;
 
diff --git a/apps/nextjs/src/components/AppComponents/Chat/export-buttons/LessonPlanProgressDropdown.test.tsx b/apps/nextjs/src/components/AppComponents/Chat/export-buttons/LessonPlanProgressDropdown.test.tsx
index 6665475c5..2e3c3f78d 100644
--- a/apps/nextjs/src/components/AppComponents/Chat/export-buttons/LessonPlanProgressDropdown.test.tsx
+++ b/apps/nextjs/src/components/AppComponents/Chat/export-buttons/LessonPlanProgressDropdown.test.tsx
@@ -19,27 +19,6 @@ const { Default, PartiallyCompleted, FullyCompleted, PartialCycles } =
   composeStories(stories);
 
 describe("LessonPlanProgressDropdown", () => {
-  it("renders the Default story with correct closed state", () => {
-    render(<Default />);
-    expect(screen.getByText("4 of 10 sections complete")).toBeInTheDocument();
-    expect(screen.queryByText("Cycles")).not.toBeInTheDocument();
-  });
-
-  it("renders the PartiallyCompleted story with correct closed state", () => {
-    render(<PartiallyCompleted />);
-    expect(screen.getByText("7 of 10 sections complete")).toBeInTheDocument();
-  });
-
-  it("renders the FullyCompleted story with correct closed state", () => {
-    render(<FullyCompleted />);
-    expect(screen.getByText("10 of 10 sections complete")).toBeInTheDocument();
-  });
-
-  it("renders the PartialCycles story with correct closed state", () => {
-    render(<PartialCycles />);
-    expect(screen.getByText("5 of 10 sections complete")).toBeInTheDocument();
-  });
-
   it("displays the dropdown menu when clicked and shows correct completed sections", async () => {
     render(<PartiallyCompleted />);
 
diff --git a/apps/nextjs/src/components/Onboarding/TermsContent.stories.tsx b/apps/nextjs/src/components/Onboarding/TermsContent.stories.tsx
index 87b806da8..472d544d5 100644
--- a/apps/nextjs/src/components/Onboarding/TermsContent.stories.tsx
+++ b/apps/nextjs/src/components/Onboarding/TermsContent.stories.tsx
@@ -8,7 +8,7 @@ const meta = {
   title: "Components/Onboarding/TermsContent",
   component: TermsContent,
   parameters: {
-    ...chromaticParams(["mobile", "desktop"]),
+    ...chromaticParams(["desktop"]),
   },
 } satisfies Meta<typeof TermsContent>;