Skip to content

Commit 79bd439

Browse files
committed
Use static resource manager instead to allow access to CDC log
1 parent feb8fc7 commit 79bd439

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

v2/datastream-to-spanner/src/test/java/com/google/cloud/teleport/v2/templates/PostgreSQLDatastreamToSpannerDataTypesIT.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,14 @@
3636
import org.apache.beam.it.common.utils.ResourceManagerUtils;
3737
import org.apache.beam.it.conditions.ChainedConditionCheck;
3838
import org.apache.beam.it.conditions.ConditionCheck;
39+
import org.apache.beam.it.gcp.cloudsql.CloudPostgresResourceManager;
3940
import org.apache.beam.it.gcp.datastream.DatastreamResourceManager;
4041
import org.apache.beam.it.gcp.datastream.PostgresqlSource;
4142
import org.apache.beam.it.gcp.pubsub.PubsubResourceManager;
4243
import org.apache.beam.it.gcp.spanner.SpannerResourceManager;
4344
import org.apache.beam.it.gcp.spanner.conditions.SpannerRowsCheck;
4445
import org.apache.beam.it.gcp.spanner.matchers.SpannerAsserts;
4546
import org.apache.beam.it.gcp.storage.GcsResourceManager;
46-
import org.apache.beam.it.jdbc.PostgresResourceManager;
4747
import org.junit.AfterClass;
4848
import org.junit.Before;
4949
import org.junit.Test;
@@ -129,7 +129,7 @@ public class PostgreSQLDatastreamToSpannerDataTypesIT extends DataStreamToSpanne
129129
"pg_dialect_data_types_test_replication_slot";
130130

131131
private static boolean initialized = false;
132-
private static PostgresResourceManager postgresResourceManager;
132+
private static CloudPostgresResourceManager postgresResourceManager;
133133
private static SpannerResourceManager spannerResourceManager;
134134
private static SpannerResourceManager pgDialectSpannerResourceManager;
135135
private static GcsResourceManager gcsResourceManager;
@@ -145,7 +145,7 @@ public void setUp() throws IOException {
145145
testInstances.add(this);
146146
if (!initialized) {
147147
LOG.info("Setting up PostgreSQL resource manager...");
148-
postgresResourceManager = PostgresResourceManager.builder(testName).build();
148+
postgresResourceManager = CloudPostgresResourceManager.builder(testName).build();
149149
LOG.info(
150150
"PostgreSQL resource manager created with URI: {}", postgresResourceManager.getUri());
151151
LOG.info("Setting up Spanner resource manager...");

0 commit comments

Comments
 (0)