@Override public Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> call() { return serviceOptions.getComputeRpcV1().listZones(optionsMap); } },
@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 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 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 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)); }