test: Add test for getUpcomingParts

Author: rjmunro

packages/job-worker/src/blueprints/__tests__/context-OnTakeContext.test.ts

@@ -1,22 +1,24 @@
 /* eslint-disable @typescript-eslint/unbound-method */
 import { IBlueprintMutatablePart, IBlueprintPiece } from '@sofie-automation/blueprints-integration'
-import { PlayoutModel } from '../../playout/model/PlayoutModel.js'
 import { WatchedPackagesHelper } from '../context/watchedPackages.js'
 import { JobContext, ProcessedShowStyleCompound } from '../../jobs/index.js'
 import { mock } from 'jest-mock-extended'
 import { PartAndPieceInstanceActionService } from '../context/services/PartAndPieceInstanceActionService.js'
 import { OnTakeContext } from '../context/index.js'
+import { DBPart } from '@sofie-automation/corelib/dist/dataModel/Part'
+import { PlayoutModelImpl } from '../../playout/model/implementation/PlayoutModelImpl.js'
 
 describe('Test blueprint api context', () => {
 	async function getTestee() {
+		const mockPlayoutModel = mock<PlayoutModelImpl>()
 		const mockActionService = mock<PartAndPieceInstanceActionService>()
 		const context = new OnTakeContext(
 			{
 				name: 'fakeContext',
 				identifier: 'action',
 			},
 			mock<JobContext>(),
-			mock<PlayoutModel>(),
+			mockPlayoutModel,
 			mock<ProcessedShowStyleCompound>(),
 			mock<WatchedPackagesHelper>(),
 			mockActionService
@@ -25,6 +27,7 @@ describe('Test blueprint api context', () => {
 		return {
 			context,
 			mockActionService,
+			mockPlayoutModel,
 		}
 	}
 
@@ -99,6 +102,20 @@ describe('Test blueprint api context', () => {
 			expect(mockActionService.getPartForPreviousPiece).toHaveBeenCalledWith({ _id: 'pieceId' })
 		})
 
+		test('getUpcomingParts', async () => {
+			const { context, mockPlayoutModel } = await getTestee()
+
+			mockPlayoutModel.getAllOrderedParts.mockReturnValue(
+				mock([
+					{ _id: 'part1', title: 'Part 1', invalid: false, floated: false } as unknown as DBPart,
+					{ _id: 'part2', title: 'Part 2', invalid: false, floated: false } as unknown as DBPart,
+				])
+			)
+
+			const parts = await context.getUpcomingParts()
+			expect(parts.map((i) => i.title)).toEqual(['Part 1', 'Part 2'])
+		})
+
 		test('insertPiece', async () => {
 			const { context, mockActionService } = await getTestee()