use of com.google.cloud.bigquery.connection.v1.UpdateConnectionRequest in project java-bigqueryconnection by googleapis.
the class ConnectionServiceClientTest method updateConnectionTest2.
@Test
public void updateConnectionTest2() throws Exception {
Connection expectedResponse = Connection.newBuilder().setName(ConnectionName.of("[PROJECT]", "[LOCATION]", "[CONNECTION]").toString()).setFriendlyName("friendlyName461933014").setDescription("description-1724546052").setCreationTime(1932333101).setLastModifiedTime(-671513446).setHasCredential(true).build();
mockConnectionService.addResponse(expectedResponse);
String name = "name3373707";
Connection connection = Connection.newBuilder().build();
FieldMask updateMask = FieldMask.newBuilder().build();
Connection actualResponse = client.updateConnection(name, connection, updateMask);
Assert.assertEquals(expectedResponse, actualResponse);
List<AbstractMessage> actualRequests = mockConnectionService.getRequests();
Assert.assertEquals(1, actualRequests.size());
UpdateConnectionRequest actualRequest = ((UpdateConnectionRequest) actualRequests.get(0));
Assert.assertEquals(name, actualRequest.getName());
Assert.assertEquals(connection, actualRequest.getConnection());
Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.cloud.bigquery.connection.v1.UpdateConnectionRequest in project java-bigqueryconnection by googleapis.
the class UpdateConnection method updateConnection.
public static void updateConnection(String projectId, String location, String connectionId, Connection connection) throws IOException {
try (ConnectionServiceClient client = ConnectionServiceClient.create()) {
ConnectionName name = ConnectionName.of(projectId, location, connectionId);
FieldMask updateMask = FieldMaskUtil.fromString("description");
UpdateConnectionRequest request = UpdateConnectionRequest.newBuilder().setName(name.toString()).setConnection(connection).setUpdateMask(updateMask).build();
Connection response = client.updateConnection(request);
System.out.println("Connection updated successfully :" + response.getDescription());
}
}
use of com.google.cloud.bigquery.connection.v1.UpdateConnectionRequest in project java-bigqueryconnection by googleapis.
the class ConnectionServiceClientTest method updateConnectionTest.
@Test
public void updateConnectionTest() throws Exception {
Connection expectedResponse = Connection.newBuilder().setName(ConnectionName.of("[PROJECT]", "[LOCATION]", "[CONNECTION]").toString()).setFriendlyName("friendlyName461933014").setDescription("description-1724546052").setCreationTime(1932333101).setLastModifiedTime(-671513446).setHasCredential(true).build();
mockConnectionService.addResponse(expectedResponse);
ConnectionName name = ConnectionName.of("[PROJECT]", "[LOCATION]", "[CONNECTION]");
Connection connection = Connection.newBuilder().build();
FieldMask updateMask = FieldMask.newBuilder().build();
Connection actualResponse = client.updateConnection(name, connection, updateMask);
Assert.assertEquals(expectedResponse, actualResponse);
List<AbstractMessage> actualRequests = mockConnectionService.getRequests();
Assert.assertEquals(1, actualRequests.size());
UpdateConnectionRequest actualRequest = ((UpdateConnectionRequest) actualRequests.get(0));
Assert.assertEquals(name.toString(), actualRequest.getName());
Assert.assertEquals(connection, actualRequest.getConnection());
Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.cloud.bigquery.connection.v1.UpdateConnectionRequest in project java-bigqueryconnection by googleapis.
the class ITSystemTest method testUpdateConnection.
@Test
public void testUpdateConnection() {
String description = "MY_DESCRIPTION";
FieldMask updateMask = FieldMaskUtil.fromString("description");
UpdateConnectionRequest request = UpdateConnectionRequest.newBuilder().setName(connection.getName()).setConnection(Connection.newBuilder().setDescription(description).build()).setUpdateMask(updateMask).build();
Connection actualConnection = client.updateConnection(request);
assertEquals(description, actualConnection.getDescription());
}
Aggregations