Create test for scanExecution lambda

backend/env.yml

@@ -42,6 +42,7 @@ staging:
   PE_API_URL: ${ssm:/crossfeed/staging/PE_API_URL}
   REPORTS_BUCKET_NAME: cisa-crossfeed-staging-reports
   CLOUDWATCH_BUCKET_NAME: cisa-crossfeed-staging-cloudwatch
+  SQS_QUEUE_URL: { Ref: WorkerQueue }
 
 prod:
   DB_DIALECT: 'postgres'
@@ -78,6 +79,7 @@ prod:
   PE_API_URL: ${ssm:/crossfeed/staging/PE_API_URL}
   REPORTS_BUCKET_NAME: cisa-crossfeed-prod-reports
   CLOUDWATCH_BUCKET_NAME: cisa-crossfeed-prod-cloudwatch
+  SQS_QUEUE_URL: { Ref: WorkerQueue }
 
 dev-vpc:
   securityGroupIds:

backend/src/tasks/scanExecution.ts

@@ -12,23 +12,9 @@ export const handler: Handler = async (event) => {
 
     console.log(commandOptions);
 
-    // Get the ARN of the SQS queue from the event
-    const sqsQueueArn: string | undefined = sqsRecord.eventSourceARN;
-
-    if (!sqsQueueArn) {
-      throw new Error('SQS Queue ARN not found in event');
-    }
-
-    // Describe the SQS queue to get its URL
-    const sqsQueue = {
-      QueueUrl: sqsQueueArn // Use the ARN as the QueueUrl
-    };
-    const queueAttributesResponse = await sqs
-      .getQueueAttributes(sqsQueue)
-      .promise();
-    const sqsQueueUrl = queueAttributesResponse.Attributes?.QueueUrl;
+    const sqsQueueUrl = process.env.SQS_QUEUE_URL!;
     console.log(sqsQueueUrl);
-    if (!sqsQueueUrl) {
+    if (sqsQueueUrl) {
       throw new Error('SQS Queue URL not found');
     }
 

backend/src/tasks/test/scanExecution.test.ts

@@ -0,0 +1,39 @@
+import { handler } from '../scanExecution';
+import { SQSRecord } from 'aws-lambda';
+
+// Mock the AWS SDK methods using aws-sdk-mock
+jest.mock('aws-sdk', () => {
+  return {
+    ECS: jest.fn(() => ({
+      runTask: jest.fn().mockReturnThis(),
+      promise: jest.fn()
+    })),
+    SQS: jest.fn(() => ({
+      sendMessage: jest.fn().mockReturnThis(),
+      promise: jest.fn()
+    }))
+  };
+});
+
+describe('Scan Execution', () => {
+  it('should handle the event', async () => {
+    const event = {
+      Records: [
+        {
+          body: 'test command',
+          eventSourceARN: 'YourSQSQueueARN'
+        } as SQSRecord
+      ]
+    };
+
+    const context = {} as any;
+    const callback = () => void 0;
+    const result = await handler(event, context, callback);
+
+    // Add your assertions here
+    expect(result.statusCode).toEqual(200);
+    expect(result.body).toContain(
+      'Fargate task started and message sent to SQS queue'
+    );
+  });
+});