serverless/test/integration/aws/iot.test.js

'use strict';

const { expect } = require('chai');
const fixtures = require('../../fixtures/programmatic');

const { publishIotData } = require('../../utils/iot');
const { confirmCloudWatchLogs } = require('../../utils/misc');
const { deployService, removeService } = require('../../utils/integration');

describe('AWS - IoT Integration Test', function () {
  this.timeout(1000 * 60 * 100); // Involves time-taking deploys
  let iotTopic;
  let serviceDir;
  let stackName;

  before(async () => {
    const serviceData = await fixtures.setup('iot');
    ({ servicePath: serviceDir } = serviceData);
    const serviceName = serviceData.serviceConfig.service;
    iotTopic = `${serviceName}/test`;
    stackName = `${serviceName}-dev`;

    return deployService(serviceDir);
  });

  after(() => {
    // Topics are ephemeral and IoT endpoint is part of the account
    return removeService(serviceDir);
  });

  describe('Basic Setup', () => {
    it('should invoke on a topic message matching the rule', async () => {
      const functionName = 'iotBasic';
      const message = JSON.stringify({ message: 'Hello from IoT!' });

      // NOTE: This test may fail on fresh accounts where the IoT endpoint has not completed provisioning
      const events = await confirmCloudWatchLogs(
        `/aws/lambda/${stackName}-${functionName}`,
        async () => publishIotData(iotTopic, message),
        {
          checkIsComplete: (soFarEvents) => {
            const logs = soFarEvents.reduce((data, event) => data + event.message, '');
            return logs.includes(message);
          },
        }
      );
      const logs = events.reduce((data, event) => data + event.message, '');
      expect(logs).to.include(message);
    });
  });
});