tests: fix for backup.BackupTest.test_backup_detect_corruption

gsmolk · gsmolk · commit 8fb9f94c6d21 · 2019-07-16T01:18:00.000+03:00
diff --git a/tests/backup.py b/tests/backup.py
@@ -469,12 +469,16 @@ def test_backup_detect_corruption(self):
             "postgres",
             "select pg_relation_filepath('t_heap')").rstrip()
 
+        node.stop()
+
         with open(os.path.join(node.data_dir, heap_path), "rb+", 0) as f:
                 f.seek(9000)
                 f.write(b"bla")
                 f.flush()
                 f.close
 
+        node.slow_start()
+
         try:
             self.backup_node(
                 backup_dir, 'node', node,
@@ -486,23 +490,33 @@ def test_backup_detect_corruption(self):
                 "\n Output: {0} \n CMD: {1}".format(
                     repr(self.output), self.cmd))
         except ProbackupException as e:
-            if self.remote:
+            if self.ptrack:
                 self.assertTrue(
-                    "ERROR: Failed to read file" in e.message and
-                    "data file checksum mismatch" in e.message,
+                    'WARNING:  page verification failed, '
+                    'calculated checksum' in e.message and
+                    'ERROR: query failed: ERROR:  '
+                    'invalid page in block 1 of relation' in e.message and
+                    'ERROR: Data files transferring failed' in e.message,
                     '\n Unexpected Error Message: {0}\n CMD: {1}'.format(
                         repr(e.message), self.cmd))
             else:
-                self.assertIn(
-                    'WARNING: Corruption detected in file',
-                    e.message,
-                    '\n Unexpected Error Message: {0}\n CMD: {1}'.format(
-                        repr(e.message), self.cmd))
-                self.assertIn(
-                    'ERROR: Data file corruption',
-                    e.message,
-                    '\n Unexpected Error Message: {0}\n CMD: {1}'.format(
-                        repr(e.message), self.cmd))
+                if self.remote:
+                    self.assertTrue(
+                        "ERROR: Failed to read file" in e.message and
+                        "data file checksum mismatch" in e.message,
+                        '\n Unexpected Error Message: {0}\n CMD: {1}'.format(
+                            repr(e.message), self.cmd))
+                else:
+                    self.assertIn(
+                        'WARNING: Corruption detected in file',
+                        e.message,
+                        '\n Unexpected Error Message: {0}\n CMD: {1}'.format(
+                            repr(e.message), self.cmd))
+                    self.assertIn(
+                        'ERROR: Data file corruption',
+                        e.message,
+                        '\n Unexpected Error Message: {0}\n CMD: {1}'.format(
+                            repr(e.message), self.cmd))
 
         # Clean after yourself
         self.del_test_dir(module_name, fname)
