finish removing DirectPathEnv

googleapis · Oct 22, 2019 · 27fa823 · 27fa823
1 parent 0cbc43e
commit 27fa823
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...oud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/it/DirectPathFallbackIT.java b/...oud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/it/DirectPathFallbackIT.java
@@ -23,7 +23,6 @@
 import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider;
 import com.google.cloud.bigtable.data.v2.BigtableDataClient;
 import com.google.cloud.bigtable.data.v2.BigtableDataSettings;
-import com.google.cloud.bigtable.test_helpers.env.DirectPathEnv;
 import com.google.cloud.bigtable.test_helpers.env.TestEnvRule;
 import io.grpc.ManagedChannelBuilder;
 import io.grpc.alts.ComputeEngineChannelBuilder;
@@ -82,8 +81,8 @@ public DirectPathFallbackIT() {
   public void setup() throws IOException {
     assume()
         .withMessage("DirectPath integration tests can only run against DirectPathEnv")
-        .that(testEnvRule.env())
-        .isInstanceOf(DirectPathEnv.class);
+        .that(testEnvRule.env().isDirectPathEnabled())
+        .isTrue();
 
     BigtableDataSettings defaultSettings = testEnvRule.env().getDataClientSettings();
     InstantiatingGrpcChannelProvider defaultTransportProvider =