skuba 9.0.0 → 9.0.1-upgrade-cdk-template-20241002233314

package/template/lambda-sqs-worker-cdk/src/services/pipelineEventSender.ts

@@ -0,0 +1,33 @@
+import { PublishCommand } from '@aws-sdk/client-sns';
+
+import { config } from 'src/config';
+
+import { sns } from './aws';
+
+export const sendPipelineEvent = async (
+  event: unknown,
+  smokeTest: boolean = false,
+): Promise<string> => {
+  const snsResponse = await sns.send(
+    new PublishCommand({
+      Message: JSON.stringify(event),
+      ...(smokeTest && {
+        MessageAttributes: {
+          // Used for connectivity tests.
+          // Subscribers should filter out messages containing this attribute.
+          SmokeTest: {
+            DataType: 'String',
+            StringValue: 'true',
+          },
+        },
+      }),
+      TopicArn: config.destinationSnsTopicArn,
+    }),
+  );
+
+  if (snsResponse.MessageId === undefined) {
+    throw Error('SNS did not return a message ID');
+  }
+
+  return snsResponse.MessageId;
+};

package/template/lambda-sqs-worker-cdk/src/testing/handler.ts

@@ -0,0 +1,13 @@
+import type { Context, SQSEvent } from 'aws-lambda';
+
+import { chance } from './types';
+
+export const createCtx = () =>
+  ({
+    awsRequestId: chance.guid({ version: 4 }),
+  }) as Context;
+
+export const createSqsEvent = (bodies: string[]) =>
+  ({
+    Records: bodies.map((body) => ({ body })),
+  }) as SQSEvent;

package/template/lambda-sqs-worker-cdk/src/testing/logging.ts

@@ -0,0 +1,19 @@
+import * as logging from 'src/framework/logging';
+
+export const logger = {
+  error: jest.fn(),
+  info: jest.fn(),
+  debug: jest.fn(),
+
+  clear: () => {
+    logger.error.mockClear();
+    logger.info.mockClear();
+    logger.debug.mockClear();
+  },
+
+  spy: () => {
+    jest.spyOn(logging.logger, 'error').mockImplementation(logger.error);
+    jest.spyOn(logging.logger, 'info').mockImplementation(logger.info);
+    jest.spyOn(logging.logger, 'debug').mockImplementation(logger.debug);
+  },
+};

package/template/lambda-sqs-worker-cdk/src/testing/services.ts

@@ -0,0 +1,28 @@
+import 'aws-sdk-client-mock-jest';
+
+import { PublishCommand } from '@aws-sdk/client-sns';
+import { mockClient } from 'aws-sdk-client-mock';
+
+import { sns as snsClient } from 'src/services/aws';
+import * as jobScorer from 'src/services/jobScorer';
+
+export const scoringService = {
+  request: jest.fn(),
+
+  clear: () => scoringService.request.mockClear(),
+
+  spy: () =>
+    jest
+      .spyOn(jobScorer.scoringService, 'request')
+      .mockImplementation(scoringService.request),
+};
+
+const snsMock = mockClient(snsClient);
+
+export const sns = {
+  publish: snsMock.on(PublishCommand),
+
+  clear: () => snsMock.resetHistory(),
+
+  client: snsMock,
+};

package/template/lambda-sqs-worker-cdk/src/testing/types.ts

@@ -0,0 +1,33 @@
+import { Chance } from 'chance';
+import { z } from 'zod';
+
+import type { JobPublishedEvent } from 'src/types/pipelineEvents';
+
+export type IdDescription = z.infer<typeof IdDescriptionSchema>;
+
+export const IdDescriptionSchema = z.object({
+  id: z.string(),
+  description: z.string(),
+});
+
+export const chance = new Chance();
+
+export const mockIdDescription = (): IdDescription => ({
+  id: chance.guid({ version: 4 }),
+  description: chance.sentence(),
+});
+
+export const mockIdDescriptionJson = (): string =>
+  JSON.stringify(mockIdDescription());
+
+export const mockJobPublishedEvent = ({
+  entityId,
+}: {
+  entityId: string;
+}): JobPublishedEvent => ({
+  data: {
+    details: chance.paragraph(),
+  },
+  entityId,
+  eventType: 'JobPublished',
+});

package/template/lambda-sqs-worker-cdk/src/types/jobScorer.ts

@@ -0,0 +1,15 @@
+import { z } from 'zod';
+
+export type JobScorerInput = z.infer<typeof JobScorerInputSchema>;
+
+export const JobScorerInputSchema = z.object({
+  id: z.string(),
+  details: z.string(),
+});
+
+export type JobScorerOutput = z.infer<typeof JobScorerOutputSchema>;
+
+export const JobScorerOutputSchema = z.object({
+  id: z.string(),
+  score: z.number(),
+});

package/template/lambda-sqs-worker-cdk/src/types/pipelineEvents.ts

@@ -0,0 +1,21 @@
+import { z } from 'zod';
+
+export type JobPublishedEvent = z.infer<typeof JobPublishedEventSchema>;
+
+export const JobPublishedEventSchema = z.object({
+  data: z.object({
+    details: z.string(),
+  }),
+  entityId: z.string(),
+  eventType: z.literal('JobPublished'),
+});
+
+export type JobScoredEvent = z.infer<typeof JobScoredEventSchema>;
+
+export const JobScoredEventSchema = z.object({
+  data: z.object({
+    score: z.number(),
+  }),
+  entityId: z.string(),
+  eventType: z.literal('JobScored'),
+});