test: multi component support in unit tests

Author: Cameron Magee

src/index.ts

@@ -125,13 +125,6 @@ export class ImagePipeline extends Construct {
 
     infrastructureConfig.addDependsOn(profile);
 
-    // const component = new imagebuilder.CfnComponent(this, 'Component', {
-    //   name: props.componentName,
-    //   platform: props.platform ?? 'Linux',
-    //   version: '1.0.0',
-    //   data: readFileSync(props.componentDocPath).toString(),
-    // });
-
     const imageRecipe = new imagebuilder.CfnImageRecipe(this, 'ImageRecipe', {
       components: [],
       name: props.imageRecipe,
@@ -143,7 +136,7 @@ export class ImagePipeline extends Construct {
       let component = new imagebuilder.CfnComponent(this, props.componentNames[index], {
         name: props.componentNames[index],
         platform: props.platform ?? 'Linux',
-        version: props.componentVersions[index],
+        version: props.componentVersions[index] ?? '0.0.1',
         data: readFileSync(document).toString(),
       });
 

test/imagepipeline.test.ts

@@ -5,8 +5,8 @@ import { ImagePipeline, ImagePipelineProps } from '../src';
 let template: Template;
 
 const props: ImagePipelineProps = {
-  componentDocuments: ['test/test_component_example.yml'],
-  componentNames: ['TestComponent'],
+  componentDocuments: ['test/test_component_example.yml', 'test/test_component_example_2.yml'],
+  componentNames: ['TestComponent', 'TestComponent2'],
   componentVersions: ['1.0.0'],
   profileName: 'TestProfile',
   infraConfigName: 'TestInfrastructureConfig',
@@ -67,7 +67,6 @@ test('Infrastructure Configuration IAM Role and Instance Profile are created', (
 });
 
 test('IAM Role contains necessary permission set', () => {
-  console.log(template.findResources);
   template.hasResourceProperties('AWS::IAM::Role',
     Match.anyValue());
 });
@@ -111,7 +110,7 @@ test.skip('Infrastructure Configuration DependsOn Instance Profile', () => {
 });
 
 test('Image Builder Component is created', () => {
-  template.resourceCountIs('AWS::ImageBuilder::Component', 1);
+  template.resourceCountIs('AWS::ImageBuilder::Component', props.componentDocuments.length);
 });
 
 test('Image Recipe is created', () => {

test/test_component_example_2.yml

@@ -0,0 +1,28 @@
+name: ComponentExample
+description: This is an example component document
+schemaVersion: 1.0
+
+phases:
+  - name: build
+    steps:
+      - name: InstallUpdates
+        action: UpdateOS
+      - name: DownloadSomeShellScriptFromS3
+        action: S3Download
+        inputs:
+          - source: 's3://some-bucket/some_shell_script.sh'
+            destination: '/tmp/some_shell_script.sh'
+  - name: validate
+    steps:
+      - name: HelloWorldStep
+        action: ExecuteBash
+        inputs:
+          commands:
+            - echo "Hello World! Validate."
+  - name: test
+    steps:
+      - name: HelloWorldStep
+        action: ExecuteBash
+        inputs:
+          commands:
+            - echo "Hello World! Test.