|
25 | 25 | import com.google.api.gax.grpc.testing.MockServiceHelper; |
26 | 26 | import com.google.api.gax.rpc.ApiClientHeaderProvider; |
27 | 27 | import com.google.api.gax.rpc.InvalidArgumentException; |
| 28 | +import com.google.api.resourcenames.ResourceName; |
28 | 29 | import com.google.common.collect.Lists; |
| 30 | +import com.google.iam.v1.GetIamPolicyRequest; |
| 31 | +import com.google.iam.v1.Policy; |
| 32 | +import com.google.iam.v1.SetIamPolicyRequest; |
| 33 | +import com.google.iam.v1.TestIamPermissionsRequest; |
| 34 | +import com.google.iam.v1.TestIamPermissionsResponse; |
29 | 35 | import com.google.protobuf.AbstractMessage; |
| 36 | +import com.google.protobuf.ByteString; |
30 | 37 | import com.google.protobuf.Empty; |
31 | 38 | import com.google.protobuf.FieldMask; |
32 | 39 | import io.grpc.Status; |
33 | 40 | import io.grpc.StatusRuntimeException; |
34 | 41 | import java.io.IOException; |
| 42 | +import java.util.ArrayList; |
35 | 43 | import java.util.Arrays; |
36 | 44 | import java.util.List; |
| 45 | +import java.util.Objects; |
37 | 46 | import java.util.UUID; |
38 | 47 | import org.junit.After; |
39 | 48 | import org.junit.AfterClass; |
@@ -580,4 +589,149 @@ public void destroySecretVersionExceptionTest() throws Exception { |
580 | 589 | // Expected exception |
581 | 590 | } |
582 | 591 | } |
| 592 | + |
| 593 | + @Test |
| 594 | + @SuppressWarnings("all") |
| 595 | + public void setIamPolicyTest() { |
| 596 | + int version = 351608024; |
| 597 | + ByteString etag = ByteString.copyFromUtf8("21"); |
| 598 | + Policy expectedResponse = Policy.newBuilder().setVersion(version).setEtag(etag).build(); |
| 599 | + mockSecretManagerService.addResponse(expectedResponse); |
| 600 | + |
| 601 | + ResourceName resource = ProjectName.of("[PROJECT]"); |
| 602 | + Policy policy = Policy.newBuilder().build(); |
| 603 | + SetIamPolicyRequest request = |
| 604 | + SetIamPolicyRequest.newBuilder().setResource(resource.toString()).setPolicy(policy).build(); |
| 605 | + |
| 606 | + Policy actualResponse = client.setIamPolicy(request); |
| 607 | + Assert.assertEquals(expectedResponse, actualResponse); |
| 608 | + |
| 609 | + List<AbstractMessage> actualRequests = mockSecretManagerService.getRequests(); |
| 610 | + Assert.assertEquals(1, actualRequests.size()); |
| 611 | + SetIamPolicyRequest actualRequest = (SetIamPolicyRequest) actualRequests.get(0); |
| 612 | + |
| 613 | + Assert.assertEquals(Objects.toString(resource), Objects.toString(actualRequest.getResource())); |
| 614 | + Assert.assertEquals(policy, actualRequest.getPolicy()); |
| 615 | + Assert.assertTrue( |
| 616 | + channelProvider.isHeaderSent( |
| 617 | + ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), |
| 618 | + GaxGrpcProperties.getDefaultApiClientHeaderPattern())); |
| 619 | + } |
| 620 | + |
| 621 | + @Test |
| 622 | + @SuppressWarnings("all") |
| 623 | + public void setIamPolicyExceptionTest() throws Exception { |
| 624 | + StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); |
| 625 | + mockSecretManagerService.addException(exception); |
| 626 | + |
| 627 | + try { |
| 628 | + ResourceName resource = ProjectName.of("[PROJECT]"); |
| 629 | + Policy policy = Policy.newBuilder().build(); |
| 630 | + SetIamPolicyRequest request = |
| 631 | + SetIamPolicyRequest.newBuilder() |
| 632 | + .setResource(resource.toString()) |
| 633 | + .setPolicy(policy) |
| 634 | + .build(); |
| 635 | + |
| 636 | + client.setIamPolicy(request); |
| 637 | + Assert.fail("No exception raised"); |
| 638 | + } catch (InvalidArgumentException e) { |
| 639 | + // Expected exception |
| 640 | + } |
| 641 | + } |
| 642 | + |
| 643 | + @Test |
| 644 | + @SuppressWarnings("all") |
| 645 | + public void getIamPolicyTest() { |
| 646 | + int version = 351608024; |
| 647 | + ByteString etag = ByteString.copyFromUtf8("21"); |
| 648 | + Policy expectedResponse = Policy.newBuilder().setVersion(version).setEtag(etag).build(); |
| 649 | + mockSecretManagerService.addResponse(expectedResponse); |
| 650 | + |
| 651 | + ResourceName resource = ProjectName.of("[PROJECT]"); |
| 652 | + GetIamPolicyRequest request = |
| 653 | + GetIamPolicyRequest.newBuilder().setResource(resource.toString()).build(); |
| 654 | + |
| 655 | + Policy actualResponse = client.getIamPolicy(request); |
| 656 | + Assert.assertEquals(expectedResponse, actualResponse); |
| 657 | + |
| 658 | + List<AbstractMessage> actualRequests = mockSecretManagerService.getRequests(); |
| 659 | + Assert.assertEquals(1, actualRequests.size()); |
| 660 | + GetIamPolicyRequest actualRequest = (GetIamPolicyRequest) actualRequests.get(0); |
| 661 | + |
| 662 | + Assert.assertEquals(Objects.toString(resource), Objects.toString(actualRequest.getResource())); |
| 663 | + Assert.assertTrue( |
| 664 | + channelProvider.isHeaderSent( |
| 665 | + ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), |
| 666 | + GaxGrpcProperties.getDefaultApiClientHeaderPattern())); |
| 667 | + } |
| 668 | + |
| 669 | + @Test |
| 670 | + @SuppressWarnings("all") |
| 671 | + public void getIamPolicyExceptionTest() throws Exception { |
| 672 | + StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); |
| 673 | + mockSecretManagerService.addException(exception); |
| 674 | + |
| 675 | + try { |
| 676 | + ResourceName resource = ProjectName.of("[PROJECT]"); |
| 677 | + GetIamPolicyRequest request = |
| 678 | + GetIamPolicyRequest.newBuilder().setResource(resource.toString()).build(); |
| 679 | + |
| 680 | + client.getIamPolicy(request); |
| 681 | + Assert.fail("No exception raised"); |
| 682 | + } catch (InvalidArgumentException e) { |
| 683 | + // Expected exception |
| 684 | + } |
| 685 | + } |
| 686 | + |
| 687 | + @Test |
| 688 | + @SuppressWarnings("all") |
| 689 | + public void testIamPermissionsTest() { |
| 690 | + TestIamPermissionsResponse expectedResponse = TestIamPermissionsResponse.newBuilder().build(); |
| 691 | + mockSecretManagerService.addResponse(expectedResponse); |
| 692 | + |
| 693 | + ResourceName resource = ProjectName.of("[PROJECT]"); |
| 694 | + List<String> permissions = new ArrayList<>(); |
| 695 | + TestIamPermissionsRequest request = |
| 696 | + TestIamPermissionsRequest.newBuilder() |
| 697 | + .setResource(resource.toString()) |
| 698 | + .addAllPermissions(permissions) |
| 699 | + .build(); |
| 700 | + |
| 701 | + TestIamPermissionsResponse actualResponse = client.testIamPermissions(request); |
| 702 | + Assert.assertEquals(expectedResponse, actualResponse); |
| 703 | + |
| 704 | + List<AbstractMessage> actualRequests = mockSecretManagerService.getRequests(); |
| 705 | + Assert.assertEquals(1, actualRequests.size()); |
| 706 | + TestIamPermissionsRequest actualRequest = (TestIamPermissionsRequest) actualRequests.get(0); |
| 707 | + |
| 708 | + Assert.assertEquals(Objects.toString(resource), Objects.toString(actualRequest.getResource())); |
| 709 | + Assert.assertEquals(permissions, actualRequest.getPermissionsList()); |
| 710 | + Assert.assertTrue( |
| 711 | + channelProvider.isHeaderSent( |
| 712 | + ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), |
| 713 | + GaxGrpcProperties.getDefaultApiClientHeaderPattern())); |
| 714 | + } |
| 715 | + |
| 716 | + @Test |
| 717 | + @SuppressWarnings("all") |
| 718 | + public void testIamPermissionsExceptionTest() throws Exception { |
| 719 | + StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); |
| 720 | + mockSecretManagerService.addException(exception); |
| 721 | + |
| 722 | + try { |
| 723 | + ResourceName resource = ProjectName.of("[PROJECT]"); |
| 724 | + List<String> permissions = new ArrayList<>(); |
| 725 | + TestIamPermissionsRequest request = |
| 726 | + TestIamPermissionsRequest.newBuilder() |
| 727 | + .setResource(resource.toString()) |
| 728 | + .addAllPermissions(permissions) |
| 729 | + .build(); |
| 730 | + |
| 731 | + client.testIamPermissions(request); |
| 732 | + Assert.fail("No exception raised"); |
| 733 | + } catch (InvalidArgumentException e) { |
| 734 | + // Expected exception |
| 735 | + } |
| 736 | + } |
583 | 737 | } |
0 commit comments