feat(apigatewayv2): support for mock integration type

packages/@aws-cdk/aws-apigatewayv2-integrations/lib/websocket/index.ts

@@ -1 +1,2 @@
 export * from './lambda';
+export * from './mock';

packages/@aws-cdk/aws-apigatewayv2-integrations/lib/websocket/mock.ts

@@ -0,0 +1,27 @@
+import {
+ WebSocketRouteIntegration,
+ WebSocketIntegrationType,
+ WebSocketRouteIntegrationConfig,
+ WebSocketRouteIntegrationBindOptions,
+} from '@aws-cdk/aws-apigatewayv2';
+
+/**
+ * Mock WebSocket Integration
+ */
+export class MockWebSocketIntegration extends WebSocketRouteIntegration {
+
+ /**
+ * @param id id of the underlying integration construct
+ */
+ constructor(id: string) {
+ super(id);
+ }
+
+ bind(options: WebSocketRouteIntegrationBindOptions): WebSocketRouteIntegrationConfig {
+ options;
+ return {
+ type: WebSocketIntegrationType.MOCK,
+ uri: '',
+ };
+ }
+}

packages/@aws-cdk/aws-apigatewayv2-integrations/test/websocket/integ.mock.expected.json

@@ -0,0 +1,108 @@
+{
+ "Resources": {
+ "mywsapi32E6CE11": {
+ "Type": "AWS::ApiGatewayV2::Api",
+ "Properties": {
+ "Name": "mywsapi",
+ "ProtocolType": "WEBSOCKET",
+ "RouteSelectionExpression": "$request.body.action"
+ }
+ },
+ "mywsapidefaultRouteDefaultIntegrationFFCB3BA9": {
+ "Type": "AWS::ApiGatewayV2::Integration",
+ "Properties": {
+ "ApiId": {
+ "Ref": "mywsapi32E6CE11"
+ },
+ "IntegrationType": "MOCK",
+ "IntegrationUri": ""
+ }
+ },
+ "mywsapidefaultRouteE9382DF8": {
+ "Type": "AWS::ApiGatewayV2::Route",
+ "Properties": {
+ "ApiId": {
+ "Ref": "mywsapi32E6CE11"
+ },
+ "RouteKey": "$default",
+ "AuthorizationType": "NONE",
+ "Target": {
+ "Fn::Join": [
+ "",
+ [
+ "integrations/",
+ {
+ "Ref": "mywsapidefaultRouteDefaultIntegrationFFCB3BA9"
+ }
+ ]
+ ]
+ }
+ }
+ },
+ "mywsapisendmessageRouteSendMessageIntegrationD29E12F9": {
+ "Type": "AWS::ApiGatewayV2::Integration",
+ "Properties": {
+ "ApiId": {
+ "Ref": "mywsapi32E6CE11"
+ },
+ "IntegrationType": "MOCK",
+ "IntegrationUri": ""
+ }
+ },
+ "mywsapisendmessageRouteAE873328": {
+ "Type": "AWS::ApiGatewayV2::Route",
+ "Properties": {
+ "ApiId": {
+ "Ref": "mywsapi32E6CE11"
+ },
+ "RouteKey": "sendmessage",
+ "AuthorizationType": "NONE",
+ "Target": {
+ "Fn::Join": [
+ "",
+ [
+ "integrations/",
+ {
+ "Ref": "mywsapisendmessageRouteSendMessageIntegrationD29E12F9"
+ }
+ ]
+ ]
+ }
+ }
+ },
+ "mystage114C35EC": {
+ "Type": "AWS::ApiGatewayV2::Stage",
+ "Properties": {
+ "ApiId": {
+ "Ref": "mywsapi32E6CE11"
+ },
+ "StageName": "dev",
+ "AutoDeploy": true
+ }
+ }
+ },
+ "Outputs": {
+ "ApiEndpoint": {
+ "Value": {
+ "Fn::Join": [
+ "",
+ [
+ "wss://",
+ {
+ "Ref": "mywsapi32E6CE11"
+ },
+ ".execute-api.",
+ {
+ "Ref": "AWS::Region"
+ },
+ ".",
+ {
+ "Ref": "AWS::URLSuffix"
+ },
+ "/dev"
+ ]
+ ]
+ }
+ }
+ }
+}

packages/@aws-cdk/aws-apigatewayv2-integrations/test/websocket/integ.mock.ts

@@ -0,0 +1,24 @@
+import { WebSocketApi, WebSocketStage } from '@aws-cdk/aws-apigatewayv2';
+import { App, CfnOutput, Stack } from '@aws-cdk/core';
+import { MockWebSocketIntegration } from '../../lib';
+
+/*
+ * Stack verification steps:
+ * 1. Verify manually that the integration has type "MOCK"
+ */
+
+const app = new App();
+const stack = new Stack(app, 'integ-mock-websocket-integration');
+
+const webSocketApi = new WebSocketApi(stack, 'mywsapi', {
+ defaultRouteOptions: { integration: new MockWebSocketIntegration('DefaultIntegration') },
+});
+const stage = new WebSocketStage(stack, 'mystage', {
+ webSocketApi,
+ stageName: 'dev',
+ autoDeploy: true,
+});
+
+webSocketApi.addRoute('sendmessage', { integration: new MockWebSocketIntegration('SendMessageIntegration') });
+
+new CfnOutput(stack, 'ApiEndpoint', { value: stage.url });

packages/@aws-cdk/aws-apigatewayv2-integrations/test/websocket/mock.test.ts

@@ -0,0 +1,23 @@
+import { Template } from '@aws-cdk/assertions';
+import { WebSocketApi } from '@aws-cdk/aws-apigatewayv2';
+import { Stack } from '@aws-cdk/core';
+import { MockWebSocketIntegration } from '../../lib';
+
+
+describe('MockWebSocketIntegration', () => {
+ test('default', () => {
+ // GIVEN
+ const stack = new Stack();
+
+ // WHEN
+ new WebSocketApi(stack, 'Api', {
+ defaultRouteOptions: { integration: new MockWebSocketIntegration('DefaultIntegration') },
+ });
+
+ // THEN
+ Template.fromStack(stack).hasResourceProperties('AWS::ApiGatewayV2::Integration', {
+ IntegrationType: 'MOCK',
+ IntegrationUri: '',
+ });
+ });
+});

packages/@aws-cdk/aws-apigatewayv2/lib/websocket/integration.ts

@@ -24,7 +24,11 @@ export enum WebSocketIntegrationType {
  /**
  * AWS Proxy Integration Type
  */
- AWS_PROXY = 'AWS_PROXY'
+ AWS_PROXY = 'AWS_PROXY',
+ /**
+ * Mock Integration Type
+ */
+ MOCK = 'MOCK'
 }
 
 /**