def _SetUp(self, release_track):
"""Setup common test components.
Args:
release_track: Release track the test is targeting.
"""
api_name = self._GetApiName(release_track)
self.track = release_track
self.base_uri = 'https://compute.googleapis.com/compute/{}/projects/{}'.format(
api_name, self.Project())
apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance('compute',
api_name,
no_http=True))
apitools_client.Mock()
self.addCleanup(apitools_client.Unmock)
self.messages = apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
self.service = apitools_client.vpnGateways
self.region_operations = apitools_client.regionOperations
self.StartPatch('time.sleep')
def __init__(self, release_track):
api_version = VERSION_MAP[release_track]
self.release_track = release_track
self.alloydb_client = apis.GetClientInstance('alloydb', api_version)
self.alloydb_messages = self.alloydb_client.MESSAGES_MODULE
self.resource_parser = resources.Registry()
self.resource_parser.RegisterApiByName('alloydb', api_version)
def _SetUp(self, release_track):
"""Setup common test components.
Args:
release_track: Release track the test is targeting.
"""
api_name = 'v1'
if release_track == calliope_base.ReleaseTrack.ALPHA:
api_name = 'alpha'
elif release_track == calliope_base.ReleaseTrack.BETA:
api_name = 'beta'
self.apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance('compute',
api_name,
no_http=True))
self.apitools_client.Mock()
self.addCleanup(self.apitools_client.Unmock)
self.messages = self.apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
self.track = release_track
self.zone_operations = self.apitools_client.zoneOperations
self.region_operations = getattr(self.apitools_client,
'regionOperations', None)
self.StartPatch('time.sleep')
def __init__(self, api_version):
self.sql_client = apis.GetClientInstance('sql', api_version)
self.sql_messages = self.sql_client.MESSAGES_MODULE
self.resource_parser = resources.Registry()
self.resource_parser.RegisterApiByName('sql', api_version)
self.resource_parser.SetParamDefault(
api='sql', collection=None, param='project',
resolver=resolvers.FromProperty(properties.VALUES.core.project))
def __init__(self, releaseTrack):
self._client = GetClientInstance(releaseTrack)
self._endpointClient = self._client.projects_locations_endpoints
self._operationsClient = self._client.projects_locations_operations
self._locationsClient = self._client.projects_locations
self._messages = GetMessagesModule(releaseTrack)
self._resource_parser = resources.Registry()
self._resource_parser.RegisterApiByName('ids',
VERSION_MAP.get(releaseTrack))
def SetUp(self):
self.parser = util.ArgumentParser()
self.parser.add_argument(
'--project',
metavar='PROJECT_ID',
dest='project',
help='Google Cloud Platform project ID to use for this invocation.',
action=actions.StoreProperty(properties.VALUES.core.project))
self.registry = resources.Registry()
def SetUp(self):
self.mock_api_adapter = self.StartPatch(
'googlecloudsdk.api_lib.container.hub.gkehub_api_adapter.APIAdapter'
)()
self.mock_kubernetes_client = self.StartPatch(
'googlecloudsdk.command_lib.container.hub.kube_util.KubernetesClient'
)()
self.parser = test_util.ArgumentParser()
register.Register.Args(self.parser)
self.registry = resources.Registry()
def testGenerateClusterUrl(self):
registry = resources.Registry()
cluster_ref = registry.Parse(
'some-cluster-name',
params={
'projectId': 'some-project-id',
'zone': 'some-zone',
},
collection='container.projects.zones.clusters')
self.assertEqual(
'https://console.cloud.google.com/kubernetes/'
'workload_/gcloud/some-zone/some-cluster-name?'
'project=some-project-id', c_util.GenerateClusterUrl(cluster_ref))
def SetUp(self):
api_name = self._GetApiName()
self.apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance(
'compute', api_name, no_http=True))
self.apitools_client.Mock()
self.addCleanup(self.apitools_client.Unmock)
self.messages = self.apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
def SetUp(self):
self.apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', self.api),
real_client=core_apis.GetClientInstance(
'compute', self.api, no_http=True))
self.apitools_client.Mock()
self.addCleanup(self.apitools_client.Unmock)
self.messages = self.apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', self.api)
self.service = self.apitools_client.instances
self.zone_operations = self.apitools_client.zoneOperations
def SetUp(self):
self.parser = util.ArgumentParser()
self.parser.add_argument(
'--project',
metavar='PROJECT_ID',
dest='project',
help='Google Cloud Platform project ID to use for this invocation.',
action=actions.StoreProperty(properties.VALUES.core.project))
self.registry = resources.Registry()
self.compute_client = client_adapter.ClientAdapter(api_version='v1')
self.client = self.compute_client.apitools_client
self.messages = self.compute_client.messages
self.StartObjectPatch(console_io, 'CanPrompt', return_value=True)
self.StartObjectPatch(console_io, 'IsInteractive', return_value=True)
def SetUp(self):
api_name = 'beta'
self.apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance(
'compute', api_name, no_http=True))
self.apitools_client.Mock()
self.addCleanup(self.apitools_client.Unmock)
self.messages = self.apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
self.track = calliope_base.ReleaseTrack.BETA
self.global_operations = self.apitools_client.globalOperations
self.region_operations = self.apitools_client.regionOperations
self.StartPatch('time.sleep')
def SetUp(self):
api_version = 'v1'
self.apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_version),
real_client=core_apis.GetClientInstance('compute',
api_version,
no_http=True))
self.apitools_client.Mock()
self.addCleanup(self.apitools_client.Unmock)
self.messages = self.apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_version)
self.track = calliope_base.ReleaseTrack.GA
self.service = self.apitools_client.instances
self.zone_operations = self.apitools_client.zoneOperations
def SetUp(self):
"""Setup common test components."""
api_name = self._GetApiName()
apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance('compute',
api_name,
no_http=True))
apitools_client.Mock()
self.addCleanup(apitools_client.Unmock)
self.messages = apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
self.service = apitools_client.externalVpnGateways
self.global_operations = apitools_client.globalOperations
def SetUp(self):
"""Setup common test components."""
api_name = self._GetApiName()
self.base_uri = 'https://compute.googleapis.com/compute/{}/projects/{}'.format(
api_name, self.Project())
apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance('compute',
api_name,
no_http=True))
apitools_client.Mock()
self.addCleanup(apitools_client.Unmock)
self.messages = apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
self.service = apitools_client.vpnTunnels
self.region_operations = apitools_client.regionOperations
def _SetUp(self, release_track):
"""Setup common test components.
Args:
release_track: Release track the test is targetting.
"""
api_name = self._GetApiName(release_track)
self.track = release_track
apitools_client = api_mock.Client(
core_apis.GetClientClass('compute', api_name),
real_client=core_apis.GetClientInstance('compute',
api_name,
no_http=True))
apitools_client.Mock()
self.addCleanup(apitools_client.Unmock)
self.messages = apitools_client.MESSAGES_MODULE
self.resources = resources.Registry()
self.resources.RegisterApiByName('compute', api_name)
self.service = apitools_client.sslPolicies
self.global_operations = apitools_client.globalOperations
def setUp(self): # pylint:disable=g-missing-super-call
self.registry = resources.Registry()
self.compute_uri = 'https://compute.googleapis.com/compute/v1'
self.project = 'long-island'
self.zone = 'local-group'
def __init__(self, api_version):
self.lux_client = apis.GetClientInstance('luxadmin', api_version)
self.lux_messages = self.lux_client.MESSAGES_MODULE
self.resource_parser = resources.Registry()
self.resource_parser.RegisterApiByName('luxadmin', api_version)
def GetResourceParser(api_version=_DEFAULT_API_VERSION):
resource_parser = resources.Registry()
resource_parser.RegisterApiByName('datastream', api_version)
return resource_parser
def GetResourceParser(release_track):
resource_parser = resources.Registry()
resource_parser.RegisterApiByName('datamigration',
GetApiVersion(release_track))
return resource_parser
