test: added some tests for custom headers

Author: nickytonline

test/__snapshots__/index.js.snap

@@ -1,5 +1,47 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
+exports[`custom headers appends custom headers in the Netlify configuration 1`] = `
+Array [
+  Object {
+    "for": "/",
+    "values": Object {
+      "X-Existing-Header": "true",
+    },
+  },
+  Object {
+    "for": "/",
+    "values": Object {
+      "X-Unit-Test": "true",
+    },
+  },
+  Object {
+    "for": "/unit-test",
+    "values": Object {
+      "X-Another-Unit-Test": "true",
+      "X-Another-Unit-Test-Again": "true",
+    },
+  },
+]
+`;
+
+exports[`custom headers sets custom headers in the Netlify configuration 1`] = `
+Array [
+  Object {
+    "for": "/",
+    "values": Object {
+      "X-Unit-Test": "true",
+    },
+  },
+  Object {
+    "for": "/unit-test",
+    "values": Object {
+      "X-Another-Unit-Test": "true",
+      "X-Another-Unit-Test-Again": "true",
+    },
+  },
+]
+`;
+
 exports[`onBuild() generates a file referencing all page sources 1`] = `
 "// This file is purely to allow nft to know about these pages. It should be temporary.
 exports.resolvePages = () => {

test/index.js

@@ -25,9 +25,14 @@ const {
   patchNextFiles,
   unpatchNextFiles,
 } = require('../plugin/src/helpers/files')
-const { getRequiredServerFiles, updateRequiredServerFiles } = require('../plugin/src/helpers/config')
+const {
+  getRequiredServerFiles,
+  updateRequiredServerFiles,
+  generateCustomHeaders,
+} = require('../plugin/src/helpers/config')
 const { dirname } = require('path')
 const { getProblematicUserRewrites } = require('../plugin/src/helpers/verification')
+const { onPostBuild } = require('../plugin/lib')
 
 const FIXTURES_DIR = `${__dirname}/fixtures`
 const SAMPLE_PROJECT_DIR = `${__dirname}/../demos/default`
@@ -246,11 +251,11 @@ describe('onBuild()', () => {
     process.env.URL = mockSiteUrl
 
     await moveNextDist()
-    
+
     const initialConfig = await getRequiredServerFiles(netlifyConfig.build.publish)
-    initialConfig.config.basePath = "/foo"
+    initialConfig.config.basePath = '/foo'
     await updateRequiredServerFiles(netlifyConfig.build.publish, initialConfig)
-    
+
     await plugin.onBuild(defaultArgs)
 
     expect(onBuildHasRun(netlifyConfig)).toBe(true)
@@ -788,3 +793,85 @@ describe('function helpers', () => {
     )
   })
 })
+
+describe.only('custom headers', () => {
+  it('sets custom headers in the Netlify configuration', async () => {
+    netlifyConfig.headers = []
+    const headers = [
+      // single header for a route
+      {
+        source: '/',
+        headers: [
+          {
+            key: 'X-Unit-Test',
+            value: 'true',
+          },
+        ],
+        regex: '^/(?:/)?$',
+      },
+      // multiple headers for a route
+      {
+        source: '/unit-test',
+        headers: [
+          {
+            key: 'X-Another-Unit-Test',
+            value: 'true',
+          },
+          {
+            key: 'X-Another-Unit-Test-Again',
+            value: 'true',
+          },
+        ],
+        regex: '^/(?:/)?$',
+      },
+    ]
+
+    generateCustomHeaders(headers, netlifyConfig.headers)
+
+    expect(netlifyConfig.headers).toMatchSnapshot()
+  })
+
+  it('appends custom headers in the Netlify configuration', async () => {
+    netlifyConfig.headers = [
+      {
+        for: '/',
+        values: {
+          'X-Existing-Header': 'true',
+        },
+      },
+    ]
+
+    const headers = [
+      // single header for a route
+      {
+        source: '/',
+        headers: [
+          {
+            key: 'X-Unit-Test',
+            value: 'true',
+          },
+        ],
+        regex: '^/(?:/)?$',
+      },
+      // multiple headers for a route
+      {
+        source: '/unit-test',
+        headers: [
+          {
+            key: 'X-Another-Unit-Test',
+            value: 'true',
+          },
+          {
+            key: 'X-Another-Unit-Test-Again',
+            value: 'true',
+          },
+        ],
+        regex: '^/(?:/)?$',
+      },
+    ]
+
+    generateCustomHeaders(headers, netlifyConfig.headers)
+
+    expect(netlifyConfig.headers).toMatchSnapshot()
+  })
+})