serverless/test/utils/iot.js

'use strict';

const awsRequest = require('@serverless/test/aws-request');
const IotService = require('aws-sdk').Iot;
const IotDataService = require('aws-sdk').IotData;

async function resolveIotEndpoint() {
  return awsRequest(IotService, 'describeEndpoint', { endpointType: 'iot:Data-ATS' }).then(
    (data) => {
      return data.endpointAddress;
    }
  );
}

async function publishIotData(topic, message) {
  return resolveIotEndpoint().then((endpoint) => {
    const params = {
      topic,
      payload: Buffer.from(message),
    };

    return awsRequest({ client: IotDataService, params: { endpoint } }, 'publish', params);
  });
}

module.exports = {
  resolveIotEndpoint,
  publishIotData,
};