@@ -30,9 +30,10 @@ def test_restore_to_latest_1(self):
3030backup_log .write (self .backup_pb (node ,options = ["--verbose" ]))
3131
3232node .stop ({"-m" :"immediate" })
33+ node .cleanup ()
3334
34- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
35- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
35+ self . assertIn ( six . b ( "INFO: restore complete" ),
36+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
3637
3738node .start ({"-t" :"600" })
3839
@@ -61,9 +62,10 @@ def test_restore_to_latest_2(self):
6162before = node .execute ("postgres" ,"SELECT * FROM pgbench_branches" )
6263
6364node .stop ({"-m" :"immediate" })
65+ node .cleanup ()
6466
65- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
66- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
67+ self . assertIn ( six . b ( "INFO: restore complete" ),
68+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]));
6769
6870node .start ({"-t" :"600" })
6971
@@ -86,9 +88,10 @@ def test_restore_to_timeline_3(self):
8688
8789target_tli = int (self .get_control_data (node )[six .b ("Latest checkpoint's TimeLineID" )])
8890node .stop ({"-m" :"immediate" })
91+ node .cleanup ()
8992
90- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
91- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
93+ self . assertIn ( six . b ( "INFO: restore complete" ),
94+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
9295
9396node .start ({"-t" :"600" })
9497
@@ -100,12 +103,11 @@ def test_restore_to_timeline_3(self):
100103backup_log .write (self .backup_pb (node ,backup_type = "full" ,options = ["--verbose" ]))
101104
102105node .stop ({"-m" :"immediate" })
106+ node .cleanup ()
103107
104- with open (path .join (node .logs_dir ,"restore_2.log" ),"wb" )as restore_log :
105- restore_log .write (self .restore_pb (
106- node ,
107- options = ["-j" ,"4" ,"--verbose" ,"--timeline=%i" % target_tli ]
108- ))
108+ self .assertIn (six .b ("INFO: restore complete" ),
109+ self .restore_pb (node ,
110+ options = ["-j" ,"4" ,"--verbose" ,"--timeline=%i" % target_tli ]))
109111
110112recovery_target_timeline = self .get_recovery_conf (node )["recovery_target_timeline" ]
111113self .assertEqual (int (recovery_target_timeline ),target_tli )
@@ -135,12 +137,11 @@ def test_restore_to_time_4(self):
135137pgbench .stdout .close ()
136138
137139node .stop ({"-m" :"immediate" })
140+ node .cleanup ()
138141
139- with open (path .join (node .logs_dir ,"restore_1.log" ),"wb" )as restore_log :
140- restore_log .write (self .restore_pb (
141- node ,
142- options = ["-j" ,"4" ,"--verbose" ,'--time="%s"' % target_time ]
143- ))
142+ self .assertIn (six .b ("INFO: restore complete" ),
143+ self .restore_pb (node ,
144+ options = ["-j" ,"4" ,"--verbose" ,'--time="%s"' % target_time ]))
144145
145146node .start ({"-t" :"600" })
146147
@@ -182,12 +183,11 @@ def test_restore_to_xid_5(self):
182183node .execute ("postgres" ,"SELECT pg_switch_xlog()" )
183184
184185node .stop ({"-m" :"fast" })
186+ node .cleanup ()
185187
186- with open (path .join (node .logs_dir ,"restore_1.log" ),"wb" )as restore_log :
187- restore_log .write (self .restore_pb (
188- node ,
189- options = ["-j" ,"4" ,"--verbose" ,'--xid=%s' % target_xid ]
190- ))
188+ self .assertIn (six .b ("INFO: restore complete" ),
189+ self .restore_pb (node ,
190+ options = ["-j" ,"4" ,"--verbose" ,'--xid=%s' % target_xid ]))
191191
192192node .start ({"-t" :"600" })
193193
@@ -224,9 +224,10 @@ def test_restore_full_ptrack_6(self):
224224before = node .execute ("postgres" ,"SELECT * FROM pgbench_branches" )
225225
226226node .stop ({"-m" :"immediate" })
227+ node .cleanup ()
227228
228- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
229- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
229+ self . assertIn ( six . b ( "INFO: restore complete" ),
230+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
230231
231232node .start ({"-t" :"600" })
232233
@@ -270,9 +271,10 @@ def test_restore_full_ptrack_ptrack_7(self):
270271before = node .execute ("postgres" ,"SELECT * FROM pgbench_branches" )
271272
272273node .stop ({"-m" :"immediate" })
274+ node .cleanup ()
273275
274- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
275- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
276+ self . assertIn ( six . b ( "INFO: restore complete" ),
277+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
276278
277279node .start ({"-t" :"600" })
278280
@@ -293,7 +295,10 @@ def test_restore_full_ptrack_stream_8(self):
293295self .skipTest ("ptrack not supported" )
294296return
295297
298+ node .append_conf ("pg_hba.conf" ,"local replication all trust" )
299+ node .append_conf ("pg_hba.conf" ,"host replication all 127.0.0.1/32 trust" )
296300node .append_conf ("postgresql.conf" ,"ptrack_enable = on" )
301+ node .append_conf ("postgresql.conf" ,"max_wal_senders = 1" )
297302node .restart ()
298303
299304with open (path .join (node .logs_dir ,"backup_1.log" ),"wb" )as backup_log :
@@ -309,9 +314,10 @@ def test_restore_full_ptrack_stream_8(self):
309314before = node .execute ("postgres" ,"SELECT * FROM pgbench_branches" )
310315
311316node .stop ({"-m" :"immediate" })
317+ node .cleanup ()
312318
313- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
314- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
319+ self . assertIn ( six . b ( "INFO: restore complete" ),
320+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
315321
316322node .start ({"-t" :"600" })
317323
@@ -333,7 +339,10 @@ def test_restore_full_ptrack_under_load_9(self):
333339self .skipTest ("ptrack not supported" )
334340return
335341
342+ node .append_conf ("pg_hba.conf" ,"local replication all trust" )
343+ node .append_conf ("pg_hba.conf" ,"host replication all 127.0.0.1/32 trust" )
336344node .append_conf ("postgresql.conf" ,"ptrack_enable = on" )
345+ node .append_conf ("postgresql.conf" ,"max_wal_senders = 1" )
337346node .restart ()
338347
339348with open (path .join (node .logs_dir ,"backup_1.log" ),"wb" )as backup_log :
@@ -358,11 +367,12 @@ def test_restore_full_ptrack_under_load_9(self):
358367
359368self .assertEqual (bbalance ,delta )
360369node .stop ({"-m" :"immediate" })
370+ node .cleanup ()
361371
362372self .wrong_wal_clean (node ,wal_segment_size )
363373
364- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
365- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
374+ self . assertIn ( six . b ( "INFO: restore complete" ),
375+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
366376
367377node .start ({"-t" :"600" })
368378
@@ -386,7 +396,10 @@ def test_restore_full_under_load_ptrack_10(self):
386396self .skipTest ("ptrack not supported" )
387397return
388398
399+ node .append_conf ("pg_hba.conf" ,"local replication all trust" )
400+ node .append_conf ("pg_hba.conf" ,"host replication all 127.0.0.1/32 trust" )
389401node .append_conf ("postgresql.conf" ,"ptrack_enable = on" )
402+ node .append_conf ("postgresql.conf" ,"max_wal_senders = 1" )
390403node .restart ()
391404
392405pgbench = node .pgbench (
@@ -412,10 +425,11 @@ def test_restore_full_under_load_ptrack_10(self):
412425self .assertEqual (bbalance ,delta )
413426
414427node .stop ({"-m" :"immediate" })
428+ node .cleanup ()
415429self .wrong_wal_clean (node ,wal_segment_size )
416430
417- with open ( path . join ( node . logs_dir , "restore_1.log" ), "wb" ) as restore_log :
418- restore_log . write ( self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
431+ self . assertIn ( six . b ( "INFO: restore complete" ),
432+ self .restore_pb (node ,options = ["-j" ,"4" ,"--verbose" ]))
419433
420434node .start ({"-t" :"600" })
421435
@@ -459,17 +473,15 @@ def test_restore_to_xid_inclusive_11(self):
459473node .execute ("postgres" ,"SELECT pg_switch_xlog()" )
460474
461475node .stop ({"-m" :"fast" })
476+ node .cleanup ()
462477
463- with open (path .join (node .logs_dir ,"restore_1.log" ),"wb" )as restore_log :
464- restore_log .write (self .restore_pb (
465- node ,
478+ self .assertIn (six .b ("INFO: restore complete" ),
479+ self .restore_pb (node ,
466480options = [
467481"-j" ,"4" ,
468482"--verbose" ,
469483'--xid=%s' % target_xid ,
470- "--inclusive=false"
471- ]
472- ))
484+ "--inclusive=false" ]))
473485
474486node .start ({"-t" :"600" })
475487