|
35 | 35 | import java.util.Arrays;
|
36 | 36 | import java.util.List;
|
37 | 37 | import java.util.UUID;
|
| 38 | +import java.util.stream.Stream; |
38 | 39 | import javax.annotation.Generated;
|
39 | 40 | import org.junit.After;
|
40 | 41 | import org.junit.AfterClass;
|
@@ -442,6 +443,41 @@ public void listUsersTest() throws Exception {
|
442 | 443 | GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
|
443 | 444 | }
|
444 | 445 |
|
| 446 | + @Test |
| 447 | + public void listUsersTest2() throws Exception { |
| 448 | + User responsesElement = User.newBuilder().build(); |
| 449 | + ListUsersResponse expectedResponse = |
| 450 | + ListUsersResponse.newBuilder() |
| 451 | + .setNextPageToken("") |
| 452 | + .addAllUsers(Arrays.asList(responsesElement)) |
| 453 | + .build(); |
| 454 | + mockIdentity.addResponse(expectedResponse); |
| 455 | + |
| 456 | + ListUsersRequest request = |
| 457 | + ListUsersRequest.newBuilder() |
| 458 | + .setPageSize(883849137) |
| 459 | + .setPageToken("pageToken873572522") |
| 460 | + .build(); |
| 461 | + |
| 462 | + ListUsersPagedResponse pagedListResponse = client.listUsers(request); |
| 463 | + |
| 464 | + Stream<User> resources = pagedListResponse.getPage().streamAll(); |
| 465 | + |
| 466 | + Assert.assertEquals(1, resources.count()); |
| 467 | + Assert.assertEquals(responsesElement, resources.findFirst().get()); |
| 468 | + |
| 469 | + List<AbstractMessage> actualRequests = mockIdentity.getRequests(); |
| 470 | + Assert.assertEquals(1, actualRequests.size()); |
| 471 | + ListUsersRequest actualRequest = ((ListUsersRequest) actualRequests.get(0)); |
| 472 | + |
| 473 | + Assert.assertEquals(request.getPageSize(), actualRequest.getPageSize()); |
| 474 | + Assert.assertEquals(request.getPageToken(), actualRequest.getPageToken()); |
| 475 | + Assert.assertTrue( |
| 476 | + channelProvider.isHeaderSent( |
| 477 | + ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), |
| 478 | + GaxGrpcProperties.getDefaultApiClientHeaderPattern())); |
| 479 | + } |
| 480 | + |
445 | 481 | @Test
|
446 | 482 | public void listUsersExceptionTest() throws Exception {
|
447 | 483 | StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
|
|
0 commit comments