fix(codebuild): add validation for source when badge is enabled

packages/@aws-cdk/aws-codebuild/lib/project.ts

@@ -644,8 +644,11 @@ export class Project extends ProjectBase {
     }
 
     // Render the source and add in the buildspec
-
     const renderSource = () => {
+      if (props.badge && !this.source.isBadgeSupported()) {
+        throw new Error(`Badge is not supported for source type ${this.source.type}`);
+      }
+
       const sourceJson = this.source.toSourceJSON();
       if (typeof buildSpec === 'string') {
         return {

packages/@aws-cdk/aws-codebuild/lib/source.ts

@@ -52,6 +52,10 @@ export abstract class BuildSource {
     return undefined;
   }
 
+  public isBadgeSupported(): boolean {
+    return false;
+  }
+
   protected toSourceProperty(): any {
     return {
     };
@@ -103,6 +107,10 @@ export abstract class GitBuildSource extends BuildSource {
       gitCloneDepth: this.cloneDepth
     };
   }
+
+  public isBadgeSupported(): boolean {
+    return true;
+  }
 }
 
 /**
@@ -134,6 +142,10 @@ export class CodeCommitSource extends GitBuildSource {
       .addResource(this.repo.repositoryArn));
   }
 
+  public isBadgeSupported(): boolean {
+    return false;
+  }
+
   protected toSourceProperty(): any {
     return {
       location: this.repo.repositoryCloneUrlHttp

packages/@aws-cdk/aws-codebuild/test/test.codebuild.ts

@@ -1046,6 +1046,40 @@ export = {
       });
     }, /Windows images do not support the Small ComputeType/);
 
+    test.done();
+  },
+
+  'badge support test'(test: Test) {
+    const stack = new cdk.Stack();
+
+    interface BadgeValidationTestCase {
+      source: codebuild.BuildSource,
+      shouldPassValidation: boolean
+    }
+
+    const repo = new codecommit.Repository(stack, 'MyRepo', { repositoryName: 'hello-cdk' });
+    const bucket = new s3.Bucket(stack, 'MyBucket');
+
+    const cases: BadgeValidationTestCase[] = [
+      { source: new codebuild.NoSource(), shouldPassValidation: false },
+      { source: new codebuild.CodePipelineSource(), shouldPassValidation: false },
+      { source: new codebuild.CodeCommitSource({ repository: repo }), shouldPassValidation: false },
+      { source: new codebuild.S3BucketSource({ bucket, path: 'path/to/source.zip' }), shouldPassValidation: false },
+      { source: new codebuild.GitHubSource({ owner: 'awslabs', repo: 'aws-cdk', oauthToken: new cdk.SecretValue()}), shouldPassValidation: true },
+      { source: new codebuild.GitHubEnterpriseSource({ httpsCloneUrl: 'url', oauthToken: new cdk.SecretValue()}), shouldPassValidation: true },
+      { source: new codebuild.BitBucketSource({ owner: 'awslabs', repo: 'aws-cdk' }), shouldPassValidation: true }
+    ];
+
+    cases.forEach(testCase => {
+      const source = testCase.source;
+      const validationBlock = () => { new codebuild.Project(stack, `MyProject-${source.type}`, { source, badge: true }); };
+      if (testCase.shouldPassValidation) {
+        test.doesNotThrow(validationBlock, Error, `Badge is not supported for source type ${source.type}`);
+      } else {
+        test.throws(validationBlock, Error, `Badge is not supported for source type ${source.type}`);
+      }
+    });
+
     test.done();
   }
 };