From 837a182f68856d1bafbcc9b8509a4c7313b38784 Mon Sep 17 00:00:00 2001
From: blam <ben@blam.sh>
Date: Sat, 27 Jun 2020 22:35:48 +0200
Subject: [PATCH] chore(scaffolder): finished off tests for the processor. It
 now works pretty well

---
 .../src/scaffolder/jobs/processor.test.ts     | 77 ++++++++++++++++++-
 1 file changed, 76 insertions(+), 1 deletion(-)

diff --git a/plugins/scaffolder-backend/src/scaffolder/jobs/processor.test.ts b/plugins/scaffolder-backend/src/scaffolder/jobs/processor.test.ts
index 6b80298926..501e72c078 100644
--- a/plugins/scaffolder-backend/src/scaffolder/jobs/processor.test.ts
+++ b/plugins/scaffolder-backend/src/scaffolder/jobs/processor.test.ts
@@ -67,7 +67,6 @@ describe('JobProcessor', () => {
         stages: [],
       });
 
-      console.warn(job.context);
       expect(job.context.entity).toBe(mockEntity);
       expect(job.context.values).toBe(mockValues);
     });
@@ -199,5 +198,81 @@ describe('JobProcessor', () => {
 
       expect(job.status).toBe('COMPLETED');
     });
+
+    it('should merge the return value from previous steps into the context of the next step', async () => {
+      const stages: StageInput[] = [
+        {
+          name: 'c/o',
+          handler: jest
+            .fn()
+            .mockResolvedValue({ first: 'ben', second: 'lambert' }),
+        },
+        {
+          name: 'g/p',
+          handler: jest
+            .fn()
+            .mockResolvedValue({ second: 'linus', third: 'lambert' }),
+        },
+        {
+          name: 'go',
+          handler: jest.fn(),
+        },
+      ];
+
+      const processor = new JobProcessor();
+      const job = processor.create({
+        entity: mockEntity,
+        values: mockValues,
+        stages,
+      });
+
+      await processor.run(job);
+
+      expect(stages[1].handler).toHaveBeenCalledWith(
+        expect.objectContaining({ first: 'ben', second: 'lambert' }),
+      );
+
+      expect(stages[2].handler).toHaveBeenCalledWith(
+        expect.objectContaining({
+          first: 'ben',
+          second: 'linus',
+          third: 'lambert',
+        }),
+      );
+    });
+
+    it('should fail the job and the step if one of them fails', async () => {
+      const fail = new Error('something went wrong here');
+      const stages: StageInput[] = [
+        {
+          name: 'c/o',
+          handler: jest.fn(),
+        },
+        {
+          name: 'g/p',
+          handler: jest.fn().mockRejectedValue(fail),
+        },
+        {
+          name: 'go',
+          handler: jest.fn(),
+        },
+      ];
+
+      const processor = new JobProcessor();
+      const job = processor.create({
+        entity: mockEntity,
+        values: mockValues,
+        stages,
+      });
+
+      await processor.run(job);
+
+      expect(job.status).toBe('FAILED');
+      expect(job.stages[0].status).toBe('COMPLETED');
+      expect(job.stages[1].status).toBe('FAILED');
+      expect(job.stages[2].status).toBe('PENDING');
+      expect(job.error?.message).toBe('something went wrong here');
+      expect(job.stages[1].log.join()).toContain('something went wrong here');
+    });
   });
 });