Pubsub in ci (#175)

* run pubsub sample in codebuild CI for both mqtt direct and websockets

Author: bretambrose

codebuild/samples/linux-smoke-tests.yml

@@ -0,0 +1,26 @@
+version: 0.2
+#this build spec assumes the ubuntu aws/codebuild/java:openjdk-8 image
+phases:
+  install:
+    commands:
+      - sudo add-apt-repository ppa:openjdk-r/ppa
+      - sudo add-apt-repository ppa:ubuntu-toolchain-r/test
+      - sudo apt-get update -y
+      - curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "/tmp/awscliv2.zip"
+      - unzip -q -d /tmp /tmp/awscliv2.zip
+      - sudo /tmp/aws/install
+  build:
+    commands:
+      - echo Build started on `date`
+      - $CODEBUILD_SRC_DIR/codebuild/samples/setup-linux.sh
+      - $CODEBUILD_SRC_DIR/codebuild/samples/pubsub-linux.sh
+  post_build:
+    commands:
+      - echo Build completed on `date`
+
+artifacts:
+  discard-paths: yes
+  files:
+    - "target/surefire-reports/**"
+    - "hs_err_pid*"
+    - "core*"

codebuild/samples/pubsub-linux.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+set -e
+
+env
+
+pushd $CODEBUILD_SRC_DIR/samples/BasicPubSub
+
+ENDPOINT=$(aws secretsmanager get-secret-value --secret-id "unit-test/endpoint" --query "SecretString" | cut -f2 -d":" | sed -e 's/[\\\"\}]//g')
+
+mvn compile
+
+echo "Mqtt Direct test"
+mvn exec:java -Dexec.mainClass="pubsub.PubSub" -Daws.crt.ci="True" -Dexec.arguments="--endpoint,$ENDPOINT,--key,/tmp/privatekey.pem,--cert,/tmp/certificate.pem"
+
+echo "Websocket test"
+mvn exec:java -Dexec.mainClass="pubsub.PubSub" -Daws.crt.ci="True" -Dexec.arguments="--endpoint,$ENDPOINT,--websockets,--region,us-east-1,--port,443"
+
+popd

codebuild/samples/setup-linux.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+set -e
+
+env
+
+# build java package
+cd $CODEBUILD_SRC_DIR
+
+ulimit -c unlimited
+mvn compile
+mvn install -DskipTests=true
+
+cert=$(aws secretsmanager get-secret-value --secret-id "unit-test/certificate" --query "SecretString" | cut -f2 -d":" | cut -f2 -d\") && echo -e "$cert" > /tmp/certificate.pem
+key=$(aws secretsmanager get-secret-value --secret-id "unit-test/privatekey" --query "SecretString" | cut -f2 -d":" | cut -f2 -d\") && echo -e "$key" > /tmp/privatekey.pem
+
+

samples/BasicPubSub/src/main/java/pubsub/PubSub.java

@@ -29,6 +29,13 @@
 import java.util.concurrent.ExecutionException;
 
 public class PubSub {
+
+    // When run normally, we want to exit nicely even if something goes wrong
+    // When run from CI, we want to let an exception escape which in turn causes the
+    // exec:java task to return a non-zero exit code
+    static String ciPropValue = System.getProperty("aws.crt.ci");
+    static boolean isCI = ciPropValue != null && Boolean.valueOf(ciPropValue);
+
     static String clientId = "test-" + UUID.randomUUID().toString();
     static String rootCaPath;
     static String certPath;
@@ -139,6 +146,7 @@ static void parseCommandLine(String[] args) {
                     }
                     break;
                 case "-w":
+                case "--websockets":
                     useWebsockets = true;
                     break;
                 case "--x509":
@@ -200,21 +208,37 @@ static void onRejectedError(RejectedError error) {
         System.out.println("Request rejected: " + error.code.toString() + ": " + error.message);
     }
 
+    /*
+     * When called during a CI run, throw an exception that will escape and fail the exec:java task
+     * When called otherwise, print what went wrong (if anything) and just continue (return from main)
+     */
+    static void onApplicationFailure(Throwable cause) {
+        if (isCI) {
+            throw new RuntimeException("BasicPubSub execution failure", cause);
+        } else if (cause != null) {
+            System.out.println("Exception encountered: " + cause.toString());
+        }
+    }
+
     public static void main(String[] args) {
+
         parseCommandLine(args);
         if (showHelp || endpoint == null) {
             printUsage();
+            onApplicationFailure(null);
             return;
         }
 
         if (!useWebsockets) {
             if (certPath == null || keyPath == null) {
                 printUsage();
+                onApplicationFailure(null);
                 return;
             }
         } else if (useX509Credentials) {
             if (x509RoleAlias == null || x509Endpoint == null || x509Thing == null || x509CertPath == null || x509KeyPath == null) {
                 printUsage();
+                onApplicationFailure(null);
                 return;
             }
         }
@@ -318,7 +342,7 @@ public void onConnectionResumed(boolean sessionPresent) {
                 disconnected.get();
             }
         } catch (CrtRuntimeException | InterruptedException | ExecutionException ex) {
-            System.out.println("Exception encountered: " + ex.toString());
+            onApplicationFailure(ex);
         }
 
         CrtResource.waitForNoResources();