test: fix clean up of protected & cdc tables

Change-Id: Id15852b2b3bbb62746861d7ec02842353612ae77
googleapis · Dec 4, 2024 · 13febd2 · 13febd2
1 parent cb4d60e
commit 13febd2
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/...-cloud-bigtable/src/test/java/com/google/cloud/bigtable/test_helpers/env/TestEnvRule.java b/...-cloud-bigtable/src/test/java/com/google/cloud/bigtable/test_helpers/env/TestEnvRule.java
@@ -24,7 +24,9 @@
 import com.google.cloud.bigtable.admin.v2.models.AppProfile;
 import com.google.cloud.bigtable.admin.v2.models.Cluster;
 import com.google.cloud.bigtable.admin.v2.models.Instance;
+import com.google.cloud.bigtable.admin.v2.models.Table;
 import com.google.cloud.bigtable.admin.v2.models.UpdateAuthorizedViewRequest;
+import com.google.cloud.bigtable.admin.v2.models.UpdateTableRequest;
 import com.google.common.collect.ImmutableSet;
 import java.io.IOException;
 import java.util.ArrayList;
@@ -178,8 +180,19 @@ private void cleanupStaleTables(String stalePrefix) {
   }
 
   private void prepTableForDelete(String tableId) {
-    // Unprotected views
     if (!(env() instanceof EmulatorEnv)) {
+      // unprotect table
+      Table table = env().getTableAdminClient().getTable(tableId);
+      if (table.isDeletionProtected() || table.getChangeStreamRetention() != null) {
+        env()
+            .getTableAdminClient()
+            .updateTable(
+                UpdateTableRequest.of(tableId)
+                    .setDeletionProtection(false)
+                    .disableChangeStreamRetention());
+      }
+
+      // Unprotected views
       for (String viewId : env().getTableAdminClient().listAuthorizedViews(tableId)) {
         try {
           env()