CI: Allow passing environment to test containers

Signed-off-by: mdouglas47 <mdouglas47@bloomberg.net>

Author: morgando

tests/docker/runit

@@ -4,6 +4,7 @@ set -e
 nodeprefix="m"
 
 script_args=()
+env_vars=()
 while [[ $1 = -* ]]; do
     if [[ "$1" = "-s" ]] || [[ "$1" = "--source" ]]; then
         shift
@@ -23,6 +24,9 @@ while [[ $1 = -* ]]; do
     elif [[ "$1" = "-a" ]] || [[ "$1" = "--file-arg" ]]; then
         shift
         script_args+=("$1")
+    elif [[ "$1" = "-e" ]] || [[ "$1" = "--env" ]]; then
+        shift
+        env_vars+=("$1")   # expect VAR=VALUE
     fi
     shift
 done
@@ -73,6 +77,20 @@ echo $cluster >common/cluster
 ssh-keygen -N "" -f common/id_rsa
 
 compose=docker-compose.yml
+
+append_env_to_compose() {
+    if ((${#env_vars[@]} > 0)); then
+        cat >> $compose <<EOF
+            environment:
+EOF
+        for e in "${env_vars[@]}"; do
+            cat >> $compose <<EOF
+            - ${e}
+EOF
+        done
+    fi
+}
+
 cat > $compose <<EOF
 version: '2'
 
@@ -104,6 +122,8 @@ EOF
 EOF
 }
 
+append_env_to_compose
+
 cat >> $compose <<EOF
         command: /client $commands
         cap_add:
@@ -140,6 +160,9 @@ EOF
 EOF
 
 }
+
+append_env_to_compose
+
 cat >> $compose <<EOF
         command: /server
         cap_add: