test(bigtable): Enable tests (#2582)

ron-gal · web-flow · commit 4285a7ee1b89 · 2025-05-12T18:48:30.000Z
Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly: - [x] Make sure to open an issue as a [bug/issue](https://togithub.com/googleapis/java-bigtable/issues/new/choose) before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [x] Ensure the tests and linter pass - [x] Code coverage does not decrease (if any source code was changed) - [x] Appropriate docs were updated (if necessary) - [x] Rollback plan is reviewed and LGTMed - [x] All new data plane features have a completed end to end testing plan Fixes #<issue_number_goes_here> ☕️ If you write sample code, please follow the [samples format]( https://togithub.com/GoogleCloudPlatform/java-docs-samples/blob/main/SAMPLE_FORMAT.md).
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
@@ -24,53 +24,76 @@
 import com.google.api.gax.rpc.NotFoundException;
 import com.google.cloud.bigtable.admin.v2.BigtableInstanceAdminClient;
 import com.google.cloud.bigtable.admin.v2.BigtableTableAdminClient;
+import com.google.cloud.bigtable.admin.v2.models.CreateInstanceRequest;
 import com.google.cloud.bigtable.admin.v2.models.CreateMaterializedViewRequest;
 import com.google.cloud.bigtable.admin.v2.models.CreateTableRequest;
+import com.google.cloud.bigtable.admin.v2.models.Instance;
 import com.google.cloud.bigtable.admin.v2.models.MaterializedView;
+import com.google.cloud.bigtable.admin.v2.models.StorageType;
 import com.google.cloud.bigtable.admin.v2.models.Table;
 import com.google.cloud.bigtable.admin.v2.models.UpdateMaterializedViewRequest;
 import com.google.cloud.bigtable.test_helpers.env.EmulatorEnv;
 import com.google.cloud.bigtable.test_helpers.env.PrefixGenerator;
 import com.google.cloud.bigtable.test_helpers.env.TestEnvRule;
 import io.grpc.StatusRuntimeException;
+import java.io.IOException;
 import java.util.List;
 import java.util.logging.Logger;
+import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.ClassRule;
-import org.junit.Ignore;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
-@Ignore("Not fully working yet in production")
 public class BigtableMaterializedViewIT {
   @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};
 
   private static BigtableInstanceAdminClient client;
+  private static BigtableTableAdminClient tableAdminClient;
   private static Table testTable;
-
-  private String instanceId = testEnvRule.env().getInstanceId();
+  private static String instanceId = "";
 
   // TODO: Update this test once emulator supports InstanceAdmin operation
   // https://github.com/googleapis/google-cloud-go/issues/1069
   @BeforeClass
-  public static void validatePlatform() {
+  public static void validatePlatform() throws IOException {
     assume()
         .withMessage("BigtableInstanceAdminClient doesn't support on Emulator")
         .that(testEnvRule.env())
         .isNotInstanceOf(EmulatorEnv.class);
+
+    createInstance();
+  }
+
+  public static void createInstance() throws IOException {
+    client = testEnvRule.env().getInstanceAdminClient();
+
+    Instance instance =
+        client.createInstance(
+            CreateInstanceRequest.of(new PrefixGenerator().newPrefix())
+                .addCluster("my-cluster", "us-east1-c", 3, StorageType.SSD));
+    instanceId = instance.getId();
+    tableAdminClient =
+        BigtableTableAdminClient.create(testEnvRule.env().getProjectId(), instanceId);
+  }
+
+  @AfterClass
+  public static void deleteInstance() {
+    if (!instanceId.isEmpty()) {
+      client.deleteInstance(instanceId);
+    }
   }
 
   @Before
   public void setUp() throws InterruptedException {
-    client = testEnvRule.env().getInstanceAdminClient();
-    testTable = createTestTable(testEnvRule.env().getTableAdminClient());
+    testTable = createTestTable(tableAdminClient);
   }
 
   @Test
