| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -30,12 +30,16 @@ | |||
| 30 | 30 | import com.google.cloud.pubsub.v1.TopicAdminClient; | |
| 31 | 31 | import com.google.common.util.concurrent.MoreExecutors; | |
| 32 | 32 | import com.google.iam.v1.Binding; | |
| 33 | + import com.google.iam.v1.GetIamPolicyRequest; | ||
| 33 | 34 | import com.google.iam.v1.Policy; | |
| 35 | + import com.google.iam.v1.SetIamPolicyRequest; | ||
| 36 | + import com.google.iam.v1.TestIamPermissionsRequest; | ||
| 34 | 37 | import com.google.protobuf.ByteString; | |
| 35 | 38 | import com.google.pubsub.v1.ProjectSubscriptionName; | |
| 36 | - import com.google.pubsub.v1.ProjectTopicName; | ||
| 37 | 39 | import com.google.pubsub.v1.PubsubMessage; | |
| 38 | 40 | import com.google.pubsub.v1.PushConfig; | |
| 41 | + import com.google.pubsub.v1.Subscription; | ||
| 42 | + import com.google.pubsub.v1.TopicName; | ||
| 39 | 43 | import java.util.Collections; | |
| 40 | 44 | import java.util.List; | |
| 41 | 45 | import java.util.UUID; | |
@@ -89,23 +93,50 @@ private String formatForTest(String resourceName) { | |||
| 89 | 93 | return resourceName + "-" + NAME_SUFFIX; | |
| 90 | 94 | } | |
| 91 | 95 | ||
| 96 | + private Subscription getSubscription( | ||
| 97 | + ProjectSubscriptionName subscriptionName, | ||
| 98 | + TopicName topicName, | ||
| 99 | + PushConfig pushConfig, | ||
| 100 | + int ackDeadline) { | ||
| 101 | + return Subscription.newBuilder() | ||
| 102 | + .setName(subscriptionName.toString()) | ||
| 103 | + .setTopic(topicName.toString()) | ||
| 104 | + .setPushConfig(pushConfig) | ||
| 105 | + .setAckDeadlineSeconds(ackDeadline) | ||
| 106 | + .build(); | ||
| 107 | + } | ||
| 108 | + | ||
| 92 | 109 | @Test | |
| 93 | 110 | public void testTopicPolicy() { | |
| 94 | - ProjectTopicName topicName = | ||
| 95 | - ProjectTopicName.of(projectId, formatForTest("testing-topic-policy")); | ||
| 111 | + TopicName topicName = | ||
| 112 | + TopicName.newBuilder() | ||
| 113 | + .setProject(projectId) | ||
| 114 | + .setTopic(formatForTest("testing-topic-policy")) | ||
| 115 | + .build(); | ||
| 96 | 116 | topicAdminClient.createTopic(topicName); | |
| 97 | - Policy policy = topicAdminClient.getIamPolicy(topicName.toString()); | ||
| 117 | + | ||
| 118 | + Policy policy = | ||
| 119 | + topicAdminClient.getIamPolicy( | ||
| 120 | + GetIamPolicyRequest.newBuilder().setResource(topicName.toString()).build()); | ||
| 98 | 121 | Binding binding = | |
| 99 | 122 | Binding.newBuilder().setRole("roles/viewer").addMembers("allAuthenticatedUsers").build(); | |
| 123 | + | ||
| 100 | 124 | Policy newPolicy = | |
| 101 | 125 | topicAdminClient.setIamPolicy( | |
| 102 | - topicName.toString(), policy.toBuilder().addBindings(binding).build()); | ||
| 126 | + SetIamPolicyRequest.newBuilder() | ||
| 127 | + .setResource(topicName.toString()) | ||
| 128 | + .setPolicy(policy.toBuilder().addBindings(binding).build()) | ||
| 129 | + .build()); | ||
| 103 | 130 | assertThat(newPolicy.getBindingsList()).contains(binding); | |
| 104 | 131 | ||
| 105 | 132 | String permissionName = "pubsub.topics.get"; | |
| 106 | 133 | List<String> permissions = | |
| 107 | 134 | topicAdminClient | |
| 108 | - .testIamPermissions(topicName.toString(), Collections.singletonList(permissionName)) | ||
| 135 | + .testIamPermissions( | ||
| 136 | + TestIamPermissionsRequest.newBuilder() | ||
| 137 | + .setResource(topicName.toString()) | ||
| 138 | + .addAllPermissions(Collections.singletonList(permissionName)) | ||
| 139 | + .build()) | ||
| 109 | 140 | .getPermissionsList(); | |
| 110 | 141 | assertThat(permissions).contains(permissionName); | |
| 111 | 142 | ||
@@ -115,19 +146,23 @@ public void testTopicPolicy() { | |||
| 115 | 146 | @Test | |
| 116 | 147 | public void testVPCPushSubscriber() { | |
| 117 | 148 | assumeTrue(IS_VPC_TEST); | |
| 118 | - ProjectTopicName topicName = | ||
| 119 | - ProjectTopicName.of(projectId, formatForTest("testing-vpc-push-subscriber-topic")); | ||
| 149 | + TopicName topicName = | ||
| 150 | + TopicName.newBuilder() | ||
| 151 | + .setProject(projectId) | ||
| 152 | + .setTopic(formatForTest("testing-vpc-push-subscriber-topic")) | ||
| 153 | + .build(); | ||
| 120 | 154 | ProjectSubscriptionName subscriptionName = | |
| 121 | 155 | ProjectSubscriptionName.of( | |
| 122 | 156 | projectId, formatForTest("testing-vpc-push-subscriber-subscription")); | |
| 123 | 157 | topicAdminClient.createTopic(topicName); | |
| 124 | 158 | ||
| 125 | 159 | try { | |
| 126 | 160 | subscriptionAdminClient.createSubscription( | |
| 127 | - subscriptionName, | ||
| 128 | - topicName, | ||
| 129 | - PushConfig.newBuilder().setPushEndpoint("https://random_point").build(), | ||
| 130 | - 10); | ||
| 161 | + getSubscription( | ||
| 162 | + subscriptionName, | ||
| 163 | + topicName, | ||
| 164 | + PushConfig.newBuilder().setPushEndpoint("https://random_point").build(), | ||
| 165 | + 10)); | ||
| 131 | 166 | subscriptionAdminClient.deleteSubscription(subscriptionName); | |
| 132 | 167 | Assert.fail("No exception raised"); | |
| 133 | 168 | } catch (PermissionDeniedException e) { | |
@@ -139,15 +174,19 @@ public void testVPCPushSubscriber() { | |||
| 139 | 174 | ||
| 140 | 175 | @Test | |
| 141 | 176 | public void testPublishSubscribe() throws Exception { | |
| 142 | - ProjectTopicName topicName = | ||
| 143 | - ProjectTopicName.of(projectId, formatForTest("testing-publish-subscribe-topic")); | ||
| 177 | + TopicName topicName = | ||
| 178 | + TopicName.newBuilder() | ||
| 179 | + .setProject(projectId) | ||
| 180 | + .setTopic(formatForTest("testing-publish-subscribe-topic")) | ||
| 181 | + .build(); | ||
| 144 | 182 | ProjectSubscriptionName subscriptionName = | |
| 145 | 183 | ProjectSubscriptionName.of( | |
| 146 | 184 | projectId, formatForTest("testing-publish-subscribe-subscription")); | |
| 147 | 185 | ||
| 148 | 186 | topicAdminClient.createTopic(topicName); | |
| 187 | + | ||
| 149 | 188 | subscriptionAdminClient.createSubscription( | |
| 150 | - subscriptionName, topicName, PushConfig.newBuilder().build(), 10); | ||
| 189 | + getSubscription(subscriptionName, topicName, PushConfig.newBuilder().build(), 10)); | ||
| 151 | 190 | ||
| 152 | 191 | final BlockingQueue<Object> receiveQueue = new LinkedBlockingQueue<>(); | |
| 153 | 192 | Subscriber subscriber = | |
| Back | FazBrowse Home | New Git URL |
0 commit comments