1
0
mirror of https://github.com/postgrespro/pg_probackup.git synced 2024-12-04 10:44:46 +02:00
pg_probackup/tests/ptrack_recovery.py

59 lines
2.4 KiB
Python
Raw Normal View History

2017-06-27 07:42:52 +02:00
import os
2017-05-03 13:14:48 +02:00
import unittest
from sys import exit
2017-06-27 07:42:52 +02:00
from .helpers.ptrack_helpers import ProbackupTest, idx_ptrack
2017-05-03 13:14:48 +02:00
2017-07-12 16:28:28 +02:00
module_name = 'ptrack_recovery'
2017-05-03 13:14:48 +02:00
2017-07-12 16:28:28 +02:00
class SimpleTest(ProbackupTest, unittest.TestCase):
2017-05-03 13:14:48 +02:00
# @unittest.skip("skip")
# @unittest.expectedFailure
2017-05-03 13:14:48 +02:00
def test_ptrack_recovery(self):
2017-06-20 12:57:23 +02:00
fname = self.id().split('.')[3]
2017-07-12 16:28:28 +02:00
node = self.make_simple_node(base_dir="{0}/{1}/node".format(module_name, fname),
2017-05-03 13:14:48 +02:00
set_replication=True,
2017-06-20 12:57:23 +02:00
initdb_params=['--data-checksums'],
2017-05-03 13:14:48 +02:00
pg_options={'ptrack_enable': 'on', 'wal_level': 'replica', 'max_wal_senders': '2'})
2017-07-12 16:28:28 +02:00
backup_dir = os.path.join(self.tmp_path, module_name, fname, 'backup')
2017-06-20 12:57:23 +02:00
self.init_pb(backup_dir)
self.add_instance(backup_dir, 'node', node)
2017-05-03 13:14:48 +02:00
node.start()
2017-06-20 12:57:23 +02:00
2017-05-03 13:14:48 +02:00
self.create_tblspace_in_node(node, 'somedata')
# Create table
2017-07-12 16:28:28 +02:00
node.safe_psql("postgres",
"create sequence t_seq; create table t_heap tablespace somedata as select i as id, md5(i::text) as text,md5(repeat(i::text,10))::tsvector as tsvector from generate_series(0,256) i")
2017-05-03 13:14:48 +02:00
# Create indexes
for i in idx_ptrack:
2017-07-12 16:28:28 +02:00
if idx_ptrack[i]['type'] != 'heap' and idx_ptrack[i]['type'] != 'seq':
node.safe_psql("postgres", "create index {0} on {1} using {2}({3}) tablespace somedata".format(
2017-05-03 13:14:48 +02:00
i, idx_ptrack[i]['relation'], idx_ptrack[i]['type'], idx_ptrack[i]['column']))
# get size of heap and indexes. size calculated in pages
idx_ptrack[i]['size'] = int(self.get_fork_size(node, i))
# get path to heap and index files
idx_ptrack[i]['path'] = self.get_fork_path(node, i)
2017-06-27 07:42:52 +02:00
if self.verbose:
print('Killing postmaster. Losing Ptrack changes')
2017-11-07 16:50:35 +02:00
node.stop(['-m', 'immediate', '-D', node.data_dir])
2017-05-03 13:14:48 +02:00
if not node.status():
node.start()
else:
2017-06-27 07:42:52 +02:00
print("Die! Die! Why won't you die?... Why won't you die?")
2017-05-03 13:14:48 +02:00
exit(1)
for i in idx_ptrack:
# get ptrack for every idx
2017-05-05 15:21:49 +02:00
idx_ptrack[i]['ptrack'] = self.get_ptrack_bits_per_page_for_fork(
2017-07-12 16:28:28 +02:00
node, idx_ptrack[i]['path'], [idx_ptrack[i]['size']])
2017-05-03 13:14:48 +02:00
# check that ptrack has correct bits after recovery
self.check_ptrack_recovery(idx_ptrack[i])
2017-06-27 07:42:52 +02:00
# Clean after yourself
2017-07-12 16:28:28 +02:00
self.del_test_dir(module_name, fname)