add test utility to normalize asset names

aws4embeddedlinux · Sep 12, 2023 · 7003947 · 7003947
1 parent 03715f7
commit 7003947
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 17 deletions.
diff --git a/test/__snapshots__/build-image-data.test.ts.snap b/test/__snapshots__/build-image-data.test.ts.snap
@@ -14,7 +14,7 @@ exports[`Build Image Data Snapshot 1`] = `
       "Properties": {
         "Content": {
           "S3Bucket": "cdk-hnb659fds-assets-111111111111-eu-central-1",
-          "S3Key": "arbitrary-file.zip",
+          "S3Key": "arbitrary-file.ext",
         },
         "Description": "/opt/awscli/aws",
       },
@@ -38,7 +38,7 @@ exports[`Build Image Data Snapshot 1`] = `
           "cdk-hnb659fds-assets-111111111111-eu-central-1",
         ],
         "SourceObjectKeys": [
-          "arbitrary-file.zip",
+          "arbitrary-file.ext",
         ],
       },
       "Type": "Custom::CDKBucketDeployment",
@@ -266,7 +266,7 @@ exports[`Build Image Data Snapshot 1`] = `
       "Properties": {
         "Code": {
           "S3Bucket": "cdk-hnb659fds-assets-111111111111-eu-central-1",
-          "S3Key": "arbitrary-file.zip",
+          "S3Key": "arbitrary-file.ext",
         },
         "Environment": {
           "Variables": {
@@ -297,7 +297,7 @@ exports[`Build Image Data Snapshot 1`] = `
       "Properties": {
         "Code": {
           "S3Bucket": "cdk-hnb659fds-assets-111111111111-eu-central-1",
-          "S3Key": "arbitrary-file.zip",
+          "S3Key": "arbitrary-file.ext",
         },
         "Description": {
           "Fn::Join": [

diff --git a/test/__snapshots__/demo-pipeline.test.ts.snap b/test/__snapshots__/demo-pipeline.test.ts.snap
@@ -171,7 +171,7 @@ exports[`Demo Pipeline Poky AMI Pipeline - check role name trim 1`] = `
             {
               "Name": "SCRIPT_URL",
               "Type": "PLAINTEXT",
-              "Value": "s3://cdk-hnb659fds-assets-12341234-eu-central-1/cb07978d00156c655bc2617e9ff6d544ec346fc8bfd37b01b644b4694d399265.sh",
+              "Value": "s3://cdk-hnb659fds-assets-12341234-eu-central-1/arbitrary-file.ext",
             },
           ],
           "Image": {
@@ -1936,7 +1936,7 @@ def handler(event, context):
           "BranchName": "main",
           "S3": {
             "Bucket": "cdk-hnb659fds-assets-12341234-eu-central-1",
-            "Key": "8328421057d696d22d2b958f640836605dbbf21ce1473e4db341cffcd075ae25.zip",
+            "Key": "arbitrary-file.ext",
           },
         },
         "RepositoryName": "layer-repo-PokyAmiPipeline2ExportsOutputFnGetAttPipelineVpc0543904ACidrBlock70DEC992",
@@ -2291,7 +2291,7 @@ exports[`Demo Pipeline Snapshot Poky AMI Pipeline 1`] = `
             {
               "Name": "SCRIPT_URL",
               "Type": "PLAINTEXT",
-              "Value": "s3://cdk-hnb659fds-assets-12341234-eu-central-1/cb07978d00156c655bc2617e9ff6d544ec346fc8bfd37b01b644b4694d399265.sh",
+              "Value": "s3://cdk-hnb659fds-assets-12341234-eu-central-1/arbitrary-file.ext",
             },
           ],
           "Image": {
@@ -4056,7 +4056,7 @@ def handler(event, context):
           "BranchName": "main",
           "S3": {
             "Bucket": "cdk-hnb659fds-assets-12341234-eu-central-1",
-            "Key": "8328421057d696d22d2b958f640836605dbbf21ce1473e4db341cffcd075ae25.zip",
+            "Key": "arbitrary-file.ext",
           },
         },
         "RepositoryName": "layer-repo-MyTestStack",

diff --git a/test/build-image-data.test.ts b/test/build-image-data.test.ts
@@ -1,6 +1,7 @@
 import * as cdk from 'aws-cdk-lib';
 import { Template } from 'aws-cdk-lib/assertions';
 import { BuildImageDataStack } from '../lib/build-image-data';
+import {  normalizedTemplateFromStack } from './util';
 
 describe('Build Image Data', () => {
   const props = {
@@ -27,13 +28,7 @@ describe('Build Image Data', () => {
     const app = new cdk.App();
     const stack = new BuildImageDataStack(app, 'MyTestStack', props);
     /* We must change some randomly generated file names used in the S3 asset construct. */
-    const templateWithRandomKeys = Template.fromStack(stack);
-    const templateWithConstKeys = JSON.parse(
-      JSON.stringify(templateWithRandomKeys.toJSON()).replace(
-        /[a-z0-9]{64}\.zip/g,
-        'arbitrary-file.zip'
-      )
-    );
+    const templateWithConstKeys = normalizedTemplateFromStack(stack);
     expect(templateWithConstKeys).toMatchSnapshot();
   });
 });
diff --git a/test/demo-pipeline.test.ts b/test/demo-pipeline.test.ts
@@ -4,6 +4,7 @@ import { DemoPipelineStack } from '../lib/demo-pipeline';
 import { Repository } from 'aws-cdk-lib/aws-ecr';
 import { Vpc } from 'aws-cdk-lib/aws-ec2';
 import { ProjectKind } from '../lib';
+import { normalizedTemplateFromStack } from './util';
 
 describe('Demo Pipeline', () => {
   const env = { account: '12341234', region: 'eu-central-1' };
@@ -35,7 +36,7 @@ describe('Demo Pipeline', () => {
       vpc,
       projectKind: ProjectKind.PokyAmi,
     });
-    const template = Template.fromStack(stack);
+    const template = normalizedTemplateFromStack(stack);
     expect(template).toMatchSnapshot();
   });
 
@@ -55,7 +56,7 @@ describe('Demo Pipeline', () => {
         projectKind: ProjectKind.PokyAmi,
       }
     );
-    const template = Template.fromStack(stack);
+    const template = normalizedTemplateFromStack(stack);
     expect(template).toMatchSnapshot();
   });
 });
diff --git a/test/util.ts b/test/util.ts
@@ -0,0 +1,17 @@
+import * as cdk from 'aws-cdk-lib';
+import { Template } from 'aws-cdk-lib/assertions';
+
+/**
+ * Remove hashed asset values and put a normalized name in instead.
+ */
+export function normalizedTemplateFromStack(stack: cdk.Stack): Template {
+    const templateWithRandomKeys = Template.fromStack(stack);
+    const templateWithConstKeys = JSON.parse(
+        JSON.stringify(templateWithRandomKeys.toJSON()).replace(
+            /[a-z0-9]{64}\.(zip|sh)/g,
+            'arbitrary-file.ext'
+        )
+    );
+
+    return templateWithConstKeys;
+}