add unit test

Author: raphaeldeem-acorns

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

@@ -9,14 +9,29 @@ describe('#compileRestApi()', () => {
   let serverless;
   let serverlessStepFunctions;
 
-  const serviceResourcesAwsResourcesObjectMock = {
-    Resources: {
-      ApiGatewayRestApi: {
-        Type: 'AWS::ApiGateway::RestApi',
-        Properties: {
-          Name: 'dev-new-service',
-          EndpointConfiguration: {
-            Types: ['EDGE'],
+  const serviceresourcesawsresourcesprivateobjectmock = {
+    resources: {
+      apigatewayrestapi: {
+        type: 'aws::apigateway::restapi',
+        properties: {
+          name: 'dev-new-service',
+          endpointconfiguration: {
+            vpcEndpointIds: ["vpc-0a60eb65b4foo", "vpc-0a60eb65b4bar"],
+            types: ['private'],
+          },
+        },
+      },
+    },
+  };
+
+  const serviceresourcesawsresourcesobjectmock = {
+    resources: {
+      apigatewayrestapi: {
+        type: 'aws::apigateway::restapi',
+        properties: {
+          name: 'dev-new-service',
+          endpointconfiguration: {
+            types: ['edge'],
           },
         },
       },
@@ -135,8 +150,14 @@ describe('#compileRestApi()', () => {
     expect(() => serverlessStepFunctions.compileRestApi()).to.not.throw(Error);
   });
 
-  it('should compile if endpointType property is PRIVATE', () => {
+  it('should not compile if endpointType property is PRIVATE and no vpcEndpointIds', () => {
+    serverlessStepFunctions.serverless.service.provider.endpointType = 'PRIVATE';
+    expect(() => serverlessStepFunctions.compileRestApi()).to.throw(Error);
+  });
+
+  it('should compile if endpointType property is PRIVATE and vpcEndpointIds', () => {
     serverlessStepFunctions.serverless.service.provider.endpointType = 'PRIVATE';
+    serverlessStepFunctions.serverless.service.provider.vpcEndpointIds = ["vpc-0a60eb65b4foo", "vpc-0a60eb65b4bar"];
     expect(() => serverlessStepFunctions.compileRestApi()).to.not.throw(Error);
   });