coder
diff --git a/‎site/src/pages/TasksPage/TasksPage.stories.tsx
Lines changed: 26 additions & 19 deletions b/‎site/src/pages/TasksPage/TasksPage.stories.tsx
Lines changed: 26 additions & 19 deletions
@@ -17,6 +17,7 @@ import {
 	withProxyProvider,
 } from "testHelpers/storybook";
 import TasksPage, { data } from "./TasksPage";
+import { reactRouterParameters } from "storybook-addon-remix-react-router";
 
 const meta: Meta<typeof TasksPage> = {
 	title: "pages/TasksPage",
@@ -132,6 +133,23 @@ const newTaskData = {
 
 export const CreateTaskSuccessfully: Story = {
 	decorators: [withProxyProvider()],
+	parameters: {
+		reactRouter: reactRouterParameters({
+			location: {
+				path: "/tasks",
+			},
+			routing: [
+				{
+					path: "/tasks",
+					useStoryElement: true,
+				},
+				{
+					path: "/tasks/:ownerName/:workspaceName",
+					element: <h1>Task page</h1>,
+				},
+			],
+		}),
+	},
 	beforeEach: () => {
 		spyOn(data, "fetchAITemplates").mockResolvedValue([MockTemplate]);
 		spyOn(data, "fetchTasks")
@@ -150,10 +168,8 @@ export const CreateTaskSuccessfully: Story = {
 			await userEvent.click(submitButton);
 		});
 
-		await step("Verify task in the table", async () => {
-			await canvas.findByRole("row", {
-				name: new RegExp(newTaskData.prompt, "i"),
-			});
+		await step("Redirects to the task page", async () => {
+			await canvas.findByText(/task page/i);
 		});
 	},
 };
@@ -187,7 +203,7 @@ export const CreateTaskError: Story = {
 	},
 };
 
-export const WithExternalAuth: Story = {
+export const WithAuthenticatedExternalAuth: Story = {
 	decorators: [withProxyProvider()],
 	beforeEach: () => {
 		spyOn(data, "fetchTasks")
@@ -201,26 +217,17 @@ export const WithExternalAuth: Story = {
 	play: async ({ canvasElement, step }) => {
 		const canvas = within(canvasElement);
 
-		await step("Run task", async () => {
-			const prompt = await canvas.findByLabelText(/prompt/i);
-			await userEvent.type(prompt, newTaskData.prompt);
-			const submitButton = canvas.getByRole("button", { name: /run task/i });
-			await waitFor(() => expect(submitButton).toBeEnabled());
-			await userEvent.click(submitButton);
-		});
-
-		await step("Verify task in the table", async () => {
-			await canvas.findByRole("row", {
-				name: new RegExp(newTaskData.prompt, "i"),
-			});
-		});
-
 		await step("Does not render external auth", async () => {
 			expect(
 				canvas.queryByText(/external authentication/),
 			).not.toBeInTheDocument();
 		});
 	},
+	parameters: {
+		chromatic: {
+			disableSnapshot: true,
+		},
+	},
 };
 
 export const MissingExternalAuth: Story = {