Added unit tests

serverless · Feb 4, 2019 · 9a25ade · 9a25ade
1 parent 73d95c2
commit 9a25ade
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/lib/plugins/aws/package/compile/events/apiGateway/lib/cors.js b/lib/plugins/aws/package/compile/events/apiGateway/lib/cors.js
@@ -18,11 +18,11 @@ module.exports = {
         origins = origin.split(',').map(a => a.trim());
       }
 
-      if (origins.length) {
+      if (!_.isEmpty(origins)) {
         origin = origins[0];
       }
 
-      if (!origin && !origins.length) {
+      if (!origin) {
         const errorMessage = 'must specify either origin or origins';
         throw new this.serverless.classes.Error(errorMessage);
       }

diff --git a/lib/plugins/aws/package/compile/events/apiGateway/lib/cors.test.js b/lib/plugins/aws/package/compile/events/apiGateway/lib/cors.test.js
@@ -64,6 +64,7 @@ describe('#compileCors()', () => {
     awsCompileApigEvents.validated.corsPreflight = {
       'users/update': {
         origin: 'http://example.com',
+        origins: [],
         headers: ['*'],
         methods: ['OPTIONS', 'PUT'],
         allowCredentials: false,

diff --git a/lib/plugins/aws/package/compile/events/apiGateway/lib/validate.test.js b/lib/plugins/aws/package/compile/events/apiGateway/lib/validate.test.js
@@ -644,6 +644,7 @@ describe('#validate()', () => {
                 origins: ['acme.com'],
                 methods: ['POST', 'OPTIONS'],
                 maxAge: 86400,
+                cacheControl: 'max-age=600, s-maxage=600, proxy-revalidate'
               },
             },
           },
@@ -659,6 +660,7 @@ describe('#validate()', () => {
       origins: ['acme.com'],
       allowCredentials: false,
       maxAge: 86400,
+      cacheControl: 'max-age=600, s-maxage=600, proxy-revalidate'
     });
   });
 
@@ -676,6 +678,7 @@ describe('#validate()', () => {
                 ],
                 allowCredentials: true,
                 maxAge: 10000,
+                cacheControl: 'max-age=600, s-maxage=600, proxy-revalidate'
               },
             },
           }, {
@@ -723,6 +726,8 @@ describe('#validate()', () => {
       .to.deep.equal(['TestHeader2', 'TestHeader']);
     expect(validated.corsPreflight.users.maxAge)
       .to.equal(86400);
+    expect(validated.corsPreflight.users.cacheControl)
+      .to.equal('max-age=600, s-maxage=600, proxy-revalidate');
     expect(validated.corsPreflight.users.allowCredentials)
       .to.equal(true);
     expect(validated.corsPreflight['users/{id}'].allowCredentials)