methods.test.js

'use strict'

const chai = require('chai')
const Serverless = require('serverless/lib/serverless')
const AwsProvider = require('serverless/lib/plugins/aws/provider')
const ServerlessApigatewayServiceProxy = require('./../index')

chai.use(require('chai-as-promised'))
const expect = require('chai').expect

describe('#getAllServiceProxies()', () => {
  let serverless
  let serverlessApigatewayServiceProxy

  beforeEach(() => {
    serverless = new Serverless({ commands: [], options: {} })
    serverless.servicePath = true
    serverless.service.service = 'apigw-service-proxy'
    const options = {
      stage: 'dev',
      region: 'us-east-1'
    }
    serverless.setProvider('aws', new AwsProvider(serverless))
    serverlessApigatewayServiceProxy = new ServerlessApigatewayServiceProxy(serverless, options)
  })

  describe('#getMethodResponses()', () => {
    it('should return a corresponding methodResponses resource', async () => {
      expect(serverlessApigatewayServiceProxy.getMethodResponses()).to.have.nested.property(
        'Properties.MethodResponses'
      )
    })

    it('should set Access-Control-Allow-Origin header when cors is true', async () => {
      const json1 = serverlessApigatewayServiceProxy.getMethodResponses({
        cors: {
          origin: '*'
        }
      })

      expect(
        json1.Properties.MethodResponses[0].ResponseParameters[
          'method.response.header.Access-Control-Allow-Origin'
        ]
      ).to.equal(true)

      const json2 = serverlessApigatewayServiceProxy.getMethodResponses({
        cors: {
          origins: ['*', 'http://example.com']
        }
      })

      expect(
        json2.Properties.MethodResponses[0].ResponseParameters[
          'method.response.header.Access-Control-Allow-Origin'
        ]
      ).to.equal(true)
    })
  })
})