def GetApiVersionForTrack(release_track=None):
if not release_track:
return core_apis.ResolveVersion(GKEHUB_API_NAME)
elif release_track == base.ReleaseTrack.ALPHA:
return GKEHUB_ALPHA_API_VERSION
elif release_track == base.ReleaseTrack.BETA:
return GKEHUB_BETA_API_VERSION
elif release_track == base.ReleaseTrack.GA:
return GKEHUB_GA_API_VERSION
return core_apis.ResolveVersion(GKEHUB_API_NAME)
def testResolveVersionWithOverride(self):
overridden_version = 'alpha'
properties.VALUES.api_client_overrides.compute.Set(overridden_version)
resolved_version = apis.ResolveVersion('compute')
self.assertEqual(overridden_version, resolved_version)
overridden_version = 'v1beta4'
properties.VALUES.api_client_overrides.sql.Set(overridden_version)
resolved_version = apis.ResolveVersion('sql')
self.assertEqual(overridden_version, resolved_version)
def testResolveVersionWithAPIAndVersionOverride(self):
# API specific override should take precedence over full surface override.
overridden_version = 'v1_staging'
overridden_api_version = 'alpha_staging'
properties.VALUES.api_client_overrides.compute.Set(overridden_version)
properties.VALUES.api_client_overrides.compute_alpha.Set(
overridden_api_version)
resolved_version = apis.ResolveVersion('compute', api_version='alpha')
self.assertEqual(overridden_api_version, resolved_version)
def __init__(self, api_default_version='v1', no_http=False, client=None):
self._api_version = core_apis.ResolveVersion(
self._API_NAME, api_default_version)
self._client = client or core_apis.GetClientInstance(
self._API_NAME, self._api_version, no_http=no_http)
# Turn the endpoint into just the host.
# eg. https://www.googleapis.com/compute/v1 -> https://www.googleapis.com
endpoint_url = core_apis.GetEffectiveApiEndpoint(
self._API_NAME, self._api_version)
self._batch_url = _GetBatchUrl(endpoint_url, self._api_version)
def SetUp(self):
self.mock_old_kubernetes_client = self.StartPatch(
'googlecloudsdk.command_lib.container.hub.kube_util.OldKubernetesClient'
)()
self.mock_kubernetes_client = self.StartPatch(
'googlecloudsdk.command_lib.container.hub.kube_util.KubernetesClient'
)()
compute_api = 'compute'
compute_api_version = core_apis.ResolveVersion(compute_api)
self.compute_messages = core_apis.GetMessagesModule(
compute_api, compute_api_version)
self.mock_compute_client = apimock.Client(
client_class=core_apis.GetClientClass(compute_api,
compute_api_version))
self.mock_compute_client.Mock()
self.addCleanup(self.mock_compute_client.Unmock)
self.StartPatch(
'googlecloudsdk.command_lib.container.hub.api_util._ComputeClient'
).return_value = self.mock_compute_client
def ListMemberships(project):
"""Lists Membership IDs in Hub.
Args:
project: the project in which Membership resources exist.
Returns:
a list of Membership resource IDs in Hub.
Raises:
apitools.base.py.HttpError: if the request returns an HTTP error
"""
parent = 'projects/{}/locations/global'.format(project)
api_version = core_apis.ResolveVersion('gkehub')
client = core_apis.GetClientInstance('gkehub', api_version)
response = client.projects_locations_memberships.List(
client.MESSAGES_MODULE.GkehubProjectsLocationsMembershipsListRequest(
parent=parent))
return [
os.path.basename(membership.name) for membership in response.resources
]
def _GetPubsubMessages():
return apis.GetMessagesModule('pubsub', apis.ResolveVersion('pubsub'))
def GetApiClientForApiVersion(api_version=None):
if not api_version:
api_version = core_apis.ResolveVersion(GKEHUB_API_NAME)
return core_apis.GetClientInstance(GKEHUB_API_NAME, api_version)
def GetMessages(version=None):
"""Get the default messages module."""
return apis.GetMessagesModule('secretmanager', version or
apis.ResolveVersion('secretmanager'))
def GetClient(version=None):
"""Get the default client."""
return apis.GetClientInstance('secretmanager', version or
apis.ResolveVersion('secretmanager'))
def _MembershipClient():
api_version = core_apis.ResolveVersion('gkehub')
return core_apis.GetClientInstance('gkehub', api_version)
def _GetDefaultVersion(): return apis.ResolveVersion(GAME_SERVICES_API)
def testResolveVersionWithDefaultOverride(self):
resolved_version = apis.ResolveVersion('compute', 'hello')
self.assertEqual('hello', resolved_version)
def testResolveVersionWithoutOverride(self):
resolved_version = apis.ResolveVersion('compute')
self.assertEqual('v1', resolved_version)
def testResolveVersionWithVersionOverride(self):
overridden_version = 'staging_alpha'
properties.VALUES.api_client_overrides.compute_alpha.Set(
overridden_version)
resolved_version = apis.ResolveVersion('compute', api_version='alpha')
self.assertEqual(overridden_version, resolved_version)
def _ComputeClient():
api_version = core_apis.ResolveVersion('compute')
return core_apis.GetClientInstance('compute', api_version)
def _GetDefaultVersion(): return apis.ResolveVersion(_API_NAME)
