From 7e6e17d307ae875b7c169bc4523fcb1de7236072 Mon Sep 17 00:00:00 2001 From: Bharath Rupireddy Date: Wed, 9 Feb 2022 14:21:34 +0000 Subject: [PATCH v4] pg_walinspect tests --- .../pg_walinspect/expected/pg_walinspect.out | 83 +++++++++++++++++++ contrib/pg_walinspect/sql/pg_walinspect.sql | 68 +++++++++++++++ 2 files changed, 151 insertions(+) create mode 100644 contrib/pg_walinspect/expected/pg_walinspect.out create mode 100644 contrib/pg_walinspect/sql/pg_walinspect.sql diff --git a/contrib/pg_walinspect/expected/pg_walinspect.out b/contrib/pg_walinspect/expected/pg_walinspect.out new file mode 100644 index 0000000000..fdcba5accc --- /dev/null +++ b/contrib/pg_walinspect/expected/pg_walinspect.out @@ -0,0 +1,83 @@ +CREATE EXTENSION pg_walinspect; +SELECT pg_current_wal_lsn() AS wal_lsn1 \gset +CREATE TABLE sample_tbl(col1 int, col2 int); +INSERT INTO sample_tbl SELECT i, i+1 FROM generate_series(1, 100) i; +SELECT pg_current_wal_lsn() AS wal_lsn2 \gset +-- =================================================================== +-- tests for input validation +-- =================================================================== +SELECT pg_get_raw_wal_record('0/0'); -- ERROR +ERROR: invalid WAL record LSN +SELECT pg_get_first_valid_wal_record_lsn('0/0'); -- ERROR +ERROR: invalid WAL record LSN +SELECT pg_get_wal_record_info('0/0'); -- ERROR +ERROR: invalid WAL record LSN +SELECT pg_get_wal_records_info('0/0', '0/0'); -- ERROR +ERROR: invalid WAL record start LSN +SELECT pg_get_wal_records_info(:'wal_lsn1', '0/0'); -- ERROR +ERROR: invalid WAL record end LSN +SELECT pg_get_wal_records_info(:'wal_lsn2', :'wal_lsn1'); -- start LSN >= end LSN, ERROR +ERROR: WAL record start LSN must be less than end LSN +SELECT pg_get_wal_stats('0/0', '0/0'); -- ERROR +ERROR: invalid WAL record start LSN +SELECT pg_get_wal_stats(:'wal_lsn1', '0/0'); -- ERROR +ERROR: invalid WAL record end LSN +SELECT pg_get_wal_stats(:'wal_lsn2', :'wal_lsn1'); -- start LSN >= end LSN, ERROR +ERROR: WAL record start LSN must be less than end LSN +-- =================================================================== +-- tests for all function executions +-- =================================================================== +SELECT COUNT(*) >= 0 AS ok FROM pg_get_first_valid_wal_record_lsn(:'wal_lsn1'); + ok +---- + t +(1 row) + +SELECT pg_get_first_valid_wal_record_lsn(:'wal_lsn1') AS valid_wal_lsn1 \gset +SELECT pg_get_first_valid_wal_record_lsn(:'wal_lsn2') AS valid_wal_lsn2 \gset +SELECT COUNT(*) >= 0 AS ok FROM pg_get_raw_wal_record(:'valid_wal_lsn1'); + ok +---- + t +(1 row) + +SELECT record AS raw_wal_rec FROM pg_get_raw_wal_record(:'valid_wal_lsn1') \gset +SELECT pg_verify_raw_wal_record(:'raw_wal_rec'); + pg_verify_raw_wal_record +-------------------------- + t +(1 row) + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_record_info(:'valid_wal_lsn1'); + ok +---- + t +(1 row) + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_records_info(:'valid_wal_lsn1', :'valid_wal_lsn2'); + ok +---- + t +(1 row) + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_stats(:'valid_wal_lsn1', :'valid_wal_lsn2'); + ok +---- + t +(1 row) + +-- =================================================================== +-- tests for filtering out WAL records of a particular table +-- =================================================================== +SELECT oid AS sample_tbl_oid FROM pg_class WHERE relname = 'sample_tbl' \gset +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_records_info(:'valid_wal_lsn1', :'valid_wal_lsn2') + WHERE block_ref LIKE concat('%', :'sample_tbl_oid', '%') AND resource_manager = 'Heap'; + ok +---- + t +(1 row) + +-- =================================================================== +-- clean up +-- =================================================================== +DROP TABLE sample_tbl; diff --git a/contrib/pg_walinspect/sql/pg_walinspect.sql b/contrib/pg_walinspect/sql/pg_walinspect.sql new file mode 100644 index 0000000000..7d0c7398c8 --- /dev/null +++ b/contrib/pg_walinspect/sql/pg_walinspect.sql @@ -0,0 +1,68 @@ +CREATE EXTENSION pg_walinspect; + +SELECT pg_current_wal_lsn() AS wal_lsn1 \gset + +CREATE TABLE sample_tbl(col1 int, col2 int); + +INSERT INTO sample_tbl SELECT i, i+1 FROM generate_series(1, 100) i; + +SELECT pg_current_wal_lsn() AS wal_lsn2 \gset + +-- =================================================================== +-- tests for input validation +-- =================================================================== + +SELECT pg_get_raw_wal_record('0/0'); -- ERROR + +SELECT pg_get_first_valid_wal_record_lsn('0/0'); -- ERROR + +SELECT pg_get_wal_record_info('0/0'); -- ERROR + +SELECT pg_get_wal_records_info('0/0', '0/0'); -- ERROR + +SELECT pg_get_wal_records_info(:'wal_lsn1', '0/0'); -- ERROR + +SELECT pg_get_wal_records_info(:'wal_lsn2', :'wal_lsn1'); -- start LSN >= end LSN, ERROR + +SELECT pg_get_wal_stats('0/0', '0/0'); -- ERROR + +SELECT pg_get_wal_stats(:'wal_lsn1', '0/0'); -- ERROR + +SELECT pg_get_wal_stats(:'wal_lsn2', :'wal_lsn1'); -- start LSN >= end LSN, ERROR + +-- =================================================================== +-- tests for all function executions +-- =================================================================== + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_first_valid_wal_record_lsn(:'wal_lsn1'); + +SELECT pg_get_first_valid_wal_record_lsn(:'wal_lsn1') AS valid_wal_lsn1 \gset + +SELECT pg_get_first_valid_wal_record_lsn(:'wal_lsn2') AS valid_wal_lsn2 \gset + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_raw_wal_record(:'valid_wal_lsn1'); + +SELECT record AS raw_wal_rec FROM pg_get_raw_wal_record(:'valid_wal_lsn1') \gset + +SELECT pg_verify_raw_wal_record(:'raw_wal_rec'); + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_record_info(:'valid_wal_lsn1'); + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_records_info(:'valid_wal_lsn1', :'valid_wal_lsn2'); + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_stats(:'valid_wal_lsn1', :'valid_wal_lsn2'); + +-- =================================================================== +-- tests for filtering out WAL records of a particular table +-- =================================================================== + +SELECT oid AS sample_tbl_oid FROM pg_class WHERE relname = 'sample_tbl' \gset + +SELECT COUNT(*) >= 0 AS ok FROM pg_get_wal_records_info(:'valid_wal_lsn1', :'valid_wal_lsn2') + WHERE block_ref LIKE concat('%', :'sample_tbl_oid', '%') AND resource_manager = 'Heap'; + +-- =================================================================== +-- clean up +-- =================================================================== + +DROP TABLE sample_tbl; -- 2.25.1