test: improve bucket_defaultAcl_update test to modify the project-editors entity to be a READER rather than OWNER

Author: BenWhitehead

google-cloud-storage/src/test/java/com/google/cloud/storage/it/ITAccessTest.java

@@ -33,6 +33,8 @@
 import com.google.cloud.RetryHelper.RetryHelperException;
 import com.google.cloud.http.BaseHttpServiceException;
 import com.google.cloud.storage.Acl;
+import com.google.cloud.storage.Acl.Entity;
+import com.google.cloud.storage.Acl.Project.ProjectRole;
 import com.google.cloud.storage.Acl.Role;
 import com.google.cloud.storage.Acl.User;
 import com.google.cloud.storage.Blob;
@@ -70,6 +72,7 @@
 import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.Callable;
+import java.util.function.Predicate;
 import java.util.stream.Collector;
 import java.util.stream.Collectors;
 import org.junit.Ignore;
@@ -207,6 +210,7 @@ public void bucket_defaultAcl_create() throws Exception {
       List<Acl> expectedAcls = dropEtags(bucket.getDefaultAcl());
       List<Acl> actualAcls = dropEtags(bucketUpdated.getDefaultAcl());
       assertThat(actualAcls).containsAtLeastElementsIn(expectedAcls);
+      assertThat(actualAcls).contains(readAll);
     }
   }
 
@@ -230,12 +234,47 @@ public void bucket_defaultAcl_update() throws Exception {
         TemporaryBucket.newBuilder().setBucketInfo(bucketInfo).setStorage(storage).build()) {
       BucketInfo bucket = tempB.getBucket();
 
-      Acl readAll = Acl.of(User.ofAllAuthenticatedUsers(), Role.READER);
-      Acl actual = retry429s(() -> storage.updateDefaultAcl(bucket.getName(), readAll), storage);
+      List<Acl> defaultAcls = bucket.getDefaultAcl();
+      System.out.println("defaultAcls = " + defaultAcls);
+      assertThat(defaultAcls).isNotEmpty();
 
-      assertThat(actual.getEntity()).isEqualTo(readAll.getEntity());
-      assertThat(actual.getRole()).isEqualTo(readAll.getRole());
+      Predicate<Acl> isProjectEditor = hasProjectRole(ProjectRole.EDITORS);
+
+      //noinspection OptionalGetWithoutIsPresent
+      Acl projectEditorAsOwner =
+          defaultAcls.stream().filter(hasRole(Role.OWNER).and(isProjectEditor)).findFirst().get();
+      System.out.println("projectEditorAsOwner = " + projectEditorAsOwner);
+
+      // lower the privileges of project editors to writer from owner
+      Entity entity = projectEditorAsOwner.getEntity();
+      System.out.println("entity = " + entity);
+      Acl projectEditorAsReader = Acl.of(entity, Role.READER);
+      System.out.println("projectEditorAsReader = " + projectEditorAsReader);
+
+      Acl actual =
+          retry429s(
+              () -> storage.updateDefaultAcl(bucket.getName(), projectEditorAsReader), storage);
+
+      assertThat(actual.getEntity()).isEqualTo(projectEditorAsReader.getEntity());
+      assertThat(actual.getRole()).isEqualTo(projectEditorAsReader.getRole());
       assertThat(actual.getEtag()).isNotEmpty();
+
+      Bucket bucketUpdated =
+          storage.get(bucket.getName(), BucketGetOption.fields(BucketField.values()));
+      assertThat(bucketUpdated.getMetageneration()).isNotEqualTo(bucket.getMetageneration());
+
+      // etags change when updates happen, drop before our comparison
+      List<Acl> expectedAcls =
+          dropEtags(
+              bucket.getDefaultAcl().stream()
+                  .filter(isProjectEditor.negate())
+                  .collect(Collectors.toList()));
+      System.out.println("expectedAcls = " + expectedAcls);
+      List<Acl> actualAcls = dropEtags(bucketUpdated.getDefaultAcl());
+      System.out.println("actualAcls = " + actualAcls);
+      assertThat(actualAcls).containsAtLeastElementsIn(expectedAcls);
+      assertThat(actualAcls).doesNotContain(projectEditorAsOwner);
+      assertThat(actualAcls).contains(projectEditorAsReader);
     }
   }
 
@@ -1098,4 +1137,18 @@ private static ImmutableList<Acl> dropEtags(List<Acl> defaultAcls) {
         .map(acl -> Acl.of(acl.getEntity(), acl.getRole()))
         .collect(ImmutableList.toImmutableList());
   }
+
+  private static Predicate<Acl> hasRole(Acl.Role expected) {
+    return acl -> acl.getRole().equals(expected);
+  }
+
+  private static Predicate<Acl> hasProjectRole(Acl.Project.ProjectRole expected) {
+    return acl -> {
+      Entity entity = acl.getEntity();
+      if (entity.getType().equals(Entity.Type.PROJECT)) {
+        return ((Acl.Project) entity).getProjectRole().equals(expected);
+      }
+      return false;
+    };
+  }
 }