Merge pull request #7 from horike37/add-integration-test

Add integration tests

Author: horike37

.gitignore

@@ -42,3 +42,4 @@ tmp
 .coveralls.yml
 tmpdirs-serverless
 .eslintcache
+.serverless

.travis.yml

@@ -8,6 +8,9 @@ matrix:
       env:
         - DISABLE_TESTS=true
         - LINTING=true
+    - node_js: '10.6'
+      env:
+        - INTEGRATION_TESTS=true
 
 sudo: false
 
@@ -17,6 +20,7 @@ install:
 script:
   - if [[ -z "$DISABLE_TESTS" ]]; then npm run test; fi
   - if [[ ! -z "$DISABLE_TESTS" && ! -z "$LINTING" ]]; then npm run lint; fi
+  - if [[ ! -z "$INTEGRATION_TESTS" && $TRAVIS_BRANCH == "master" && $TRAVIS_PULL_REQUEST == "false" ]]; then npm run integration-test; fi
 
 after_success:
   - cat ./coverage/lcov.info | ./node_modules/coveralls/bin/coveralls.js && rm -rf ./coverage

__tests__/integration/kinesis/service/serverless.yml

@@ -0,0 +1,25 @@
+service: kinesis-proxy
+
+provider:
+  name: aws
+  runtime: nodejs10.x
+
+plugins:
+  localPath: './../'
+  modules:
+    - serverless-apigateway-service-proxy
+
+custom:
+  apiGatewayServiceProxies:
+    - kinesis:
+        path: /kinesis
+        method: post
+        streamName: { Ref: 'YourStream' }
+        cors: true
+
+resources:
+  Resources:
+    YourStream:
+      Type: AWS::Kinesis::Stream
+      Properties:
+        ShardCount: 1

__tests__/integration/kinesis/tests.js

@@ -0,0 +1,40 @@
+'use strict'
+
+const expect = require('chai').expect
+const fetch = require('node-fetch')
+const { deployService, removeService, getApiGatewayEndpoint } = require('./../../utils')
+
+describe('Kinesis Proxy Integration Test', () => {
+  let endpoint
+  let stackName
+  let stage
+  const config = '__tests__/integration/kinesis/service/serverless.yml'
+
+  beforeAll(async () => {
+    stage = Math.random()
+      .toString(32)
+      .substring(2)
+    stackName = 'kinesis-proxy-' + stage
+    deployService(stage, config)
+    endpoint = await getApiGatewayEndpoint(stackName)
+  })
+
+  afterAll(() => {
+    removeService(stage, config)
+  })
+
+  it('should get correct response from kinesis proxy endpoint', async () => {
+    const testEndpoint = `${endpoint}/kinesis`
+
+    const response = await fetch(testEndpoint, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ Data: 'some data', PartitionKey: 'some key' })
+    })
+    expect(response.headers.get('access-control-allow-origin')).to.deep.equal('*')
+    expect(response.status).to.be.equal(200)
+    const body = await response.json()
+    expect(body).to.have.own.property('ShardId')
+    expect(body).to.have.own.property('SequenceNumber')
+  })
+})

__tests__/integration/sqs/service/serverless.yml

@@ -0,0 +1,23 @@
+service: sqs-proxy
+
+provider:
+  name: aws
+  runtime: nodejs10.x
+
+plugins:
+  localPath: './../'
+  modules:
+    - serverless-apigateway-service-proxy
+
+custom:
+  apiGatewayServiceProxies:
+    - sqs:
+        path: /sqs
+        method: post
+        queueName: {"Fn::GetAtt":[ "SQSQueue", "QueueName" ]}
+        cors: true
+
+resources:
+  Resources:
+    SQSQueue:
+      Type: "AWS::SQS::Queue"

__tests__/integration/sqs/tests.js

@@ -0,0 +1,45 @@
+'use strict'
+
+const expect = require('chai').expect
+const fetch = require('node-fetch')
+const { deployService, removeService, getApiGatewayEndpoint } = require('./../../utils')
+
+describe('SQS Proxy Integration Test', () => {
+  let endpoint
+  let stackName
+  let stage
+  const config = '__tests__/integration/sqs/service/serverless.yml'
+
+  beforeAll(async () => {
+    stage = Math.random()
+      .toString(32)
+      .substring(2)
+    stackName = 'sqs-proxy-' + stage
+    deployService(stage, config)
+    endpoint = await getApiGatewayEndpoint(stackName)
+  })
+
+  afterAll(() => {
+    removeService(stage, config)
+  })
+
+  it('should get correct response from sqs proxy endpoint', async () => {
+    const testEndpoint = `${endpoint}/sqs`
+
+    const response = await fetch(testEndpoint, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ message: 'testtest' })
+    })
+    expect(response.headers.get('access-control-allow-origin')).to.deep.equal('*')
+    expect(response.status).to.be.equal(200)
+    const body = await response.json()
+    expect(body.SendMessageResponse.SendMessageResult).to.have.own.property(
+      'MD5OfMessageAttributes'
+    )
+    expect(body.SendMessageResponse.SendMessageResult).to.have.own.property('MD5OfMessageBody')
+    expect(body.SendMessageResponse.SendMessageResult).to.have.own.property('MessageId')
+    expect(body.SendMessageResponse.SendMessageResult).to.have.own.property('SequenceNumber')
+    expect(body.SendMessageResponse.ResponseMetadata).to.have.own.property('RequestId')
+  })
+})

__tests__/setup-tests.js

@@ -0,0 +1,3 @@
+'use strict'
+
+jest.setTimeout(300000)

__tests__/utils.js

@@ -0,0 +1,26 @@
+'use strict'
+
+const _ = require('lodash')
+const execSync = require('child_process').execSync
+const aws = require('aws-sdk')
+const cloudformation = new aws.CloudFormation({ region: 'us-east-1' })
+
+module.exports = {
+  async getApiGatewayEndpoint(stackName) {
+    const result = await cloudformation.describeStacks({ StackName: stackName }).promise()
+
+    const endpointOutput = _.find(result.Stacks[0].Outputs, { OutputKey: 'ServiceEndpoint' })
+      .OutputValue
+    return endpointOutput.match(/https:\/\/.+\.execute-api\..+\.amazonaws\.com.+/)[0]
+  },
+
+  deployService(stage, config) {
+    execSync(`npx serverless deploy --stage ${stage} --config ${config}`, {
+      stdio: 'inherit'
+    })
+  },
+
+  removeService(stage, config) {
+    execSync(`npx serverless remove --stage ${stage} --config ${config}`, { stdio: 'inherit' })
+  }
+}

lib/index.js

@@ -106,7 +106,7 @@ class ServerlessApigatewayServiceProxy {
     let message = ''
     let serviceProxyMessages = ''
 
-    const endpointInfo = this.gatheredData.info.endpoint
+    const endpointInfo = this.gatheredData.info.endpoints
     message += `${chalk.yellow.underline('Serverless APIGateway Service Proxy OutPuts')}\n`
     message += `${chalk.yellow('endpoints:')}`