publish.test.js

const { CreateTableCommand, DeleteTableCommand, DescribeTableCommand } = require('@aws-sdk/client-dynamodb');
const { DynamoDBDocumentClient, GetCommand } = require('@aws-sdk/lib-dynamodb');
const { testSha, testDatetime, testMessage } = require('./testdata/env');
const { publishPayload, ddbClient } = require('./publish');

const TABLE_NAME = 'metadata';
const TABLE_ARN = `arn:aws:dynamodb:local:000000000000:table/${TABLE_NAME}`;
const REGION = 'local';
const PAYLOAD = {
  commit_branch: 'main',
  commit_datetime: testDatetime,
  commit_message: testMessage,
  commit_sha: testSha,
  repository: 'babbel/publish-build-metadata',
  slices: ['api', 'consumer'],
  ttl: 1000,
};

beforeEach(async () => {
  // setup dynamodb endpoint
  process.env.AWS_ENDPOINT_URL = 'http://localhost:8000';

  // setup inputs
  await createDynamoDBTable();
});


afterEach(async () => {
  await deleteDynamoDBTable();
});


test('content should be published', async () => {
  const publishResult = await publishPayload(TABLE_ARN, PAYLOAD);

  expect(publishResult['$metadata'].httpStatusCode).toEqual(200);

  const docClient = DynamoDBDocumentClient.from(client());
  const result = await docClient.send(new GetCommand(
    {
      TableName: TABLE_NAME,
      Key: {
        repository: PAYLOAD.repository,
        commit_sha: PAYLOAD.commit_sha,
      }
    }
  ));

  expect(result.Item).toEqual(PAYLOAD);
});


//
// Test Helpers
//
function client() {
  return ddbClient(REGION);
}

async function createDynamoDBTable() {
  const params = {
    TableName: TABLE_NAME,
    KeySchema: [
      {
        AttributeName: 'repository',
        KeyType: 'HASH',
      },
      {
        AttributeName: 'commit_sha',
        KeyType: 'RANGE',
      },
    ],
    AttributeDefinitions: [
      {
        AttributeName: 'repository',
        AttributeType: 'S',
      },
      {
        AttributeName: 'commit_sha',
        AttributeType: 'S',
      },
    ],
    ProvisionedThroughput: {
      ReadCapacityUnits: 1,
      WriteCapacityUnits: 1,
    },
    StreamSpecification: {
      StreamEnabled: false,
    },
  };

  try {
    await client().send(new CreateTableCommand(params));
  } catch(err) {
    if (err == "ResourceInUseException") {
      await deleteDynamoDBTable();
      await createDynamoDBTable();
    }
  }

  const describeParams = {
    TableName: TABLE_NAME
  };

  return await client().send(new DescribeTableCommand(describeParams));
}

async function deleteDynamoDBTable() {
  const params = {
    TableName: TABLE_NAME
  };

  return await client().send(new DeleteTableCommand(params));
}