From 86739799005abfccc1fc01a79e55c37e7202be66 Mon Sep 17 00:00:00 2001
From: watany <76135106+watany-dev@users.noreply.github.com>
Date: Wed, 22 Nov 2023 09:15:59 +0000
Subject: [PATCH] integ

---
 .../integ.distribution-function-runtime.ts    | 41 +++++++++++++++++++
 .../test/integ.distribution-function.ts       |  1 -
 2 files changed, 41 insertions(+), 1 deletion(-)
 create mode 100644 packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function-runtime.ts

diff --git a/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function-runtime.ts b/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function-runtime.ts
new file mode 100644
index 0000000000000..4359252deb8df
--- /dev/null
+++ b/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function-runtime.ts
@@ -0,0 +1,41 @@
+import * as cdk from 'aws-cdk-lib';
+import { IntegTest } from '@aws-cdk/integ-tests-alpha';
+import { TestOrigin } from './test-origin';
+import * as cloudfront from 'aws-cdk-lib/aws-cloudfront';
+
+const app = new cdk.App();
+const stack = new cdk.Stack(app, 'integ-distribution-function', { env: { region: 'eu-west-1' } });
+
+const cfFunctionRequest = new cloudfront.Function(stack, 'FunctionRequest', {
+  code: cloudfront.FunctionCode.fromInline('function handler(event) { return event.request }'),
+});
+
+const cfFunctionResponse = new cloudfront.Function(stack, 'FunctionResponse', {
+  code: cloudfront.FunctionCode.fromInline('function handler(event) { return event.response }'),
+});
+
+new cloudfront.Distribution(stack, 'Dist', {
+  defaultBehavior: {
+    origin: new TestOrigin('www.example.com'),
+    cachePolicy: cloudfront.CachePolicy.CACHING_DISABLED,
+    functionAssociations: [{
+      function: cfFunctionRequest,
+      eventType: cloudfront.FunctionEventType.VIEWER_REQUEST,
+    }, {
+      function: cfFunctionResponse,
+      eventType: cloudfront.FunctionEventType.VIEWER_RESPONSE,
+    }],
+  },
+});
+
+new cdk.CfnOutput(stack, 'RequestFunctionArn', { value: cfFunctionRequest.functionArn });
+new cdk.CfnOutput(stack, 'RequestFunctionStage', { value: cfFunctionRequest.functionStage });
+
+new cdk.CfnOutput(stack, 'ResponseFunctionArn', { value: cfFunctionResponse.functionArn });
+new cdk.CfnOutput(stack, 'ResponseFunctionStage', { value: cfFunctionResponse.functionStage });
+
+new IntegTest(app, 'CF2Runtime', {
+  testCases: [stack],
+});
+
+app.synth();
diff --git a/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function.ts b/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function.ts
index f177a67fbe2d2..cf56ff5e90be8 100644
--- a/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function.ts
+++ b/packages/@aws-cdk-testing/framework-integ/test/aws-cloudfront/test/integ.distribution-function.ts
@@ -7,7 +7,6 @@ const stack = new cdk.Stack(app, 'integ-distribution-function', { env: { region:
 
 const cfFunction = new cloudfront.Function(stack, 'Function', {
   code: cloudfront.FunctionCode.fromInline('function handler(event) { return event.request }'),
-  runtime: cloudfront.FunctionRuntime.JS_2_0,
 });
 
 new cloudfront.Distribution(stack, 'Dist', {