github.com/pingcap/br@v5.3.0-alpha.0.20220125034240-ec59c7b6ce30+incompatible/tests/lightning_restore/run.sh (about)

     1  #!/bin/sh
     2  #
     3  # Copyright 2019 PingCAP, Inc.
     4  #
     5  # Licensed under the Apache License, Version 2.0 (the "License");
     6  # you may not use this file except in compliance with the License.
     7  # You may obtain a copy of the License at
     8  #
     9  #     http://www.apache.org/licenses/LICENSE-2.0
    10  #
    11  # Unless required by applicable law or agreed to in writing, software
    12  # distributed under the License is distributed on an "AS IS" BASIS,
    13  # See the License for the specific language governing permissions and
    14  # limitations under the License.
    15  
    16  set -eu
    17  
    18  # Populate the mydumper source
    19  DBPATH="$TEST_DIR/restore.mydump"
    20  TABLE_COUNT=35
    21  
    22  mkdir -p $DBPATH
    23  echo 'CREATE DATABASE restore_tsr;' > "$DBPATH/restore_tsr-schema-create.sql"
    24  for i in $(seq "$TABLE_COUNT"); do
    25      echo "CREATE TABLE tbl$i(i TINYINT);" > "$DBPATH/restore_tsr.tbl$i-schema.sql"
    26      echo "INSERT INTO tbl$i VALUES (1);" > "$DBPATH/restore_tsr.tbl$i.sql"
    27  done
    28  
    29  # Count OpenEngine and CloseEngine events.
    30  # Abort if number of unbalanced OpenEngine is >= 4
    31  export GO_FAILPOINTS='github.com/pingcap/br/pkg/lightning/backend/FailIfEngineCountExceeds=return(4)'
    32  
    33  # Start importing
    34  run_sql 'DROP DATABASE IF EXISTS restore_tsr'
    35  run_lightning -d "$DBPATH"
    36  echo "Import finished"
    37  
    38  # Verify all data are imported
    39  for i in $(seq "$TABLE_COUNT"); do
    40      run_sql "SELECT sum(i) FROM restore_tsr.tbl$i;"
    41      check_contains 'sum(i): 1'
    42  done