159 lines
7.8 KiB
Perl
159 lines
7.8 KiB
Perl
#!/usr/bin/perl
|
|
|
|
use strict;
|
|
use warnings;
|
|
use String::Util qw(trim);
|
|
use File::Basename;
|
|
use File::Compare;
|
|
use PostgresNode;
|
|
use Test::More;
|
|
|
|
# Expected folder where expected output will be present
|
|
my $expected_folder = "t/expected";
|
|
|
|
# Results/out folder where generated results files will be placed
|
|
my $results_folder = "t/results";
|
|
|
|
# Check if results folder exists or not, create if it doesn't
|
|
unless (-d $results_folder)
|
|
{
|
|
mkdir $results_folder or die "Can't create folder $results_folder: $!\n";;
|
|
}
|
|
|
|
# Check if expected folder exists or not, bail out if it doesn't
|
|
unless (-d $expected_folder)
|
|
{
|
|
BAIL_OUT "Expected files folder $expected_folder doesn't exist: \n";;
|
|
}
|
|
|
|
# Get filename of the this perl file
|
|
my $perlfilename = basename($0);
|
|
|
|
#Remove .pl from filename and store in a variable
|
|
$perlfilename =~ s/\.[^.]+$//;
|
|
my $filename_without_extension = $perlfilename;
|
|
|
|
# Create expected filename with path
|
|
my $expected_filename = "${filename_without_extension}.out";
|
|
my $expected_filename_with_path = "${expected_folder}/${expected_filename}" ;
|
|
|
|
# Create results filename with path
|
|
my $out_filename = "${filename_without_extension}.out";
|
|
my $out_filename_with_path = "${results_folder}/${out_filename}" ;
|
|
|
|
# Delete already existing result out file, if it exists.
|
|
if ( -f $out_filename_with_path)
|
|
{
|
|
unlink($out_filename_with_path) or die "Can't delete already existing $out_filename_with_path: $!\n";
|
|
}
|
|
|
|
# Create new PostgreSQL node and do initdb
|
|
my $node = PostgresNode->get_new_node('test');
|
|
my $pgdata = $node->data_dir;
|
|
$node->dump_info;
|
|
$node->init;
|
|
|
|
# Update postgresql.conf to include/load pg_stat_monitor library
|
|
open my $conf, '>>', "$pgdata/postgresql.conf";
|
|
print $conf "shared_preload_libraries = 'pg_stat_monitor'\n";
|
|
print $conf "pg_stat_monitor.pgsm_overflow_target = 0\n";
|
|
print $conf "pg_stat_monitor.pgsm_max = 1\n";
|
|
print $conf "pg_stat_monitor.pgsm_query_shared_buffer = 1\n";
|
|
print $conf "pg_stat_monitor.pgsm_query_max_len =10000\n";
|
|
print $conf " pg_stat_monitor.pgsm_enable_query_plan = 'yes'\n";
|
|
print $conf "pg_stat_monitor.pgsm_track_planning = 'yes'\n";
|
|
print $conf "pg_stat_monitor.pgsm_track = 'top'\n";
|
|
print $conf "pg_stat_monitor.pgsm_extract_comments = 'no'\n";
|
|
close $conf;
|
|
|
|
# Start server
|
|
my $rt_value = $node->start;
|
|
ok($rt_value == 1, "Start Server");
|
|
|
|
# Create extension and change out file permissions
|
|
my ($cmdret, $stdout, $stderr) = $node->psql('postgres', 'CREATE EXTENSION pg_stat_monitor;', extra_params => ['-a']);
|
|
ok($cmdret == 0, "Create PGSM Extension");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
chmod(0640 , $out_filename_with_path)
|
|
or die("unable to set permissions for $out_filename_with_path");
|
|
|
|
# Run required commands/queries and dump output to out file.
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'create table foo (id int generated by default as identity,col1 varchar(100) not null,primary key(id));', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Create Table foo");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT pg_stat_monitor_reset();', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Reset PGSM Extension");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT * from pg_stat_monitor_settings;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Print PGSM Extension Settings");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', '\i scripts/data_1.sql' , extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Run sql file: scripts/data.sql");
|
|
#TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT substr(query, 0, 50), length(query), bucket, queryid, calls, elevel, sqlcode, message from pg_stat_monitor;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "SELECT substr(query, 0, 50), calls, message from pg_stat_monitor");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT count(queryid) from pg_stat_monitor;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "SELECT count(queryid) from pg_stat_monitor");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_max = 100\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_query_max_len = 1024\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_track_utility = 'yes'\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_normalized_query = 'yes'\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_max_buckets = 10\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_bucket_time = 60\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_histogram_min = 0\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_histogram_max = 100000\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_histogram_buckets = 10\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_query_shared_buffer = 20\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_overflow_target = 1\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_enable_query_plan = 'no'\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_track = 'top'\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_extract_comments = 'no'\n");
|
|
$node->append_conf('postgresql.conf', "pg_stat_monitor.pgsm_track_planning = 'no'\n");
|
|
$node->restart();
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT pg_stat_monitor_reset();', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Reset PGSM Extension");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT * from pg_stat_monitor_settings;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Print PGSM Extension Settings");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', '\i scripts/data_2.sql' , extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Run sql file: scripts/data2.sql");
|
|
#TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT substr(query, 0, 50), length(query), bucket, queryid, calls, elevel, sqlcode, message from pg_stat_monitor;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "SELECT substr(query, 0, 50), calls, message from pg_stat_monitor");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT count(queryid) from pg_stat_monitor;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "SELECT count(queryid) from pg_stat_monitor");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
($cmdret, $stdout, $stderr) = $node->psql('postgres', 'SELECT * from pg_stat_monitor_settings;', extra_params => ['-a', '-Pformat=aligned','-Ptuples_only=off']);
|
|
ok($cmdret == 0, "Print PGSM Extension Settings");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
# Drop extension
|
|
$stdout = $node->safe_psql('postgres', 'Drop extension pg_stat_monitor;', extra_params => ['-a']);
|
|
ok($cmdret == 0, "Drop PGSM Extension");
|
|
TestLib::append_to_file($out_filename_with_path, $stdout . "\n");
|
|
|
|
# Stop the server
|
|
$node->stop;
|
|
|
|
# Done testing for this testcase file.
|
|
done_testing();
|
|
|
|
|