fix: add decorator to right place

Parkreiner · Parkreiner · commit 90f21462d49b · 2025-04-29T22:31:34.000-04:00
diff --git a/site/.storybook/preview.jsx b/site/.storybook/preview.jsx
@@ -32,11 +32,18 @@ import { QueryClient, QueryClientProvider, parseQueryArgs } from "react-query";
 import { withRouter } from "storybook-addon-remix-react-router";
 import "theme/globalFonts";
 import themes from "../src/theme";
+import { TimeSyncProvider } from "../src/hooks/useTimeSync";
 
 DecoratorHelpers.initializeThemeState(Object.keys(themes), "dark");
 
 /** @type {readonly Decorator[]} */
-export const decorators = [withRouter, withQuery, withHelmet, withTheme];
+export const decorators = [
+	withRouter,
+	withQuery,
+	withHelmet,
+	withTheme,
+	withTimeSyncProvider,
+];
 
 /** @type {Preview["parameters"]} */
 export const parameters = {
@@ -101,6 +108,20 @@ function withHelmet(Story) {
 	);
 }
 
+const storyDate = new Date("March 15, 2022");
+function withTimeSyncProvider(Story) {
+	return (
+		<TimeSyncProvider
+			options={{
+				initialDatetime: storyDate,
+				resyncOnNewSubscription: false,
+			}}
+		>
+			<Story />
+		</TimeSyncProvider>
+	);
+}
+
 /** @type {Decorator} */
 function withQuery(Story, { parameters }) {
 	const queryClient = new QueryClient({
diff --git a/site/src/pages/DeploymentSettingsPage/NotificationsPage/storybookUtils.ts b/site/src/pages/DeploymentSettingsPage/NotificationsPage/storybookUtils.ts
@@ -215,6 +215,5 @@ export const baseMeta = {
 		withAuthProvider,
 		withDashboardProvider,
 		withOrganizationSettingsProvider,
-		withTimeSyncProvider,
 	],
 } satisfies Meta<typeof NotificationsPage>;
diff --git a/site/src/testHelpers/storybook.tsx b/site/src/testHelpers/storybook.tsx
@@ -6,12 +6,11 @@ import { hasFirstUserKey, meKey } from "api/queries/users";
 import type { Entitlements } from "api/typesGenerated";
 import { GlobalSnackbar } from "components/GlobalSnackbar/GlobalSnackbar";
 import { AuthProvider } from "contexts/auth/AuthProvider";
-import { TimeSyncProvider } from "hooks/useTimeSync";
 import { DashboardContext } from "modules/dashboard/DashboardProvider";
 import { DeploymentConfigContext } from "modules/management/DeploymentConfigProvider";
 import { OrganizationSettingsContext } from "modules/management/OrganizationSettingsLayout";
 import { permissionChecks } from "modules/permissions";
-import type { FC, ReactElement } from "react";
+import type { FC } from "react";
 import { useQueryClient } from "react-query";
 import {
 	MockAppearanceConfig,
@@ -157,20 +156,6 @@ export const withGlobalSnackbar = (Story: FC) => (
 	</>
 );
 
-const storyDate = new Date("March 15, 2022");
-export const withTimeSyncProvider = (Story: FC): ReactElement => {
-	return (
-		<TimeSyncProvider
-			options={{
-				initialDatetime: storyDate,
-				resyncOnNewSubscription: false,
-			}}
-		>
-			<Story />
-		</TimeSyncProvider>
-	);
-};
-
 export const withOrganizationSettingsProvider = (Story: FC) => {
 	return (
 		<OrganizationSettingsContext.Provider
