Skip to content

Commit

Permalink
add test coverage to addAndSelectStep
Browse files Browse the repository at this point in the history
  • Loading branch information
@jerader
jerader committed Dec 17, 2024
1 parent f1ebb02 commit f1250ce
Showing 1 changed file with 264 additions and 0 deletions.
264 changes: 264 additions & 0 deletions protocol-designer/src/ui/steps/actions/__tests__/addAndSelectStep.test.ts
View file
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
import { describe, expect, it, vi, beforeEach } from 'vitest'
import { fixture12Trough, fixtureTiprack1000ul } from '@opentrons/shared-data'
import { addAndSelectStep } from '../thunks'
import { PRESAVED_STEP_ID } from '../../../../steplist/types'
import { addHint } from '../../../../tutorial/actions'
import { selectors as labwareIngredSelectors } from '../../../../labware-ingred/selectors'
import * as fileDataSelectors from '../../../../file-data/selectors'
import { getInitialDeckSetup } from '../../../../step-forms/selectors'
import type { StepType } from '../../../../form-types'
import type { LabwareDefinition2 } from '@opentrons/shared-data'

vi.mock('../../../../tutorial/actions')
vi.mock('../../../../ui/modules/selectors')
Expand Down Expand Up @@ -95,4 +97,266 @@ describe('addAndSelectStep', () => {
],
])
})
it('should dispatch a magnet module selected action if the step type is magnet', () => {
vi.mocked(getInitialDeckSetup).mockReturnValue({
modules: {
modId: {
type: 'magneticModuleType',
id: 'modId',
slot: '1',
model: 'magneticModuleV1',
moduleState: {} as any,
},
},
labware: {},
pipettes: {},
additionalEquipmentOnDeck: {},
})
const stepType: StepType = 'magnet'
const payload = {
stepType,
}
addAndSelectStep(payload)(dispatch, getState)
expect(dispatch.mock.calls).toEqual([
[
{
type: 'ADD_STEP',
payload: {
id: PRESAVED_STEP_ID,
stepType: 'magnet',
},
meta: {
robotStateTimeline: 'mockGetRobotStateTimelineValue',
},
},
],
[
{
type: 'SELECT_DROPDOWN_ITEM',
payload: {
selection: { id: 'modId', text: 'Selected', field: '1' },
mode: 'add',
},
},
],
])
})
it('should dispatch a temperature module selected action if the step type is temperature and only 1 temp mod', () => {
vi.mocked(getInitialDeckSetup).mockReturnValue({
modules: {
modId: {
type: 'temperatureModuleType',
id: 'modId',
slot: 'B2',
model: 'temperatureModuleV1',
moduleState: {} as any,
},
},
labware: {},
pipettes: {},
additionalEquipmentOnDeck: {},
})
const stepType: StepType = 'temperature'
const payload = {
stepType,
}
addAndSelectStep(payload)(dispatch, getState)
expect(dispatch.mock.calls).toEqual([
[
{
type: 'ADD_STEP',
payload: {
id: PRESAVED_STEP_ID,
stepType: 'temperature',
},
meta: {
robotStateTimeline: 'mockGetRobotStateTimelineValue',
},
},
],
[
{
type: 'SELECT_DROPDOWN_ITEM',
payload: {
selection: { id: 'modId', text: 'Selected', field: '1' },
mode: 'add',
},
},
],
])
})
it('should not dispatch hs module selected action if the step type is hs and 2 mods', () => {
vi.mocked(getInitialDeckSetup).mockReturnValue({
modules: {
modId: {
type: 'heaterShakerModuleType',
id: 'modId',
slot: 'B2',
model: 'heaterShakerModuleV1',
moduleState: {} as any,
},
modId2: {
type: 'heaterShakerModuleType',
id: 'modId2',
slot: 'A1',
model: 'heaterShakerModuleV1',
moduleState: {} as any,
},
},
labware: {},
pipettes: {},
additionalEquipmentOnDeck: {},
})
const stepType: StepType = 'heaterShaker'
const payload = {
stepType,
}
addAndSelectStep(payload)(dispatch, getState)
expect(dispatch.mock.calls).toEqual([
[
{
type: 'ADD_STEP',
payload: {
id: PRESAVED_STEP_ID,
stepType: 'heaterShaker',
},
meta: {
robotStateTimeline: 'mockGetRobotStateTimelineValue',
},
},
],
])
})
it('should dispatch labware selected action if the step type is mix and only 1 labware', () => {
vi.mocked(getInitialDeckSetup).mockReturnValue({
modules: {},
labware: {
labware: {
id: 'labware',
def: fixture12Trough as LabwareDefinition2,
labwareDefURI: 'mockDefUri',
slot: 'A1',
},
labware2: {
id: 'labware2',
def: fixtureTiprack1000ul as LabwareDefinition2,
labwareDefURI: 'mockDefUri',
slot: 'B1',
},
},
pipettes: {},
additionalEquipmentOnDeck: {},
})
const stepType: StepType = 'mix'
const payload = {
stepType,
}
addAndSelectStep(payload)(dispatch, getState)
expect(dispatch.mock.calls).toEqual([
[
{
type: 'ADD_STEP',
payload: {
id: PRESAVED_STEP_ID,
stepType: 'mix',
},
meta: {
robotStateTimeline: 'mockGetRobotStateTimelineValue',
},
},
],
[
{
type: 'SELECT_DROPDOWN_ITEM',
payload: {
selection: { id: 'labware', text: 'Selected', field: '1' },
mode: 'add',
},
},
],
])
})
it('should not dispatch labware selected action if the step type is moveLiquid and 2 labware', () => {
vi.mocked(getInitialDeckSetup).mockReturnValue({
modules: {},
labware: {
labware: {
id: 'labware',
def: fixture12Trough as LabwareDefinition2,
labwareDefURI: 'mockDefUri',
slot: 'A1',
},
labware2: {
id: 'labware2',
def: fixture12Trough as LabwareDefinition2,
labwareDefURI: 'mockDefUri',
slot: 'B1',
},
},
pipettes: {},
additionalEquipmentOnDeck: {},
})
const stepType: StepType = 'moveLiquid'
const payload = {
stepType,
}
addAndSelectStep(payload)(dispatch, getState)
expect(dispatch.mock.calls).toEqual([
[
{
type: 'ADD_STEP',
payload: {
id: PRESAVED_STEP_ID,
stepType: 'moveLiquid',
},
meta: {
robotStateTimeline: 'mockGetRobotStateTimelineValue',
},
},
],
])
})
it('should dispatch move labware selected action if the step type is moveLabware and only 1 labware', () => {
vi.mocked(getInitialDeckSetup).mockReturnValue({
modules: {},
labware: {
labware2: {
id: 'labware2',
def: fixtureTiprack1000ul as LabwareDefinition2,
labwareDefURI: 'mockDefUri',
slot: 'B1',
},
},
pipettes: {},
additionalEquipmentOnDeck: {},
})
const stepType: StepType = 'moveLabware'
const payload = {
stepType,
}
addAndSelectStep(payload)(dispatch, getState)
expect(dispatch.mock.calls).toEqual([
[
{
type: 'ADD_STEP',
payload: {
id: PRESAVED_STEP_ID,
stepType: 'moveLabware',
},
meta: {
robotStateTimeline: 'mockGetRobotStateTimelineValue',
},
},
],
[
{
type: 'SELECT_DROPDOWN_ITEM',
payload: {
selection: { id: 'labware2', text: 'Selected', field: '1' },
mode: 'add',
},
},
],
])
})
})

0 comments on commit f1250ce

Please sign in to comment.