def _SetupMockXpnClient(self):
self.mock_client = mock.Client(
core_apis.GetClientClass('compute', _DEFAULT_API_VERSION),
real_client=core_apis.GetClientInstance('compute',
_DEFAULT_API_VERSION,
no_http=True))
self.mock_client.Mock()
self.addCleanup(self.mock_client.Unmock)
self.messages = self.mock_client.MESSAGES_MODULE
self.xpn_client = xpn_api.XpnClient(
compute_client=client_adapter.ClientAdapter(
api_version=_DEFAULT_API_VERSION, client=self.mock_client))
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 Start(self):
"""Starts this mock."""
self.resources = resources.REGISTRY.Clone()
self.resources.RegisterApiByName('compute', self.api_version)
self.adapter = client_adapter.ClientAdapter(self.api_version, no_http=True)
self.messages = self.adapter.messages
self.batch_responder = BatchResponder()
self.adapter.BatchRequests = self.batch_responder.BatchRequests
make_requests_patch = mock.patch.object(self.adapter, 'MakeRequests')
self._cleanups.append(make_requests_patch.stop)
self.make_requests = make_requests_patch.start()
adapter_patch = mock.patch(
'googlecloudsdk.api_lib.compute.client_adapter.ClientAdapter',
return_value=self.adapter)
self._cleanups.append(adapter_patch.stop)
adapter_patch.start()
return self
def SetUp(self):
self.api_version = 'v1'
self.client_class = core_apis.GetClientClass('compute',
self.api_version)
self.messages = self.client_class.MESSAGES_MODULE
status_enum = self.messages.Operation.StatusValueValuesEnum
self.status_done = status_enum.DONE
self.status_pending = status_enum.PENDING
self.error_entry = self.messages.Operation.ErrorValue.ErrorsValueListEntry
self.operation_x_ref = resources.REGISTRY.Create(
_GLOBAL_OPERATIONS_COLLECTION,
project='mickey',
operation='operationX')
self.operation_y_ref = resources.REGISTRY.Create(
_GLOBAL_OPERATIONS_COLLECTION,
project='mickey',
operation='operationY')
self.instance_x_ref = resources.REGISTRY.Create(
'compute.instances',
project='mickey',
zone='disney',
instance='Super-Cheese-X')
self.instance_y_ref = resources.REGISTRY.Create(
'compute.instances',
project='mickey',
zone='disney',
instance='Super-Cheese-Y')
self.adapter = client_adapter.ClientAdapter(self.api_version,
no_http=True)
self.batch_fake = waiter_test_base.OperationBatchFake(
self.adapter.apitools_client.instances, 'compute.instances',
self.adapter.apitools_client.globalOperations)
def client(self):
"""Specifies the compute client."""
if self._client is None:
self._client = client_adapter.ClientAdapter(self._api_version)
return self._client
def GetXpnClient(api_version=_DEFAULT_API_VERSION):
return XpnClient(client_adapter.ClientAdapter(api_version))
def SetUp(self):
self.adapter = client_adapter.ClientAdapter('v1', no_http=True)
self.client = self.adapter.apitools_client
self.messages = self.client.MESSAGES_MODULE
def SetUp(self):
self.client = client_adapter.ClientAdapter('v1', no_http=True)
self.messages = core_apis.GetMessagesModule('compute', 'v1')
self.scopes = [scope.ScopeEnum.ZONE]
def GetXpnClient():
return XpnClient(client_adapter.ClientAdapter(_API_VERSION))
