From e38848f6dafa189388e714c9160565969723298e Mon Sep 17 00:00:00 2001
From: corymhall <43035978+corymhall@users.noreply.github.com>
Date: Fri, 1 Apr 2022 13:06:35 +0000
Subject: [PATCH] chore(aws-cognito): fix broken integration tests

---
 ...integ.user-pool-client-explicit-props.expected.json | 10 ++++++++++
 .../test/integ.user-pool-client-explicit-props.ts      |  6 +++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.expected.json b/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.expected.json
index c2c5b4d18b269..33d544ca1751a 100644
--- a/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.expected.json
+++ b/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.expected.json
@@ -20,6 +20,16 @@
         },
         "EmailVerificationMessage": "The verification code to your new account is {####}",
         "EmailVerificationSubject": "Verify your new account",
+        "Schema": [
+          {
+            "AttributeDataType": "String",
+            "Name": "attribute_one"
+          },
+          {
+            "AttributeDataType": "String",
+            "Name": "attribute_two"
+          }
+        ],
         "SmsVerificationMessage": "The verification code to your new account is {####}",
         "VerificationMessageTemplate": {
           "DefaultEmailOption": "CONFIRM_WITH_CODE",
diff --git a/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.ts b/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.ts
index d8a058a86d9c8..6f71e03dbed6e 100644
--- a/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.ts
+++ b/packages/@aws-cdk/aws-cognito/test/integ.user-pool-client-explicit-props.ts
@@ -1,11 +1,15 @@
 import { App, RemovalPolicy, Stack } from '@aws-cdk/core';
-import { OAuthScope, UserPool, ClientAttributes } from '../lib';
+import { OAuthScope, UserPool, ClientAttributes, StringAttribute } from '../lib';
 
 const app = new App();
 const stack = new Stack(app, 'integ-user-pool-client-explicit-props');
 
 const userpool = new UserPool(stack, 'myuserpool', {
   removalPolicy: RemovalPolicy.DESTROY,
+  customAttributes: {
+    attribute_one: new StringAttribute(),
+    attribute_two: new StringAttribute(),
+  },
 });
 
 userpool.addClient('myuserpoolclient', {