@Override public void run(Compute compute, Void arg) { for (Zone zone : compute.listZones().iterateAll()) { System.out.println(zone); } } }
@Test public void testListZones() { String cursor = "cursor"; compute = options.getService(); ImmutableList<Zone> zoneList = ImmutableList.of(ZONE, ZONE); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> result = Tuple.of(cursor, Iterables.transform(zoneList, Zone.TO_PB_FUNCTION)); EasyMock.expect(computeRpcMock.listZones(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Zone> page = compute.listZones(); assertEquals(cursor, page.getNextPageToken()); assertArrayEquals(zoneList.toArray(), Iterables.toArray(page.getValues(), Zone.class)); }
@Test public void testListZonesWithOptions() { String cursor = "cursor"; compute = options.getService(); ImmutableList<Zone> zoneList = ImmutableList.of(ZONE, ZONE); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> result = Tuple.of(cursor, Iterables.transform(zoneList, Zone.TO_PB_FUNCTION)); EasyMock.expect(computeRpcMock.listZones(ZONE_LIST_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Zone> page = compute.listZones(ZONE_LIST_PAGE_SIZE, ZONE_LIST_PAGE_TOKEN, ZONE_LIST_FILTER); assertEquals(cursor, page.getNextPageToken()); assertArrayEquals(zoneList.toArray(), Iterables.toArray(page.getValues(), Zone.class)); }
@Test public void testListEmptyZones() { ImmutableList<com.google.api.services.compute.model.Zone> zones = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Zone>>of(null, zones); EasyMock.expect(computeRpcMock.listZones(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); compute = options.getService(); Page<Zone> page = compute.listZones(); assertNull(page.getNextPageToken()); assertArrayEquals(zones.toArray(), Iterables.toArray(page.getValues(), Zone.class)); }
@Test public void testListZonesWithFilter() { Page<Zone> zonePage = compute.listZones( Compute.ZoneListOption.filter(Compute.ZoneFilter.equals(Compute.ZoneField.NAME, ZONE))); Iterator<Zone> zoneIterator = zonePage.iterateAll().iterator(); assertEquals(ZONE, zoneIterator.next().getZoneId().getZone()); assertFalse(zoneIterator.hasNext()); }
@Test public void testListZonesNextPage() { String cursor = "cursor"; String nextCursor = "nextCursor"; compute = options.getService(); ImmutableList<Zone> zoneList = ImmutableList.of(ZONE, ZONE); ImmutableList<Zone> nextZoneList = ImmutableList.of(ZONE); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> result = Tuple.of(cursor, Iterables.transform(zoneList, Zone.TO_PB_FUNCTION)); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> nextResult = Tuple.of(nextCursor, Iterables.transform(nextZoneList, Zone.TO_PB_FUNCTION)); Map<ComputeRpc.Option, ?> nextOptions = ImmutableMap.of(ComputeRpc.Option.PAGE_TOKEN, cursor); EasyMock.expect(computeRpcMock.listZones(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.expect(computeRpcMock.listZones(nextOptions)).andReturn(nextResult); EasyMock.replay(computeRpcMock); Page<Zone> page = compute.listZones(); assertEquals(cursor, page.getNextPageToken()); assertArrayEquals(zoneList.toArray(), Iterables.toArray(page.getValues(), Zone.class)); page = page.getNextPage(); assertEquals(nextCursor, page.getNextPageToken()); assertArrayEquals(nextZoneList.toArray(), Iterables.toArray(page.getValues(), Zone.class)); }
@Test public void testListZones() { Page<Zone> zonePage = compute.listZones(); Iterator<Zone> zoneIterator = zonePage.iterateAll().iterator(); while (zoneIterator.hasNext()) { Zone zone = zoneIterator.next(); assertNotNull(zone.getZoneId()); assertNotNull(zone.getGeneratedId()); assertNotNull(zone.getCreationTimestamp()); assertNotNull(zone.getDescription()); assertNotNull(zone.getStatus()); assertNotNull(zone.getRegion()); } }
@Test public void testListZonesWithSelectedFields() { Page<Zone> zonePage = compute.listZones(Compute.ZoneListOption.fields(Compute.ZoneField.CREATION_TIMESTAMP)); Iterator<Zone> zoneIterator = zonePage.iterateAll().iterator(); while (zoneIterator.hasNext()) { Zone zone = zoneIterator.next(); assertNotNull(zone.getZoneId()); assertNull(zone.getGeneratedId()); assertNotNull(zone.getCreationTimestamp()); assertNull(zone.getDescription()); assertNull(zone.getStatus()); assertNull(zone.getRegion()); } }