serverless/test/utils/api-gateway.js

'use strict'

const _ = require('lodash')
const awsRequest = require('@serverless/test/aws-request')
const APIGatewayService = require('aws-sdk').APIGateway

async function createRestApi(name) {
  const params = {
    name,
  }

  return awsRequest(APIGatewayService, 'createRestApi', params)
}

async function deleteRestApi(restApiId) {
  const params = {
    restApiId,
  }

  return awsRequest(APIGatewayService, 'deleteRestApi', params)
}

async function getResources(restApiId) {
  const params = {
    restApiId,
  }

  return awsRequest(APIGatewayService, 'getResources', params).then(
    (data) => data.items,
  )
}

async function findRestApis(name) {
  const params = {
    limit: 500,
  }

  async function recursiveFind(found, position) {
    if (position) params.position = position
    return awsRequest(APIGatewayService, 'getRestApis', params).then(
      (result) => {
        const matches = result.items.filter((restApi) =>
          restApi.name.match(name),
        )
        if (matches.length) {
          _.merge(found, matches)
        }
        if (result.position) return recursiveFind(found, result.position)
        return found
      },
    )
  }

  return recursiveFind([])
}

module.exports = {
  createRestApi,
  deleteRestApi,
  getResources,
  findRestApis,
}