test: add chat-quick-bottons stories

apps/nextjs/src/components/AppComponents/Chat/chat-quick-buttons.stories.tsx

@@ -0,0 +1,133 @@
+import type { Meta, StoryObj } from "@storybook/react";
+
+import {
+  ChatContext,
+  type ChatContextProps,
+} from "@/components/ContextProviders/ChatProvider";
+import { lessonPlanTrackingContext } from "@/lib/analytics/lessonPlanTrackingContext";
+
+import ChatQuickButtons from "./chat-quick-buttons";
+
+const DummyMessage = {};
+
+const ChatDecorator: Story["decorators"] = (Story, { parameters }) => (
+  <ChatContext.Provider
+    value={
+      {
+        messages: [DummyMessage],
+        ...parameters.chatContext,
+      } as unknown as ChatContextProps
+    }
+  >
+    <Story />
+  </ChatContext.Provider>
+);
+
+const LessonPlanTrackingContextDecorator: Story["decorators"] = (Story) => (
+  <lessonPlanTrackingContext.Provider
+    value={{
+      onClickContinue: () => {},
+      onClickRetry: () => {},
+      onClickStartFromExample: () => {},
+      onClickStartFromFreeText: () => {},
+      onStreamFinished: () => {},
+      onSubmitText: () => {},
+    }}
+  >
+    <Story />
+  </lessonPlanTrackingContext.Provider>
+);
+
+const meta: Meta<typeof ChatQuickButtons> = {
+  title: "Components/Chat/ChatQuickButtons",
+  component: ChatQuickButtons,
+  tags: ["autodocs"],
+  decorators: [ChatDecorator, LessonPlanTrackingContextDecorator],
+};
+
+export default meta;
+type Story = StoryObj<typeof ChatQuickButtons>;
+
+export const Idle: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "Idle",
+    },
+  },
+};
+
+export const Loading: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "Loading",
+    },
+  },
+};
+
+export const LoadingWithoutMessages: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "Loading",
+      messages: [],
+    },
+  },
+};
+
+export const RequestMade: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "RequestMade",
+    },
+  },
+};
+
+export const StreamingLessonPlan: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "StreamingLessonPlan",
+    },
+  },
+};
+
+export const StreamingChatResponse: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "StreamingChatResponse",
+    },
+  },
+};
+
+export const Moderating: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      ailaStreamingStatus: "Moderating",
+    },
+  },
+};
+
+export const StreamingWithQueuedUserAction: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      queuedUserAction: "regenerate",
+      ailaStreamingStatus: "StreamingLessonPlan",
+    },
+  },
+};
+
+export const ModeratingWithQueuedUserAction: Story = {
+  args: {},
+  parameters: {
+    chatContext: {
+      queuedUserAction: "regenerate",
+      ailaStreamingStatus: "Moderating",
+    },
+  },
+};

apps/nextjs/src/lib/analytics/lessonPlanTrackingContext.tsx

@@ -30,7 +30,7 @@ type LessonPlanTrackingContext = {
   onClickStartFromFreeText: (text: string) => void;
 };
 
-const lessonPlanTrackingContext =
+export const lessonPlanTrackingContext =
   createContext<LessonPlanTrackingContext | null>(null);
 
 export type LessonPlanTrackingProviderProps = Readonly<{