Add tests for getMethodAuthorization

Author: vlewin

lib/deploy/events/apiGateway/methods.test.js

@@ -242,4 +242,78 @@ describe('#methods()', () => {
       .to.equal('\'*\'');
     });
   });
+
+  describe('#getMethodAuthorization()', () => {
+    it('should return resource properties with AuthorizationType: NONE if no authorizer provided', () => {
+      const event = {
+        path: 'foo/bar1',
+        method: 'post'
+      };
+
+      expect(serverlessStepFunctions.getMethodAuthorization(event)
+              .Properties.AuthorizationType).to.equal('NONE');
+    });
+
+    it('should return resource properties with AuthorizationType: AWS_IAM', () => {
+
+      const event = {
+        authorizer: {
+          type: 'AWS_IAM',
+          authorizerId: 'foo12345',
+        },
+      };
+
+      expect(serverlessStepFunctions.getMethodAuthorization(event)
+              .Properties.AuthorizationType).to.equal('AWS_IAM');
+    });
+
+    it('should return resource properties with AuthorizationType: CUSTOM and authotizerId', () => {
+
+      const event = {
+        authorizer: {
+          type: 'CUSTOM',
+          authorizerId: 'foo12345',
+        },
+      };
+
+      expect(serverlessStepFunctions.getMethodAuthorization(event)
+              .Properties.AuthorizationType).to.equal('CUSTOM');
+      expect(serverlessStepFunctions.getMethodAuthorization(event)
+              .Properties.AuthorizerId).to.equal('foo12345');
+
+    });
+
+    it('should return resource properties with AuthorizationType: CUSTOM and resource reference', () => {
+      const event = {
+        authorizer: {
+          name: 'authorizer',
+          arn: { 'Fn::GetAtt': ['SomeLambdaFunction', 'Arn'] },
+          resultTtlInSeconds: 300,
+          identitySource: 'method.request.header.Authorization',
+        },
+      };
+
+      const autorization = serverlessStepFunctions.getMethodAuthorization(event)
+      expect(autorization.Properties.AuthorizationType)
+              .to.equal('CUSTOM');
+
+      expect(autorization.Properties.AuthorizerId)
+              .to.deep.equal({ Ref: 'AuthorizerApiGatewayAuthorizer' });
+    });
+
+    it('should return resource properties with AuthorizationType: COGNITO_USER_POOLS and resource reference', () => {
+      const event = {
+        authorizer: {
+          name: 'authorizer',
+          arn: 'arn:aws:cognito-idp:us-east-1:xxx:userpool/us-east-1_ZZZ',
+        },
+      };
+
+      const autorization = serverlessStepFunctions.getMethodAuthorization(event)
+      expect(autorization.Properties.AuthorizationType)
+              .to.equal('COGNITO_USER_POOLS');
+      expect(autorization.Properties.AuthorizerId)
+              .to.deep.equal({ Ref: 'AuthorizerApiGatewayAuthorizer' });
+    });
+  });
 });