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)
