OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	02/11/2025 08:19:48 AM	rwxr-xr-x
📄 __init__.py	715 bytes	10/23/2024 10:32:22 PM	rw-r--r--
📄 client_generation_test.py	2.1 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 extended_descriptor.py	21.15 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 gen_client.py	11.42 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 gen_client_lib.py	10.29 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 gen_client_test.py	6.09 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 message_registry.py	21 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 service_registry.py	22.51 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 test_utils.py	1.41 KB	10/23/2024 10:32:22 PM	rw-r--r--
📁 testdata	-	10/23/2024 10:32:22 PM	rwxr-xr-x
📄 util.py	14.93 KB	10/23/2024 10:32:22 PM	rw-r--r--
📄 util_test.py	2.91 KB	10/23/2024 10:32:22 PM	rw-r--r--

Editing: gen_client_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 for gen_client module."""

import os
import unittest

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

def GetTestDataPath(*path):
    return os.path.join(os.path.dirname(__file__), 'testdata', *path)

def _GetContent(file_path):
    with open(file_path) as f:
        return f.read()

class ClientGenCliTest(unittest.TestCase):

def testHelp_NotEnoughArguments(self):
        with self.assertRaisesRegex(SystemExit, '0'):
            with test_utils.CaptureOutput() as (_, err):
                gen_client.main([gen_client.__file__, '-h'])
                err_output = err.getvalue()
                self.assertIn('usage:', err_output)
                self.assertIn('error: too few arguments', err_output)

def testGenClient_SimpleDocNoInit(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--init-file', 'none',
                '--infile', GetTestDataPath('dns', 'dns_v1.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--root_package', 'google.apis',
                'client'
            ])
            expected_files = (
                set(['dns_v1_client.py', 'dns_v1_messages.py']))
            self.assertEqual(expected_files, set(os.listdir(tmp_dir_path)))

def testGenClient_SimpleDocEmptyInit(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--init-file', 'empty',
                '--infile', GetTestDataPath('dns', 'dns_v1.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--root_package', 'google.apis',
                'client'
            ])
            expected_files = (
                set(['dns_v1_client.py', 'dns_v1_messages.py', '__init__.py']))
            self.assertEqual(expected_files, set(os.listdir(tmp_dir_path)))
            init_file = _GetContent(os.path.join(tmp_dir_path, '__init__.py'))
            self.assertEqual("""\"""Package marker file.\"""

from __future__ import absolute_import

import pkgutil

__path__ = pkgutil.extend_path(__path__, __name__)
""", init_file)

def testGenClient_SimpleDocWithV4(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--infile', GetTestDataPath('dns', 'dns_v1.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--apitools_version', '0.4.12',
                '--root_package', 'google.apis',
                'client'
            ])
            self.assertEqual(
                set(['dns_v1_client.py', 'dns_v1_messages.py', '__init__.py']),
                set(os.listdir(tmp_dir_path)))

def testGenClient_SimpleDocWithV5(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--infile', GetTestDataPath('dns', 'dns_v1.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--apitools_version', '0.5.0',
                '--root_package', 'google.apis',
                'client'
            ])
            self.assertEqual(
                set(['dns_v1_client.py', 'dns_v1_messages.py', '__init__.py']),
                set(os.listdir(tmp_dir_path)))

def testGenClient_ApiVersioning(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--infile', GetTestDataPath(
                    'dns', 'dns_2015-08-07-preview.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--version-identifier', 'v2015_08_07_preview',
                '--root_package', 'google.apis',
                'client'
            ])
            self.assertEqual(
                set([
                    'dns_v2015_08_07_preview_client.py',
                    'dns_v2015_08_07_preview_messages.py',
                    '__init__.py']),
                set(os.listdir(tmp_dir_path)))
            client_file = _GetContent(
                os.path.join(tmp_dir_path, 'dns_v2015_08_07_preview_client.py'))
            # Check that "apiVersion" system parameter values from discovery doc
            # appear in generated client.
            self.assertIn('2015-01-01-preview', client_file)
            self.assertIn('2015-02-02-preview', client_file)
            self.assertIn('2015-03-03-preview', client_file)

def testGenPipPackage_SimpleDoc(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--infile', GetTestDataPath('dns', 'dns_v1.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--root_package', 'google.apis',
                'pip_package'
            ])
            self.assertEqual(
                set(['apitools', 'setup.py']),
                set(os.listdir(tmp_dir_path)))

def testGenProto_SimpleDoc(self):
        with test_utils.TempDir() as tmp_dir_path:
            gen_client.main([
                gen_client.__file__,
                '--infile', GetTestDataPath('dns', 'dns_v1.json'),
                '--outdir', tmp_dir_path,
                '--overwrite',
                '--root_package', 'google.apis',
                'proto'
            ])
            self.assertEqual(
                set(['dns_v1_messages.proto', 'dns_v1_services.proto']),
                set(os.listdir(tmp_dir_path)))