Search in sources :

Example 21 with Binding

use of org.flyte.api.v1.Binding in project java-bigqueryconnection by googleapis.

the class ITSystemTest method testShareConnection.

@Test
public void testShareConnection() {
    ResourceName resource = ConnectionName.of(PROJECT_ID, LOCATION, CONNECTION_ID);
    Binding binding = Binding.newBuilder().addMembers("group:example-analyst-group@google.com").setRole("roles/bigquery.connectionUser").build();
    Policy policy = Policy.newBuilder().addBindings(binding).build();
    SetIamPolicyRequest request = SetIamPolicyRequest.newBuilder().setResource(resource.toString()).setPolicy(policy).build();
    Policy actualPolicy = client.setIamPolicy(request);
    assertEquals(ImmutableList.of(binding), actualPolicy.getBindingsList());
}
Also used : Binding(com.google.iam.v1.Binding) Policy(com.google.iam.v1.Policy) SetIamPolicyRequest(com.google.iam.v1.SetIamPolicyRequest) ResourceName(com.google.api.resourcenames.ResourceName) Test(org.junit.Test)

Example 22 with Binding

use of org.flyte.api.v1.Binding in project java-bigqueryconnection by googleapis.

the class ConnectionServiceClientTest method setIamPolicyTest2.

@Test
public void setIamPolicyTest2() throws Exception {
    Policy expectedResponse = Policy.newBuilder().setVersion(351608024).addAllBindings(new ArrayList<Binding>()).setEtag(ByteString.EMPTY).build();
    mockConnectionService.addResponse(expectedResponse);
    String resource = "resource-341064690";
    Policy policy = Policy.newBuilder().build();
    Policy actualResponse = client.setIamPolicy(resource, policy);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<AbstractMessage> actualRequests = mockConnectionService.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    SetIamPolicyRequest actualRequest = ((SetIamPolicyRequest) actualRequests.get(0));
    Assert.assertEquals(resource, actualRequest.getResource());
    Assert.assertEquals(policy, actualRequest.getPolicy());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : Policy(com.google.iam.v1.Policy) Binding(com.google.iam.v1.Binding) AbstractMessage(com.google.protobuf.AbstractMessage) SetIamPolicyRequest(com.google.iam.v1.SetIamPolicyRequest) ByteString(com.google.protobuf.ByteString) Test(org.junit.Test)

Example 23 with Binding

use of org.flyte.api.v1.Binding in project java-bigqueryconnection by googleapis.

the class ConnectionServiceClientTest method getIamPolicyTest.

@Test
public void getIamPolicyTest() throws Exception {
    Policy expectedResponse = Policy.newBuilder().setVersion(351608024).addAllBindings(new ArrayList<Binding>()).setEtag(ByteString.EMPTY).build();
    mockConnectionService.addResponse(expectedResponse);
    ResourceName resource = ConnectionName.of("[PROJECT]", "[LOCATION]", "[CONNECTION]");
    GetPolicyOptions options = GetPolicyOptions.newBuilder().build();
    Policy actualResponse = client.getIamPolicy(resource, options);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<AbstractMessage> actualRequests = mockConnectionService.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    GetIamPolicyRequest actualRequest = ((GetIamPolicyRequest) actualRequests.get(0));
    Assert.assertEquals(resource.toString(), actualRequest.getResource());
    Assert.assertEquals(options, actualRequest.getOptions());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : Policy(com.google.iam.v1.Policy) Binding(com.google.iam.v1.Binding) AbstractMessage(com.google.protobuf.AbstractMessage) ResourceName(com.google.api.resourcenames.ResourceName) GetPolicyOptions(com.google.iam.v1.GetPolicyOptions) GetIamPolicyRequest(com.google.iam.v1.GetIamPolicyRequest) Test(org.junit.Test)

Example 24 with Binding

use of org.flyte.api.v1.Binding in project java-bigqueryconnection by googleapis.

the class ShareConnection method shareConnection.

public static void shareConnection(String projectId, String location, String connectionId) throws IOException {
    try (ConnectionServiceClient client = ConnectionServiceClient.create()) {
        ResourceName resource = ConnectionName.of(projectId, location, connectionId);
        Binding binding = Binding.newBuilder().addMembers("group:example-analyst-group@google.com").setRole("roles/bigquery.connectionUser").build();
        Policy policy = Policy.newBuilder().addBindings(binding).build();
        SetIamPolicyRequest request = SetIamPolicyRequest.newBuilder().setResource(resource.toString()).setPolicy(policy).build();
        client.setIamPolicy(request);
        System.out.println("Connection shared successfully");
    }
}
Also used : Binding(com.google.iam.v1.Binding) Policy(com.google.iam.v1.Policy) ConnectionServiceClient(com.google.cloud.bigqueryconnection.v1.ConnectionServiceClient) SetIamPolicyRequest(com.google.iam.v1.SetIamPolicyRequest) ResourceName(com.google.api.resourcenames.ResourceName)

Example 25 with Binding

use of org.flyte.api.v1.Binding in project java-pubsub by googleapis.

the class SubscriptionAdminClientTest method getIamPolicyTest.

@Test
public void getIamPolicyTest() throws Exception {
    Policy expectedResponse = Policy.newBuilder().setVersion(351608024).addAllBindings(new ArrayList<Binding>()).setEtag(ByteString.EMPTY).build();
    mockIAMPolicy.addResponse(expectedResponse);
    GetIamPolicyRequest request = GetIamPolicyRequest.newBuilder().setResource(ProjectName.of("[PROJECT]").toString()).setOptions(GetPolicyOptions.newBuilder().build()).build();
    Policy actualResponse = client.getIamPolicy(request);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<AbstractMessage> actualRequests = mockIAMPolicy.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    GetIamPolicyRequest actualRequest = ((GetIamPolicyRequest) actualRequests.get(0));
    Assert.assertEquals(request.getResource(), actualRequest.getResource());
    Assert.assertEquals(request.getOptions(), actualRequest.getOptions());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : ExpirationPolicy(com.google.pubsub.v1.ExpirationPolicy) Policy(com.google.iam.v1.Policy) DeadLetterPolicy(com.google.pubsub.v1.DeadLetterPolicy) RetryPolicy(com.google.pubsub.v1.RetryPolicy) Binding(com.google.iam.v1.Binding) AbstractMessage(com.google.protobuf.AbstractMessage) GetIamPolicyRequest(com.google.iam.v1.GetIamPolicyRequest) Test(org.junit.Test)

Aggregations

Binding (com.google.iam.v1.Binding)104 Policy (com.google.iam.v1.Policy)101 Test (org.junit.Test)87 AbstractMessage (com.google.protobuf.AbstractMessage)82 SetIamPolicyRequest (com.google.iam.v1.SetIamPolicyRequest)47 GetIamPolicyRequest (com.google.iam.v1.GetIamPolicyRequest)43 ResourceName (com.google.api.resourcenames.ResourceName)32 ByteString (com.google.protobuf.ByteString)32 Node (org.flyte.api.v1.Node)8 CryptoKeyName (com.google.cloud.kms.v1.CryptoKeyName)6 KeyManagementServiceClient (com.google.cloud.kms.v1.KeyManagementServiceClient)6 MockIAMPolicy (com.google.iam.v1.MockIAMPolicy)6 TaskNode (org.flyte.api.v1.TaskNode)6 Binding (org.kie.workbench.common.dmn.api.definition.v1_1.Binding)6 Test (org.junit.jupiter.api.Test)5 SecretManagerServiceClient (com.google.cloud.secretmanager.v1.SecretManagerServiceClient)4 SecretName (com.google.cloud.secretmanager.v1.SecretName)4 ArrayList (java.util.ArrayList)4 Binding (org.flyte.api.v1.Binding)4 WorkflowTemplate (org.flyte.api.v1.WorkflowTemplate)4