|
1 | 1 | import { captureLambdaHandler, logger } from '@aws-github-runner/aws-powertools-util'; |
2 | 2 | import { Context, SQSEvent, SQSRecord } from 'aws-lambda'; |
3 | 3 |
|
4 | | -import { addMiddleware, adjustPool, scaleDownHandler, scaleUpHandler, ssmHousekeeper, jobRetryCheck, cleanupOrgRunnersHandler } from './lambda'; |
| 4 | +import { |
| 5 | + addMiddleware, |
| 6 | + adjustPool, |
| 7 | + scaleDownHandler, |
| 8 | + scaleUpHandler, |
| 9 | + ssmHousekeeper, |
| 10 | + jobRetryCheck, |
| 11 | + cleanupOrgRunnersHandler, |
| 12 | +} from './lambda'; |
5 | 13 | import { adjust } from './pool/pool'; |
6 | 14 | import ScaleError from './scale-runners/ScaleError'; |
7 | 15 | import { scaleDown } from './scale-runners/scale-down'; |
@@ -62,32 +70,14 @@ const context: Context = { |
62 | 70 | }, |
63 | 71 | }; |
64 | 72 |
|
65 | | -<<<<<<< HEAD |
66 | 73 | vi.mock('./pool/pool'); |
67 | 74 | vi.mock('./scale-runners/scale-down'); |
68 | 75 | vi.mock('./scale-runners/scale-up'); |
69 | 76 | vi.mock('./scale-runners/ssm-housekeeper'); |
70 | 77 | vi.mock('./scale-runners/job-retry'); |
| 78 | +vi.mock('./scale-runners/cleanup-org-runners'); |
71 | 79 | vi.mock('@aws-github-runner/aws-powertools-util'); |
72 | 80 | vi.mock('@aws-github-runner/aws-ssm-util'); |
73 | | -||||||| parent of b4283485 (feat: add lambda function to cleanup org runners) |
74 | | -jest.mock('./pool/pool'); |
75 | | -jest.mock('./scale-runners/scale-down'); |
76 | | -jest.mock('./scale-runners/scale-up'); |
77 | | -jest.mock('./scale-runners/ssm-housekeeper'); |
78 | | -jest.mock('./scale-runners/job-retry'); |
79 | | -jest.mock('@aws-github-runner/aws-powertools-util'); |
80 | | -jest.mock('@aws-github-runner/aws-ssm-util'); |
81 | | -======= |
82 | | -jest.mock('./pool/pool'); |
83 | | -jest.mock('./scale-runners/scale-down'); |
84 | | -jest.mock('./scale-runners/scale-up'); |
85 | | -jest.mock('./scale-runners/ssm-housekeeper'); |
86 | | -jest.mock('./scale-runners/job-retry'); |
87 | | -jest.mock('./scale-runners/cleanup-org-runners'); |
88 | | -jest.mock('@aws-github-runner/aws-powertools-util'); |
89 | | -jest.mock('@aws-github-runner/aws-ssm-util'); |
90 | | ->>>>>>> b4283485 (feat: add lambda function to cleanup org runners) |
91 | 81 |
|
92 | 82 | describe('Test scale up lambda wrapper.', () => { |
93 | 83 | it('Do not handle multiple record sets.', async () => { |
|
0 commit comments