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
2017-05-22 13:17:43 +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 )