mirror of
https://github.com/postgres/postgres.git
synced 2026-04-21 22:28:15 -04:00
1c5bf1185a
Test ALTER SUBSCRIPTION ... SERVER and ALTER SUBSCRIPTION ... CONNECTION, including invalidation. Also run perltidy on the test file. Discussion: https://postgr.es/m/CAExHW5vV5znEvecX=ra2-v7UBj9-M6qvdDzuB78M-TxbYD1PEA@mail.gmail.com Suggested-by: Ashutosh Bapat <ashutosh.bapat.oss@gmail.com>
102 lines
3.5 KiB
Perl
102 lines
3.5 KiB
Perl
|
|
# Copyright (c) 2021-2026, PostgreSQL Global Development Group
|
|
|
|
# Test postgres_fdw foreign server for use with a subscription.
|
|
|
|
use strict;
|
|
use warnings FATAL => 'all';
|
|
use PostgreSQL::Test::Cluster;
|
|
use PostgreSQL::Test::Utils;
|
|
use Test::More;
|
|
|
|
# Initialize publisher node
|
|
my $node_publisher = PostgreSQL::Test::Cluster->new('publisher');
|
|
$node_publisher->init(allows_streaming => 'logical');
|
|
$node_publisher->start;
|
|
|
|
# Create subscriber node
|
|
my $node_subscriber = PostgreSQL::Test::Cluster->new('subscriber');
|
|
$node_subscriber->init;
|
|
$node_subscriber->start;
|
|
|
|
# Create some preexisting content on publisher
|
|
$node_publisher->safe_psql('postgres',
|
|
"CREATE TABLE tab_ins AS SELECT a, a + 1 as b FROM generate_series(1,1002) AS a"
|
|
);
|
|
|
|
# Setup structure on subscriber
|
|
$node_subscriber->safe_psql('postgres', "CREATE EXTENSION postgres_fdw");
|
|
$node_subscriber->safe_psql('postgres',
|
|
"CREATE TABLE tab_ins (a int, b int)");
|
|
|
|
# Setup logical replication
|
|
my $publisher_connstr = $node_publisher->connstr . ' dbname=postgres';
|
|
$node_publisher->safe_psql('postgres',
|
|
"CREATE PUBLICATION tap_pub FOR TABLE tab_ins");
|
|
|
|
my $publisher_host = $node_publisher->host;
|
|
my $publisher_port = $node_publisher->port;
|
|
$node_subscriber->safe_psql('postgres',
|
|
"CREATE SERVER tap_server FOREIGN DATA WRAPPER postgres_fdw OPTIONS (host '$publisher_host', port '$publisher_port', dbname 'postgres')"
|
|
);
|
|
|
|
$node_subscriber->safe_psql('postgres',
|
|
"CREATE USER MAPPING FOR PUBLIC SERVER tap_server");
|
|
|
|
$node_subscriber->safe_psql('postgres',
|
|
"CREATE SUBSCRIPTION tap_sub SERVER tap_server PUBLICATION tap_pub WITH (password_required=false)"
|
|
);
|
|
|
|
# Wait for initial table sync to finish
|
|
$node_subscriber->wait_for_subscription_sync($node_publisher, 'tap_sub');
|
|
|
|
my $result =
|
|
$node_subscriber->safe_psql('postgres', "SELECT MAX(a) FROM tab_ins");
|
|
is($result, qq(1002), 'check that initial data was copied to subscriber');
|
|
|
|
$node_publisher->safe_psql('postgres',
|
|
"INSERT INTO tab_ins SELECT a, a + 1 FROM generate_series(1003,1050) a");
|
|
|
|
$node_publisher->wait_for_catchup('tap_sub');
|
|
|
|
$result =
|
|
$node_subscriber->safe_psql('postgres', "SELECT MAX(a) FROM tab_ins");
|
|
is($result, qq(1050), 'check that inserted data was copied to subscriber');
|
|
|
|
# change to CONNECTION and confirm invalidation
|
|
my $log_offset = -s $node_subscriber->logfile;
|
|
$node_subscriber->safe_psql('postgres',
|
|
"ALTER SUBSCRIPTION tap_sub CONNECTION '$publisher_connstr'");
|
|
$node_subscriber->wait_for_log(
|
|
qr/logical replication worker for subscription "tap_sub" will restart because of a parameter change/,
|
|
$log_offset);
|
|
|
|
$node_publisher->safe_psql('postgres',
|
|
"INSERT INTO tab_ins SELECT a, a + 1 FROM generate_series(1051,1057) a");
|
|
|
|
$node_publisher->wait_for_catchup('tap_sub');
|
|
|
|
$result =
|
|
$node_subscriber->safe_psql('postgres', "SELECT MAX(a) FROM tab_ins");
|
|
is($result, qq(1057),
|
|
'check subscription after ALTER SUBSCRIPTION ... CONNECTION');
|
|
|
|
# change back to SERVER and confirm invalidation
|
|
$log_offset = -s $node_subscriber->logfile;
|
|
$node_subscriber->safe_psql('postgres',
|
|
"ALTER SUBSCRIPTION tap_sub SERVER tap_server");
|
|
$node_subscriber->wait_for_log(
|
|
qr/logical replication worker for subscription "tap_sub" will restart because of a parameter change/,
|
|
$log_offset);
|
|
|
|
$node_publisher->safe_psql('postgres',
|
|
"INSERT INTO tab_ins SELECT a, a + 1 FROM generate_series(1058,1073) a");
|
|
|
|
$node_publisher->wait_for_catchup('tap_sub');
|
|
|
|
$result =
|
|
$node_subscriber->safe_psql('postgres', "SELECT MAX(a) FROM tab_ins");
|
|
is($result, qq(1073),
|
|
'check subscription after ALTER SUBSCRIPTION ... SERVER');
|
|
|
|
done_testing();
|