fix cli test

aws · Mar 30, 2023 · 85d828f · 85d828f
1 parent 6681eb1
commit 85d828f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tools/@aws-cdk/remodel/lib/cli.ts b/tools/@aws-cdk/remodel/lib/cli.ts
@@ -26,6 +26,7 @@ import {
   rewriteDepVersions,
   rewriteSourceFiles,
   replaceInFiles,
+  rewriteSourceFile,
 } from './util';
 
 
@@ -629,6 +630,19 @@ async function reformatCliPackage(dir: string) {
 
     return lines.join('\n');
   });
+
+  // Use cloud-assembly-schema from aws-cdk-lib in test so it matches correctly
+  await rewriteSourceFile(
+    path.join(dir, 'test', 'api', 'exec.test.ts'),
+    async (fileContent: string) => {
+      const lines = fileContent.split('\n')
+        .map((line) => {
+          return line.replace('@aws-cdk/cloud-assembly-schema', 'aws-cdk-lib/cloud-assembly-schema');
+        });
+
+      return lines.join('\n');
+    },
+  );
 }
 
 async function reformatIntegPackage(dir: string) {