#!/usr/bin/perl

use strict;
use warnings;
use File::Basename;
use Text::Trim qw(trim);
use Test::More;
use lib 't';
use pgsm;

# Get file name and CREATE out file name and dirs WHERE requried
PGSM::setup_files_dir(basename($0));

# CREATE new PostgreSQL node and do initdb
my $node = PGSM->pgsm_init_pg();

$node->append_conf(
	'postgresql.conf', qq(
shared_preload_libraries = 'pg_stat_monitor'
# Set bucket duration to 1800 seconds so bucket doesn't change.
pg_stat_monitor.pgsm_bucket_time = 1800
track_io_timing = on
log_temp_files = 0
work_mem = 64kB
pg_stat_monitor.pgsm_track_utility = off
pg_stat_monitor.pgsm_normalized_query = on
));

# Start server
$node->start;

# CREATE EXTENSION and change out file permissions
my ($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'CREATE EXTENSION pg_stat_statements;',
	extra_params => ['-a']);
is($cmdret, 0, "CREATE PGSS EXTENSION");
PGSM::append_to_file($stdout);

# CREATE EXTENSION and change out file permissions
($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'CREATE EXTENSION pg_stat_monitor;',
	extra_params => ['-a']);
is($cmdret, 0, "CREATE PGSM EXTENSION");
PGSM::append_to_file($stdout);

# Run required commands/queries and dump output to out file.
($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'SELECT pg_stat_monitor_reset();',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "Reset PGSM EXTENSION");
PGSM::append_to_file($stdout);

# Run 'SELECT ***' two times and dump output to out file
($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	"SELECT name, setting, unit, context, vartype, source, min_val, max_val, enumvals, boot_val, reset_val, pending_restart FROM pg_settings WHERE name LIKE '%pg_stat_monitor%';",
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "Print PGSM EXTENSION Settings");
PGSM::append_to_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'CREATE TABLE t1 (a int);',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "CREATE Table t1");
PGSM::append_to_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'CREATE INDEX idx_t1_a ON t1 (a);',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "CREATE index.");
PGSM::append_to_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'INSERT INTO t1 SELECT generate_series(1, 1000000);',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "INSERT 10000 records.");
PGSM::append_to_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql('postgres', 'ANALYZE t1;',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "ANALYZE t1.");
PGSM::append_to_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'SELECT * FROM t1 AS xx INNER JOIN t1 AS tt ON xx.a = tt.a;',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
is($cmdret, 0, "SELECT * FROM t1.");
PGSM::append_to_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'SELECT substr(query, 0, 130) AS query, calls, rows_retrieved, cpu_user_time, cpu_sys_time, local_blks_hit, local_blks_read, local_blks_dirtied, local_blks_written, temp_blks_read, temp_blks_written FROM pg_stat_monitor WHERE query LIKE \'%t1%\' ORDER BY query, calls DESC;',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
PGSM::append_to_debug_file($stdout);

($cmdret, $stdout, $stderr) = $node->psql(
	'postgres',
	'SELECT temp_files, temp_bytes FROM pg_stat_database db;',
	extra_params => [ '-a', '-Pformat=aligned', '-Ptuples_only=off' ]);
PGSM::append_to_debug_file($stdout);

# Compare values for query 'SELECT * FROM t1'
($cmdret, $stdout, $stderr) = $node->psql('postgres',
	'SELECT pgsm.temp_blks_read <> 0 FROM pg_stat_monitor AS pgsm WHERE pgsm.query LIKE \'%FROM t1%\';'
);
trim($stdout);
is($stdout, 't', "Check: temp_blks_read should not be 0.");

($cmdret, $stdout, $stderr) = $node->psql('postgres',
	'SELECT pgsm.temp_blks_written <> 0 FROM pg_stat_monitor AS pgsm WHERE pgsm.query LIKE \'%FROM t1%\';'
);
trim($stdout);
is($stdout, 't', "Check: temp_blks_written should not be 0.");

if ($PGSM::PG_MAJOR_VERSION >= 15)
{
	($cmdret, $stdout, $stderr) = $node->psql('postgres',
		'SELECT sum(pgsm.temp_blk_read_time) <> 0 FROM pg_stat_monitor AS pgsm WHERE pgsm.query LIKE \'%FROM t1%\';'
	);
	trim($stdout);
	is($stdout, 't', "Check: temp_blk_read_time should not be 0.");

	($cmdret, $stdout, $stderr) = $node->psql('postgres',
		'SELECT sum(pgsm.temp_blk_write_time) <> 0 FROM pg_stat_monitor AS pgsm WHERE pgsm.query LIKE \'%FROM t1%\''
	);
	trim($stdout);
	is($stdout, 't', "Check: temp_blk_write_time should not be 0.");
}

# Stop the server
$node->stop;

# Done testing for this testcase file.
done_testing();