lambda-at-edge.test.js

const { createComponent, assertHasCacheBehavior } = require('../test-utils')

const { mockCreateDistribution, mockCreateDistributionPromise } = require('aws-sdk')

describe('Input origin as a custom url', () => {
  let component

  beforeEach(async () => {
    mockCreateDistributionPromise.mockResolvedValueOnce({
      Distribution: {
        Id: 'distribution123'
      }
    })

    component = await createComponent()
  })

  it('creates distribution with lambda associations for each event type', async () => {
    await component.default({
      origins: [
        {
          url: 'https://exampleorigin.com',
          pathPatterns: {
            '/some/path': {
              ttl: 10,
              'lambda@edge': {
                'viewer-request': {
                  arn: 'arn:aws:lambda:us-east-1:123:function:viewerRequestFunction',
                  includeBody: true
                },
                'origin-request': {
                  arn: 'arn:aws:lambda:us-east-1:123:function:originRequestFunction',
                  includeBody: false
                },
                'origin-response': 'arn:aws:lambda:us-east-1:123:function:originResponseFunction',
                'viewer-response': 'arn:aws:lambda:us-east-1:123:function:viewerResponseFunction'
              }
            }
          }
        }
      ]
    })

    assertHasCacheBehavior(mockCreateDistribution, {
      PathPattern: '/some/path',
      LambdaFunctionAssociations: {
        Quantity: 4,
        Items: [
          {
            EventType: 'viewer-request',
            LambdaFunctionARN: 'arn:aws:lambda:us-east-1:123:function:viewerRequestFunction',
            IncludeBody: true
          },
          {
            EventType: 'origin-request',
            LambdaFunctionARN: 'arn:aws:lambda:us-east-1:123:function:originRequestFunction',
            IncludeBody: false
          },
          {
            EventType: 'origin-response',
            LambdaFunctionARN: 'arn:aws:lambda:us-east-1:123:function:originResponseFunction',
            IncludeBody: false
          },
          {
            EventType: 'viewer-response',
            LambdaFunctionARN: 'arn:aws:lambda:us-east-1:123:function:viewerResponseFunction',
            IncludeBody: false
          }
        ]
      }
    })

    expect(mockCreateDistribution.mock.calls[0][0]).toMatchSnapshot()
  })

  it('throws error when event type provided is not valid', async () => {
    expect.assertions(1)

    try {
      await component.default({
        origins: [
          {
            url: 'https://exampleorigin.com',
            pathPatterns: {
              '/some/path': {
                ttl: 10,
                'lambda@edge': {
                  'invalid-eventtype': 'arn:aws:lambda:us-east-1:123:function:viewerRequestFunction'
                }
              }
            }
          }
        ]
      })
    } catch (err) {
      expect(err.message).toEqual(
        '"invalid-eventtype" is not a valid lambda trigger. See https://docs.aws.amazon.com/AmazonCloudFront/latest/DeveloperGuide/lambda-cloudfront-trigger-events.html for valid event types.'
      )
    }
  })

  it('throws error when includeBody given for event types other than request', async () => {
    expect.assertions(1)

    try {
      await component.default({
        origins: [
          {
            url: 'https://exampleorigin.com',
            pathPatterns: {
              '/some/path': {
                ttl: 10,
                'lambda@edge': {
                  'viewer-response': {
                    arn: 'arn:aws:lambda:us-east-1:123:function:viewerRequestFunction',
                    includeBody: true
                  }
                }
              }
            }
          }
        ]
      })
    } catch (err) {
      expect(err.message).toEqual('"includeBody" not allowed for viewer-response lambda triggers.')
    }
  })
})