From 77e6cdb3966e90e1dfef2cc40e17838804306196 Mon Sep 17 00:00:00 2001
From: RidRisR <79858083+RidRisR@users.noreply.github.com>
Date: Mon, 19 Aug 2024 02:28:09 +0800
Subject: [PATCH] fix test

---
 tests/realtikvtest/brietest/brie_test.go | 49 ++++++------------------
 1 file changed, 11 insertions(+), 38 deletions(-)

diff --git a/tests/realtikvtest/brietest/brie_test.go b/tests/realtikvtest/brietest/brie_test.go
index a09aaac13eb7f..2f9b0a4bc87f9 100644
--- a/tests/realtikvtest/brietest/brie_test.go
+++ b/tests/realtikvtest/brietest/brie_test.go
@@ -137,42 +137,15 @@ func TestExistedTables(t *testing.T) {
 	sqlTmp := strings.ReplaceAll(tmp, "'", "''")
 	executor.ResetGlobalBRIEQueueForTest()
 	tk.MustExec("use test;")
-	for i := 0; i < 10; i++ {
-		tableName := fmt.Sprintf("foo%d", i)
-		tk.MustExec(fmt.Sprintf("create table %s(pk int primary key auto_increment, v varchar(255));", tableName))
-		tk.MustExec(fmt.Sprintf("insert into %s(v) values %s;", tableName, strings.TrimSuffix(strings.Repeat("('hello, world'),", 100), ",")))
-	}
-
-	done := make(chan struct{})
-	go func() {
-		defer close(done)
-		backupQuery := fmt.Sprintf("BACKUP DATABASE * TO 'local://%s'", sqlTmp)
-		_ = tk.MustQuery(backupQuery)
-	}()
-	select {
-	case <-time.After(20 * time.Second):
-		t.Fatal("Backup operation exceeded")
-	case <-done:
-	}
-
-	done = make(chan struct{})
-	go func() {
-		defer close(done)
-		restoreQuery := fmt.Sprintf("RESTORE DATABASE * FROM 'local://%s'", sqlTmp)
-		res, err := tk.Exec(restoreQuery)
-		require.NoError(t, err)
-
-		_, err = session.ResultSetToStringSlice(context.Background(), tk.Session(), res)
-		require.ErrorContains(t, err, "table already exists")
-	}()
-	select {
-	case <-time.After(20 * time.Second):
-		t.Fatal("Restore operation exceeded")
-	case <-done:
-	}
-
-	for i := 0; i < 10; i++ {
-		tableName := fmt.Sprintf("foo%d", i)
-		tk.MustExec(fmt.Sprintf("drop table %s;", tableName))
-	}
+	tk.MustExec("create table foo(pk int primary key auto_increment, v varchar(255));")
+	tk.MustExec("create table baa(pk int primary key auto_increment, v varchar(255));")
+	tk.MustExec("insert into foo(v) values " + strings.TrimSuffix(strings.Repeat("('hello, world'),", 100), ",") + ";")
+	tk.MustExec("insert into baa(v) values " + strings.TrimSuffix(strings.Repeat("('hello, world'),", 100), ",") + ";")
+	backupQuery := fmt.Sprintf("BACKUP DATABASE `test` TO 'local://%s'", sqlTmp)
+	tk.MustQuery(backupQuery)
+	restoreQuery := fmt.Sprintf("RESTORE DATABASE `test` FROM 'local://%s'", sqlTmp)
+	res, err := tk.Exec(restoreQuery)
+	require.NoError(t, err)
+	_, err = session.ResultSetToStringSlice(context.Background(), tk.Session(), res)
+	require.ErrorContains(t, err, "table already exists")
 }