|
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