3636import org .apache .beam .it .common .utils .ResourceManagerUtils ;
3737import org .apache .beam .it .conditions .ChainedConditionCheck ;
3838import org .apache .beam .it .conditions .ConditionCheck ;
39+ import org .apache .beam .it .gcp .cloudsql .CloudPostgresResourceManager ;
3940import org .apache .beam .it .gcp .datastream .DatastreamResourceManager ;
4041import org .apache .beam .it .gcp .datastream .PostgresqlSource ;
4142import org .apache .beam .it .gcp .pubsub .PubsubResourceManager ;
4243import org .apache .beam .it .gcp .spanner .SpannerResourceManager ;
4344import org .apache .beam .it .gcp .spanner .conditions .SpannerRowsCheck ;
4445import org .apache .beam .it .gcp .spanner .matchers .SpannerAsserts ;
4546import org .apache .beam .it .gcp .storage .GcsResourceManager ;
46- import org .apache .beam .it .jdbc .PostgresResourceManager ;
4747import org .junit .AfterClass ;
4848import org .junit .Before ;
4949import 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