From e513ec10daa79c137b06e917b3a523f654548701 Mon Sep 17 00:00:00 2001
From: Matt Clay <matt@mystile.com>
Date: Thu, 31 Aug 2017 17:44:33 -0700
Subject: [PATCH] Add safety check for Shippable "Rebuild with SSH" (#28857)

* Add safety check for Shippable "Rebuild with SSH".
* Run shippable.py for the `other` test.
---
 test/runner/shippable.py      | 103 ++++++++++++++++++++++++++++++++++
 test/utils/shippable/other.sh |   2 +
 2 files changed, 105 insertions(+)
 create mode 100755 test/runner/shippable.py

diff --git a/test/runner/shippable.py b/test/runner/shippable.py
new file mode 100755
index 0000000000..c9526818d2
--- /dev/null
+++ b/test/runner/shippable.py
@@ -0,0 +1,103 @@
+#!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
+"""Verify the current Shippable run has the required number of jobs."""
+
+from __future__ import absolute_import, print_function
+
+# noinspection PyCompatibility
+import argparse
+import errno
+import os
+import sys
+
+from lib.http import (
+    HttpClient,
+)
+
+from lib.util import (
+    display,
+    ApplicationError,
+    ApplicationWarning,
+    MissingEnvironmentVariable,
+)
+
+
+try:
+    import argcomplete
+except ImportError:
+    argcomplete = None
+
+
+def main():
+    """Main program function."""
+    try:
+        args = parse_args()
+        display.verbosity = args.verbosity
+        display.color = args.color
+
+        try:
+            run_id = os.environ['SHIPPABLE_BUILD_ID']
+        except KeyError as ex:
+            raise MissingEnvironmentVariable(ex.args[0])
+
+        client = HttpClient(args)
+        response = client.get('https://api.shippable.com/jobs?runIds=%s' % run_id)
+        jobs = response.json()
+
+        if len(jobs) == 1:
+            raise ApplicationError('Shippable run %s has only one job. Did you use the "Rebuild with SSH" option?' % run_id)
+    except ApplicationWarning as ex:
+        display.warning(str(ex))
+        exit(0)
+    except ApplicationError as ex:
+        display.error(str(ex))
+        exit(1)
+    except KeyboardInterrupt:
+        exit(2)
+    except IOError as ex:
+        if ex.errno == errno.EPIPE:
+            exit(3)
+        raise
+
+
+def parse_args():
+    """Parse command line arguments."""
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('-e', '--explain',
+                        action='store_true',
+                        help='explain commands that would be executed')
+
+    parser.add_argument('-v', '--verbose',
+                        dest='verbosity',
+                        action='count',
+                        default=0,
+                        help='display more output')
+
+    parser.add_argument('--color',
+                        metavar='COLOR',
+                        nargs='?',
+                        help='generate color output: %(choices)s',
+                        choices=('yes', 'no', 'auto'),
+                        const='yes',
+                        default='auto')
+
+    if argcomplete:
+        argcomplete.autocomplete(parser)
+
+    args = parser.parse_args()
+
+    if args.color == 'yes':
+        args.color = True
+    elif args.color == 'no':
+        args.color = False
+    elif 'SHIPPABLE' in os.environ:
+        args.color = True
+    else:
+        args.color = sys.stdout.isatty()
+
+    return args
+
+
+if __name__ == '__main__':
+    main()
diff --git a/test/utils/shippable/other.sh b/test/utils/shippable/other.sh
index 3f5b75316d..c5b2fd0073 100755
--- a/test/utils/shippable/other.sh
+++ b/test/utils/shippable/other.sh
@@ -2,6 +2,8 @@
 
 set -o pipefail
 
+shippable.py
+
 retry.py apt-get update -qq
 retry.py apt-get install -qq \
     shellcheck \