diff --git a/br/tests/br_file_corruption/run.sh b/br/tests/br_file_corruption/run.sh index 6feb08f41e1f6..35a7698bb9fef 100644 --- a/br/tests/br_file_corruption/run.sh +++ b/br/tests/br_file_corruption/run.sh @@ -26,11 +26,24 @@ run_br --pd $PD_ADDR backup full -s "local://$TEST_DIR/$DB" filename=$(find $TEST_DIR/$DB -regex ".*.sst" | head -n 1) filename_temp=$filename"_temp" +filename_bak=$filename"_bak" echo "corruption" > $filename_temp cat $filename >> $filename_temp + +# file lost +mv $filename $filename_bak +export GO_FAILPOINTS="github.com/pingcap/tidb/br/pkg/utils/set-import-attempt-to-one=return(true)" +restore_fail=0 +run_br --pd $PD_ADDR restore full -s "local://$TEST_DIR/$DB" || restore_fail=1 +export GO_FAILPOINTS="" +if [ $restore_fail -ne 1 ]; then + echo 'restore success' + exit 1 +fi + +# file corruption mv $filename_temp $filename truncate --size=-11 $filename - export GO_FAILPOINTS="github.com/pingcap/tidb/br/pkg/utils/set-import-attempt-to-one=return(true)" restore_fail=0 run_br --pd $PD_ADDR restore full -s "local://$TEST_DIR/$DB" || restore_fail=1 diff --git a/br/tests/br_pitr/run.sh b/br/tests/br_pitr/run.sh index 1ae66f17a1a5a..35143c0afa0b3 100644 --- a/br/tests/br_pitr/run.sh +++ b/br/tests/br_pitr/run.sh @@ -165,8 +165,22 @@ restart_services echo "corrupt a log file" filename=$(find $TEST_DIR/$PREFIX/log -regex ".*\.log" | grep -v "schema-meta" | tail -n 1) filename_temp=$filename"_temp" +filename_bak=$filename"_bak" echo "corruption" > $filename_temp cat $filename >> $filename_temp + +# file lost +mv $filename $filename_bak +export GO_FAILPOINTS="github.com/pingcap/tidb/br/pkg/utils/set-import-attempt-to-one=return(true)" +restore_fail=0 +run_br --pd $PD_ADDR restore point -s "local://$TEST_DIR/$PREFIX/log" --full-backup-storage "local://$TEST_DIR/$PREFIX/full" || restore_fail=1 +export GO_FAILPOINTS="" +if [ $restore_fail -ne 1 ]; then + echo 'pitr success' + exit 1 +fi + +# file corruption mv $filename_temp $filename truncate --size=-11 $filename export GO_FAILPOINTS="github.com/pingcap/tidb/br/pkg/utils/set-import-attempt-to-one=return(true)"