br: fix integration test cases with --with-sys-table

Signed-off-by: zhanggaoming <[email protected]>

br/pkg/restore/client.go

@@ -988,6 +988,26 @@ func (rc *Client) CheckSysTableCompatibility(dom *domain.Domain, tables []*metau
 					backupCol.Name, backupCol.FieldType.String())
 			}
 		}
+		// check whether the columns of table in cluster are less than the backup data
+		clusterColMap := make(map[string]*model.ColumnInfo)
+		for i := range ti.Columns {
+			col := ti.Columns[i]
+			clusterColMap[col.Name.L] = col
+		}
+		// order can be different
+		for i := range backupTi.Columns {
+			col := backupTi.Columns[i]
+			clusterCol := clusterColMap[col.Name.L]
+			if clusterCol == nil {
+				log.Error("missing column in cluster data",
+					zap.Stringer("table", table.Info.Name),
+					zap.String("col", fmt.Sprintf("%s %s", col.Name, col.FieldType.String())))
+				return errors.Annotatef(berrors.ErrRestoreIncompatibleSys,
+					"missing column in cluster data, table: %s, col: %s %s",
+					table.Info.Name.O,
+					col.Name, col.FieldType.String())
+			}
+		}
 	}
 	return nil
 }

br/pkg/restore/client_test.go

@@ -203,6 +203,16 @@ func TestCheckSysTableCompatibility(t *testing.T) {
 		Info: mockedUserTI,
 	}})
 	require.NoError(t, err)
+	userTI.Columns = userTI.Columns[:len(userTI.Columns)-1]
+
+	// user table in cluster have less columns(failed)
+	mockedUserTI = userTI.Clone()
+	mockedUserTI.Columns = append(mockedUserTI.Columns, &model.ColumnInfo{Name: model.NewCIStr("new-name")})
+	err = client.CheckSysTableCompatibility(cluster.Domain, []*metautil.Table{{
+		DB:   tmpSysDB,
+		Info: mockedUserTI,
+	}})
+	require.True(t, berrors.ErrRestoreIncompatibleSys.Equal(err))
 
 	// column order mismatch(success)
 	mockedUserTI = userTI.Clone()

br/tests/br_full_cluster_restore/run.sh

@@ -55,7 +55,8 @@ restart_services
 # mock incompatible manually
 run_sql "alter table mysql.user add column xx int;"
 run_br restore full --with-sys-table --log-file $br_log_file -s "local://$backup_dir" > $res_file 2>&1 || true
-check_contains "the target cluster is not compatible with the backup data"
+run_sql "select count(*) from mysql.user"
+check_contains "count(*): 6"
 
 echo "--> incompatible system table: less column on target cluster"
 restart_services