add test for EditInstrumentModal

protocol-designer/src/organisms/EditInstrumentsModal/__tests__/EditInstrumentsModal.test.tsx

@@ -1,4 +1,4 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest'
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
 import { fireEvent, screen } from '@testing-library/react'
 
 import { FLEX_ROBOT_TYPE } from '@opentrons/shared-data'
@@ -7,15 +7,23 @@ import { renderWithProviders } from '../../../__testing-utils__'
 import { i18n } from '../../../assets/localization'
 
 import { getRobotType } from '../../../file-data/selectors'
+import {
+  getAdditionalEquipment,
+  getInitialDeckSetup,
+} from '../../../step-forms/selectors'
+import { getHas96Channel } from '../../../utils'
 import { usePipetteConfig } from '../usePipetteConfig'
 import { PipetteOverview } from '../PipetteOverview'
 import { PipetteConfiguration } from '../PipetteConfiguration'
 
 import { EditInstrumentsModal } from '..'
 
 import type { ComponentProps } from 'react'
+import { fi } from 'date-fns/locale'
 
 vi.mock('../../../file-data/selectors')
+vi.mock('../../../step-forms/selectors')
+vi.mock('../../../utils')
 vi.mock('../usePipetteConfig')
 vi.mock('../PipetteOverview')
 vi.mock('../PipetteConfiguration')
@@ -39,7 +47,7 @@ describe('EditInstrumentsModal', () => {
     vi.mocked(usePipetteConfig).mockReturnValue({
       page: 'add',
       mount: 'left',
-      pipetteType: '96',
+      pipetteType: 'single',
       pipetteGen: 'flex',
       pipetteVolume: 'p1000',
       selectedTips: ['A1'],
@@ -55,12 +63,90 @@ describe('EditInstrumentsModal', () => {
     vi.mocked(PipetteConfiguration).mockReturnValue(
       <div>mock PipetteConfiguration</div>
     )
+    vi.mocked(getInitialDeckSetup).mockReturnValue({
+      pipettes: {},
+      additionalEquipmentOnDeck: {},
+      modules: {},
+      labware: {},
+    })
+    vi.mocked(getHas96Channel).mockReturnValue(false)
+    vi.mocked(getAdditionalEquipment).mockReturnValue({})
+  })
+
+  afterEach(() => {
+    vi.resetAllMocks()
+  })
+
+  it('should render text and buttons - pipette configuration', () => {
+    render(props)
+    screen.getByText('Edit Pipette')
+    screen.getByText('mock PipetteConfiguration')
+    screen.getByText('Save')
+    screen.getByText('Back')
   })
 
-  it('should render text and buttons', () => {
+  it('should render text and buttons - pipette overview', () => {
+    vi.mocked(usePipetteConfig).mockReturnValue({
+      page: 'overview',
+      mount: 'left',
+      pipetteType: 'single',
+      pipetteGen: 'flex',
+      pipetteVolume: 'p1000',
+      selectedTips: ['A1'],
+      setPage: vi.fn(),
+      setMount: vi.fn(),
+      setPipetteType: vi.fn(),
+      setPipetteGen: vi.fn(),
+      setPipetteVolume: vi.fn(),
+      setSelectedTips: vi.fn(),
+      resetFields: vi.fn(),
+    })
     render(props)
     screen.getByText('Edit Instruments')
+    screen.getByText('mock PipetteOverview')
     screen.getByText('Save')
     screen.getByText('Cancel')
   })
+
+  it('should render text and buttons - pipette overview', () => {
+    vi.mocked(usePipetteConfig).mockReturnValue({
+      page: 'overview',
+      mount: 'left',
+      pipetteType: 'single',
+      pipetteGen: 'flex',
+      pipetteVolume: 'p1000',
+      selectedTips: ['A1'],
+      setPage: vi.fn(),
+      setMount: vi.fn(),
+      setPipetteType: vi.fn(),
+      setPipetteGen: vi.fn(),
+      setPipetteVolume: vi.fn(),
+      setSelectedTips: vi.fn(),
+      resetFields: vi.fn(),
+    })
+    render(props)
+    fireEvent.click(screen.getByText('Save'))
+    expect(mockOnClose).toHaveBeenCalled()
+  })
+
+  it('should render text and buttons - pipette overview', () => {
+    vi.mocked(usePipetteConfig).mockReturnValue({
+      page: 'overview',
+      mount: 'left',
+      pipetteType: 'single',
+      pipetteGen: 'flex',
+      pipetteVolume: 'p1000',
+      selectedTips: ['A1'],
+      setPage: vi.fn(),
+      setMount: vi.fn(),
+      setPipetteType: vi.fn(),
+      setPipetteGen: vi.fn(),
+      setPipetteVolume: vi.fn(),
+      setSelectedTips: vi.fn(),
+      resetFields: vi.fn(),
+    })
+    render(props)
+    fireEvent.click(screen.getByText('Cancel'))
+    expect(mockOnClose).toHaveBeenCalled()
+  })
 })

protocol-designer/src/organisms/EditInstrumentsModal/index.tsx

@@ -64,6 +64,8 @@ export function EditInstrumentsModal(
     resetFields,
   } = pipetteConfig
 
+  console.log('obj', JSON.stringify(pipettesOnDeck, null, 4))
+
   const selectedPipette =
     pipetteType === '96' || pipetteGen === 'GEN1'
       ? `${pipetteVolume}_${pipetteType}`