From 085f2f0631063b45132338a166ff8a5c89ffd7e6 Mon Sep 17 00:00:00 2001
From: Jonatan Witoszek <jonatanwitoszek@gmail.com>
Date: Wed, 11 Feb 2026 18:51:44 +0100
Subject: [PATCH 1/2] Add stories for new modals

This reverts commit d5726544b90da0e4a57ba8043213054c76ff0637.
---
 .../AssignModelDialog.stories.tsx             | 68 +++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 src/components/AssignModelDialog/AssignModelDialog.stories.tsx

diff --git a/src/components/AssignModelDialog/AssignModelDialog.stories.tsx b/src/components/AssignModelDialog/AssignModelDialog.stories.tsx
new file mode 100644
index 00000000000..978d97c2848
--- /dev/null
+++ b/src/components/AssignModelDialog/AssignModelDialog.stories.tsx
@@ -0,0 +1,68 @@
+import type { Meta, StoryObj } from "@storybook/react-vite";
+import { PageFactory } from "@storybookUtils/AssignDialogShared/factories";
+import { withMockedFilters } from "@storybookUtils/AssignDialogShared/storyDecorators";
+import { ComponentProps } from "react";
+import { fn } from "storybook/test";
+
+import AssignModelDialog from "./AssignModelDialog";
+
+type Props = ComponentProps<typeof AssignModelDialog>;
+
+const meta: Meta<typeof AssignModelDialog> = {
+  title: "Components/Dialogs/AssignModelDialog",
+  component: AssignModelDialog,
+  decorators: [withMockedFilters],
+  loaders: [async () => ({ pages: await PageFactory.buildList(8) })],
+  render: (args: Props, { loaded }: { loaded: { pages: Props["pages"] } }) => (
+    <AssignModelDialog {...args} pages={args.pages ?? loaded.pages} />
+  ),
+  argTypes: {
+    confirmButtonState: {
+      control: "inline-radio",
+      options: ["default", "loading", "success", "error"],
+    },
+    onClose: { table: { disable: true } },
+    onFetchMore: { table: { disable: true } },
+    onSubmit: { table: { disable: true } },
+    onFilterChange: { table: { disable: true } },
+    pages: { table: { disable: true } },
+    excludedFilters: { table: { disable: true } },
+    initialConstraints: { table: { disable: true } },
+    labels: { table: { disable: true } },
+  },
+  args: {
+    open: true,
+    loading: false,
+    hasMore: false,
+    confirmButtonState: "default",
+    onClose: fn(),
+    onFetchMore: fn(),
+    onSubmit: fn(),
+    onFilterChange: fn(),
+  },
+};
+
+export default meta;
+type Story = StoryObj<typeof AssignModelDialog>;
+
+export const Default: Story = {};
+
+export const SingleSelection: Story = {
+  args: { selectionMode: "single" },
+};
+
+export const Loading: Story = {
+  args: { loading: true, pages: [] },
+};
+
+export const Empty: Story = {
+  args: { pages: [] },
+};
+
+export const WithLockedPageType: Story = {
+  args: {
+    initialConstraints: {
+      pageTypes: [{ id: "page-type-1", name: "Blog Post" }],
+    },
+  },
+};

From 9ee357eaec43a6ba5ec59a5376124829fb838233 Mon Sep 17 00:00:00 2001
From: Jonatan Witoszek <jonatanwitoszek@gmail.com>
Date: Thu, 12 Feb 2026 16:19:08 +0100
Subject: [PATCH 2/2] fix ts error

---
 .../AssignModelDialog/AssignModelDialog.stories.tsx           | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/components/AssignModelDialog/AssignModelDialog.stories.tsx b/src/components/AssignModelDialog/AssignModelDialog.stories.tsx
index 978d97c2848..376ed5059a5 100644
--- a/src/components/AssignModelDialog/AssignModelDialog.stories.tsx
+++ b/src/components/AssignModelDialog/AssignModelDialog.stories.tsx
@@ -1,6 +1,6 @@
 import type { Meta, StoryObj } from "@storybook/react-vite";
+import { withFilterApolloMocks } from "@storybookUtils/AssignDialogShared/decorators";
 import { PageFactory } from "@storybookUtils/AssignDialogShared/factories";
-import { withMockedFilters } from "@storybookUtils/AssignDialogShared/storyDecorators";
 import { ComponentProps } from "react";
 import { fn } from "storybook/test";
 
@@ -11,7 +11,7 @@ type Props = ComponentProps<typeof AssignModelDialog>;
 const meta: Meta<typeof AssignModelDialog> = {
   title: "Components/Dialogs/AssignModelDialog",
   component: AssignModelDialog,
-  decorators: [withMockedFilters],
+  decorators: [withFilterApolloMocks],
   loaders: [async () => ({ pages: await PageFactory.buildList(8) })],
   render: (args: Props, { loaded }: { loaded: { pages: Props["pages"] } }) => (
     <AssignModelDialog {...args} pages={args.pages ?? loaded.pages} />