[catapult] - Copy Telemetry's gsutilz over to third_party.

third_party/gsutil/third_party/apitools/apitools/gen/client_generation_test.py

Index: third_party/gsutil/third_party/apitools/apitools/gen/client_generation_test.py
diff --git a/third_party/gsutil/third_party/apitools/apitools/gen/client_generation_test.py b/third_party/gsutil/third_party/apitools/apitools/gen/client_generation_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..f34d95459dadaf25cd7e6ad1678fe749b9477d04
--- /dev/null
+++ b/third_party/gsutil/third_party/apitools/apitools/gen/client_generation_test.py
@@ -0,0 +1,72 @@
+"""Test gen_client against all the APIs we use regularly."""
+
+import contextlib
+import logging
+import os
+import shutil
+import subprocess
+import sys
+import tempfile
+
+
+import unittest2
+
+_API_LIST = [
+    'drive.v2',
+    'bigquery.v2',
+    'compute.v1',
+    'storage.v1',
+]
+
+
+@contextlib.contextmanager
+def TempDir():
+    original_dir = os.getcwd()
+    path = tempfile.mkdtemp()
+    try:
+        os.chdir(path)
+        yield path
+    finally:
+        os.chdir(original_dir)
+        shutil.rmtree(path)
+
+
+class ClientGenerationTest(unittest2.TestCase):
+
+    def setUp(self):
+        super(ClientGenerationTest, self).setUp()
+        self.gen_client_binary = 'gen_client'
+
+    # unittest in 2.6 doesn't have skipIf.
+    @unittest2.skipUnless(sys.version_info[0] == 2 and
+                          sys.version_info[1] == 7,
+                          'Only runs in Python 2.7')
+    def testGeneration(self):
+        for api in _API_LIST:
+            with TempDir():
+                args = [
+                    self.gen_client_binary,
+                    '--client_id=12345',
+                    '--client_secret=67890',
+                    '--discovery_url=%s' % api,
+                    '--outdir=generated',
+                    '--overwrite',
+                    'client',
+                ]
+                logging.info('Testing API %s with command line: %s',
+                             api, ' '.join(args))
+                retcode = subprocess.call(args)
+                if retcode == 128:
+                    logging.error('Failed to fetch discovery doc, continuing.')
+                    continue
+                self.assertEqual(0, retcode)
+
+                with tempfile.NamedTemporaryFile() as out:
+                    cmdline_args = [
+                        os.path.join(
+                            'generated', api.replace('.', '_') + '.py'),
+                        'help',
+                    ]
+                    retcode = subprocess.call(cmdline_args, stdout=out)
+                # appcommands returns 1 on help
+                self.assertEqual(1, retcode)