update test

Opentrons · Oct 8, 2024 · 168241f · 168241f
1 parent 64d4324
commit 168241f
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/protocol-designer/src/pages/Designer/ProtocolSteps/StepForm/StepTools/MagnetTools/index.tsx b/protocol-designer/src/pages/Designer/ProtocolSteps/StepForm/StepTools/MagnetTools/index.tsx
@@ -23,6 +23,7 @@ import {
 import {
   getMagnetLabwareEngageHeight,
   getMagneticLabwareOptions,
+  getSingleMagneticModuleId,
 } from '../../../../../../ui/modules/selectors'
 import { ToggleExpandStepFormField } from '../../../../../../molecules'
 import {
@@ -42,6 +43,8 @@ export function MagnetTools(props: StepFormProps): JSX.Element {
   const deckSetup = useSelector(getInitialDeckSetup)
   const modulesOnDeck = getModulesOnDeckByType(deckSetup, MAGNETIC_MODULE_TYPE)
 
+  console.log(modulesOnDeck)
+
   const moduleModel = moduleEntities[formData.moduleId].model
 
   const slotInfo = moduleLabwareOptions[0].name.split('in')

diff --git a/...signer/src/pages/Designer/ProtocolSteps/StepForm/StepTools/__tests__/MagnetTools.test.tsx b/...signer/src/pages/Designer/ProtocolSteps/StepForm/StepTools/__tests__/MagnetTools.test.tsx
@@ -6,7 +6,10 @@ import {
   getMagneticLabwareOptions,
   getMagnetLabwareEngageHeight,
 } from '../../../../../../ui/modules/selectors'
-import { getModuleEntities } from '../../../../../../step-forms/selectors'
+import {
+  getInitialDeckSetup,
+  getModuleEntities,
+} from '../../../../../../step-forms/selectors'
 import { MagnetTools } from '../MagnetTools'
 import type { ComponentProps } from 'react'
 import type * as ModulesSelectors from '../../../../../../ui/modules/selectors'
@@ -77,12 +80,26 @@ describe('MagnetTools', () => {
       },
     })
     vi.mocked(getMagnetLabwareEngageHeight).mockReturnValue(null)
+    vi.mocked(getInitialDeckSetup).mockReturnValue({
+      labware: {},
+      modules: {
+        module: {
+          id: 'mockId',
+          slot: '10',
+          type: 'magneticModuleType',
+          moduleState: { engaged: false, type: 'magneticModuleType' },
+          model: 'magneticModuleV1',
+        },
+      },
+      additionalEquipmentOnDeck: {},
+      pipettes: {},
+    })
   })
 
   it('renders the text and a switch button for v2', () => {
     render(props)
     screen.getByText('Module')
-    screen.getByText('abc')
+    screen.getByText('10')
     screen.getByText('mock labware')
     screen.getByText('mock module')
     screen.getByText('Magnet state')