Fixed code

Author: TetyanaYahodska

compute/cloud-client/src/main/java/compute/disks/consistencygroup/CreateDiskConsistencyGroup.java

@@ -17,6 +17,7 @@
 package compute.disks.consistencygroup;
 
 // [START compute_consistency_group_create]
+import com.google.cloud.compute.v1.InsertResourcePolicyRequest;
 import com.google.cloud.compute.v1.Operation;
 import com.google.cloud.compute.v1.ResourcePoliciesClient;
 import com.google.cloud.compute.v1.ResourcePolicy;
@@ -53,8 +54,14 @@ public static Operation.Status createDiskConsistencyGroup(
                   ResourcePolicy.newBuilder().getDiskConsistencyGroupPolicy())
               .build();
 
+      InsertResourcePolicyRequest request = InsertResourcePolicyRequest.newBuilder()
+              .setProject(project)
+              .setRegion(region)
+              .setResourcePolicyResource(resourcePolicy)
+              .build();
+
       Operation response =
-          regionResourcePoliciesClient.insertAsync(project, region, resourcePolicy).get();
+          regionResourcePoliciesClient.insertAsync(request).get();
 
       if (response.hasError()) {
         throw new Error("Error creating consistency group! " + response.getError());

compute/cloud-client/src/test/java/compute/disks/ConsistencyGroupIT.java

@@ -26,11 +26,11 @@
 
 import com.google.api.gax.longrunning.OperationFuture;
 import com.google.cloud.compute.v1.AddResourcePoliciesRegionDiskRequest;
+import com.google.cloud.compute.v1.InsertResourcePolicyRequest;
 import com.google.cloud.compute.v1.Operation;
 import com.google.cloud.compute.v1.RegionDisksClient;
 import com.google.cloud.compute.v1.RemoveResourcePoliciesRegionDiskRequest;
 import com.google.cloud.compute.v1.ResourcePoliciesClient;
-import com.google.cloud.compute.v1.ResourcePolicy;
 import compute.disks.consistencygroup.AddDiskToConsistencyGroup;
 import compute.disks.consistencygroup.CreateDiskConsistencyGroup;
 import compute.disks.consistencygroup.DeleteDiskConsistencyGroup;
@@ -51,30 +51,22 @@ public class ConsistencyGroupIT {
 
   @Test
   public void testCreateDiskConsistencyGroupResourcePolicy() throws Exception {
-    ResourcePolicy resourcePolicy =
-            ResourcePolicy.newBuilder()
-                    .setName(CONSISTENCY_GROUP_NAME)
-                    .setRegion(REGION)
-                    .setDiskConsistencyGroupPolicy(
-                            ResourcePolicy.newBuilder().getDiskConsistencyGroupPolicy())
-                    .build();
     try (MockedStatic<ResourcePoliciesClient> mockedResourcePoliciesClient =
                  mockStatic(ResourcePoliciesClient.class)) {
       Operation operation = mock(Operation.class);
       ResourcePoliciesClient mockClient = mock(ResourcePoliciesClient.class);
       OperationFuture mockFuture = mock(OperationFuture.class);
 
       mockedResourcePoliciesClient.when(ResourcePoliciesClient::create).thenReturn(mockClient);
-      when(mockClient.insertAsync(PROJECT_ID, REGION, resourcePolicy))
+      when(mockClient.insertAsync(any(InsertResourcePolicyRequest.class)))
               .thenReturn(mockFuture);
       when(mockFuture.get()).thenReturn(operation);
       when(operation.getStatus()).thenReturn(Operation.Status.DONE);
 
       Operation.Status status = CreateDiskConsistencyGroup.createDiskConsistencyGroup(
               PROJECT_ID, REGION, CONSISTENCY_GROUP_NAME);
 
-      verify(mockClient, times(1))
-              .insertAsync(PROJECT_ID, REGION, resourcePolicy);
+      verify(mockClient, times(1)).insertAsync(any(InsertResourcePolicyRequest.class));
       verify(mockFuture, times(1)).get();
       assertEquals(Operation.Status.DONE, status);
     }