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))