chore(aws-cognito): fix broken integration tests

aws · Apr 1, 2022 · e38848f · e38848f
1 parent fd1fff9
commit e38848f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 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
@@ -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
@@ -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', {