apitools/gen/client_generation_test.py

#
# Copyright 2015 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Test gen_client against all the APIs we use regularly."""

import importlib
import logging
import os
import six
import subprocess
import sys
import tempfile

from apitools.gen import gen_client
from apitools.gen import test_utils

if six.PY2:
    import unittest2 as unittest
else:
    import unittest

_API_LIST = [
    'bigquery.v2',
    'compute.v1',
    'drive.v3',
    'storage.v1',
]


class ClientGenerationTest(unittest.TestCase):

    def setUp(self):
        super(ClientGenerationTest, self).setUp()
        self.gen_client_binary = 'gen_client'

    @test_utils.SkipOnWindows
    def testGeneration(self):
        for api in _API_LIST:
            with test_utils.TempDir(change_to=True):
                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 = gen_client.main(args)
                if retcode == 128:
                    logging.error('Failed to fetch discovery doc, continuing.')
                    continue
                self.assertEqual(0, retcode)

                sys.path.insert(0, os.path.join(os.getcwd(), 'generated'))
                # Ensure we can import the generated client.
                importlib.import_module('{}_{}_client'.format(
                    *api.split('.')))