fix estop stories

Opentrons · Mar 7, 2024 · c3ba64e · c3ba64e
1 parent b02c741
commit c3ba64e
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 6 deletions.
diff --git a/app/src/organisms/EmergencyStop/DesktopEstopMissingModal.stories.tsx b/app/src/organisms/EmergencyStop/DesktopEstopMissingModal.stories.tsx
@@ -32,4 +32,7 @@ const Template: Story<
 export const EstopMissing = Template.bind({})
 EstopMissing.args = {
   robotName: 'Flexy',
+  closeModal: () => {},
+  isDismissedModal: false,
+  setIsDismissedModal: () => {},
 }
diff --git a/app/src/organisms/EmergencyStop/DesktopEstopPressedModal.stories.tsx b/app/src/organisms/EmergencyStop/DesktopEstopPressedModal.stories.tsx
@@ -1,6 +1,7 @@
 import * as React from 'react'
 import { Provider } from 'react-redux'
 import { createStore } from 'redux'
+import { QueryClient, QueryClientProvider } from 'react-query'
 
 import { configReducer } from '../../redux/config/reducer'
 import { EstopPressedModal } from '.'
@@ -20,16 +21,21 @@ const dummyConfig = {
 } as any
 
 const store: Store<any> = createStore(configReducer, dummyConfig)
+const queryClient = new QueryClient()
 
 const Template: Story<
   React.ComponentProps<typeof EstopPressedModal>
 > = args => (
-  <Provider store={store}>
-    <EstopPressedModal {...args} />
-  </Provider>
+  <QueryClientProvider client={queryClient}>
+    <Provider store={store}>
+      <EstopPressedModal {...args} />
+    </Provider>
+  </QueryClientProvider>
 )
 
 export const EstopPressed = Template.bind({})
 EstopPressed.args = {
   isEngaged: true,
+  closeModal: () => {},
+  setIsDismissedModal: () => {},
 }
diff --git a/app/src/organisms/EmergencyStop/TouchscreenEstopMissingModal.stories.tsx b/app/src/organisms/EmergencyStop/TouchscreenEstopMissingModal.stories.tsx
@@ -34,4 +34,7 @@ const Template: Story<
 export const EstopMissing = Template.bind({})
 EstopMissing.args = {
   robotName: 'Flexy',
+  closeModal: () => {},
+  isDismissedModal: false,
+  setIsDismissedModal: () => {},
 }
diff --git a/app/src/organisms/EmergencyStop/TouchscreenEstopPressedModal.stories.tsx b/app/src/organisms/EmergencyStop/TouchscreenEstopPressedModal.stories.tsx
@@ -1,6 +1,7 @@
 import * as React from 'react'
 import { Provider } from 'react-redux'
 import { createStore } from 'redux'
+import { QueryClient, QueryClientProvider } from 'react-query'
 
 import { touchScreenViewport } from '../../DesignTokens/constants'
 import { configReducer } from '../../redux/config/reducer'
@@ -22,13 +23,16 @@ const dummyConfig = {
 } as any
 
 const store: Store<any> = createStore(configReducer, dummyConfig)
+const queryClient = new QueryClient()
 
 const Template: Story<
   React.ComponentProps<typeof EstopPressedModal>
 > = args => (
-  <Provider store={store}>
-    <EstopPressedModal {...args} />
-  </Provider>
+  <QueryClientProvider client={queryClient}>
+    <Provider store={store}>
+      <EstopPressedModal {...args} />
+    </Provider>
+  </QueryClientProvider>
 )
 
 export const EstopPressed = Template.bind({})