def testNeq(self): client1 = object() client2 = object() service = object() regional_lister1 = lister.RegionalLister(client1, service) regional_lister2 = lister.RegionalLister(client2, service) self.assertNotEqual(regional_lister1, regional_lister2)
def testEq(self): client = object() service = object() regional_lister1 = lister.RegionalLister(client, service) regional_lister2 = lister.RegionalLister(client, service) self.assertEqual(regional_lister1, regional_lister2) self.assertEqual(hash(regional_lister1), hash(regional_lister2))
def testAggregatedList(self): lister_patcher = mock.patch( 'googlecloudsdk.api_lib.compute.lister.GetRegionalResourcesDicts', autospec=True) self.addCleanup(lister_patcher.stop) self.mock_get_regional_resources = lister_patcher.start() self.mock_get_regional_resources.return_value = [1, 2, 3] self.api_mock = utils.ComputeApiMock('v1').Start() self.addCleanup(self.api_mock.Stop) resource_registry = resources.REGISTRY.Clone() resource_registry.RegisterApiByName('compute', 'v1') project = resource_registry.Parse( 'https://compute.googleapis.com/compute/v1/projects/lister-project') frontend = lister._Frontend('filter', 123, lister.AllScopes([project], False, True)) regional_lister = lister.RegionalLister(self.api_mock.adapter, 'service') result = list(regional_lister(frontend)) self.assertListEqual(result, [1, 2, 3]) self.mock_get_regional_resources.assert_called_once_with( service='service', project=self.Project(), requested_regions=[], filter_expr='filter', http=self.api_mock.adapter.apitools_client.http, batch_url=self.api_mock.adapter.batch_url, errors=[])
def _Run(args, holder): """Issues requests necessary to list service attachments.""" client = holder.client request_data = lister.ParseRegionalFlags(args, holder.resources) list_implementation = lister.RegionalLister( client, client.apitools_client.serviceAttachments) return lister.Invoke(request_data, list_implementation)
def Run(self, args): holder = base_classes.ComputeApiHolder(self.ReleaseTrack()) client = holder.client request_data = lister.ParseRegionalFlags(args, holder.resources) list_implementation = lister.RegionalLister( client, client.apitools_client.targetVpnGateways) return lister.Invoke(request_data, list_implementation)
def Run(self, args): holder = base_classes.ComputeApiHolder(self.ReleaseTrack()) request_data = lister.ParseRegionalFlags(args, holder.resources) compute_client = holder.client list_implementation = lister.RegionalLister( compute_client, compute_client.apitools_client.packetMirrorings) return lister.Invoke(request_data, list_implementation)
def Run(self, args): holder = base_classes.ComputeApiHolder(self.ReleaseTrack()) client = holder.client request_data = lister.ParseRegionalFlags(args, holder.resources) list_implementation = lister.RegionalLister( client, client.apitools_client.subnetworks) for resource in lister.Invoke(request_data, list_implementation): if args.network is None: yield resource elif 'network' in resource: network_ref = holder.resources.Parse(resource['network']) if network_ref.Name() == args.network: yield resource
def testDeepcopy(self): regional_lister = lister.RegionalLister('client', 'service') self.assertEqual(regional_lister, copy.deepcopy(regional_lister))
def testRepr(self): regional_lister = lister.RegionalLister('client', 'service') # Python 2: u"RegionalLister(u'client', u'service')" # Python 3: "RegionalLister('client', 'service')" expected = 'RegionalLister({}, {})'.format(repr('client'), repr('service')) self.assertEqual(repr(regional_lister), expected)