chore: test getSegment

Author: Mint de Wit

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

@@ -53,6 +53,14 @@ describe('Test blueprint api context', () => {
 			expect(mockActionService.getResolvedPieceInstances).toHaveBeenCalledWith('current')
 		})
 
+		test('getSegment', async () => {
+			const { context, mockActionService } = await getTestee()
+
+			await context.getSegment('current')
+			expect(mockActionService.getSegment).toHaveBeenCalledTimes(1)
+			expect(mockActionService.getSegment).toHaveBeenCalledWith('current')
+		})
+
 		test('findLastPieceOnLayer', async () => {
 			const { context, mockActionService } = await getTestee()
 

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

@@ -53,6 +53,14 @@ describe('Test blueprint api context', () => {
 			expect(mockActionService.getResolvedPieceInstances).toHaveBeenCalledWith('current')
 		})
 
+		test('getSegment', async () => {
+			const { context, mockActionService } = await getTestee()
+
+			await context.getSegment('current')
+			expect(mockActionService.getSegment).toHaveBeenCalledTimes(1)
+			expect(mockActionService.getSegment).toHaveBeenCalledWith('current')
+		})
+
 		test('findLastPieceOnLayer', async () => {
 			const { context, mockActionService } = await getTestee()
 

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

@@ -55,6 +55,14 @@ describe('Test blueprint api context', () => {
 			expect(mockActionService.getResolvedPieceInstances).toHaveBeenCalledWith('current')
 		})
 
+		test('getSegment', async () => {
+			const { context, mockActionService } = await getTestee()
+
+			await context.getSegment('current')
+			expect(mockActionService.getSegment).toHaveBeenCalledTimes(1)
+			expect(mockActionService.getSegment).toHaveBeenCalledWith('current')
+		})
+
 		test('findLastPieceOnLayer', async () => {
 			const { context, mockActionService } = await getTestee()