fix unit tests

Signed-off-by: Francis <colifran@amazon.com>
aws · May 2, 2024 · b02ce68 · b02ce68
1 parent 5056e40
commit b02ce68
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/packages/aws-cdk-lib/aws-kinesis/test/stream.test.ts b/packages/aws-cdk-lib/aws-kinesis/test/stream.test.ts
@@ -1,7 +1,7 @@
 import { Match, Template } from '../../assertions';
 import * as iam from '../../aws-iam';
 import * as kms from '../../aws-kms';
-import { App, Duration, Stack, CfnParameter, RemovalPolicy } from '../../core';
+import { App, Duration, Stack, CfnParameter, RemovalPolicy, CfnDeletionPolicy } from '../../core';
 import { Stream, StreamEncryption, StreamMode } from '../lib';
 
 describe('Kinesis data streams', () => {
@@ -1270,8 +1270,8 @@ describe('Kinesis data streams', () => {
     new Stream(stack, 'Stream');
 
     // THEN
-    Template.fromStack(stack).hasResourceProperties('AWS::Kinesis::Stream', {
-      DeletionPolicy: RemovalPolicy.RETAIN,
+    Template.fromStack(stack).hasResource('AWS::Kinesis::Stream', {
+      DeletionPolicy: CfnDeletionPolicy.RETAIN,
     });
   });
 
@@ -1285,8 +1285,8 @@ describe('Kinesis data streams', () => {
     });
 
     // THEN
-    Template.fromStack(stack).hasResourceProperties('AWS::Kinesis::Stream', {
-      DeletionPolicy: RemovalPolicy.DESTROY,
+    Template.fromStack(stack).hasResource('AWS::Kinesis::Stream', {
+      DeletionPolicy: CfnDeletionPolicy.DELETE,
     });
   });
 });