test: Add missing test table cleanup for MaterializedViewIT

trollyxia · trollyxia · commit 8469813029d3 · 2025-12-02T10:59:11.000-05:00
Change-Id: I1281568894ac2d04f0aafe0aa7f4efa7570ae58a
diff --git a/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/it/BigtableMaterializedViewIT.java b/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/it/BigtableMaterializedViewIT.java
@@ -36,6 +36,7 @@
 import java.io.IOException;
 import java.util.List;
 import java.util.logging.Logger;
+import org.junit.After;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.ClassRule;
@@ -46,8 +47,11 @@
 
 @RunWith(JUnit4.class)
 public class BigtableMaterializedViewIT {
-  @ClassRule public static final TestEnvRule testEnvRule = new TestEnvRule();
-  @Rule public final PrefixGenerator prefixGenerator = new PrefixGenerator();
+
+  @ClassRule
+  public static final TestEnvRule testEnvRule = new TestEnvRule();
+  @Rule
+  public final PrefixGenerator prefixGenerator = new PrefixGenerator();
   private static final Logger LOGGER = Logger.getLogger(BigtableMaterializedViewIT.class.getName());
   private static final int[] BACKOFF_DURATION = {2, 4, 8, 16, 32, 64, 128, 256, 512, 1024};
 
@@ -71,6 +75,13 @@ public void setUp() throws InterruptedException, IOException {
     testTable = createTestTable(testEnvRule.env().getTableAdminClient());
   }
 
+  @After
+  public void tearDown() throws InterruptedException {
+    if (testTable != null) {
+      testEnvRule.env().getTableAdminClient().deleteTable(testTable.getId());
+    }
+  }
+
   @Test
   public void createMaterializedViewAndGetMaterializedViewTest() {
     String materializedViewId = prefixGenerator.newPrefix();
