Skip to content

Commit c4b6ef8

Browse files
craig[bot]msbutler
andcommitted
154339: backup: deflake TestFullClusterRestoreWithUserIDs r=kev-cao a=msbutler Fixes #154006 Release note: none 154340: external connection: pass name as parameter r=kev-cao a=msbutler Epic: none Release note: none 154394: roachtest: set positive workload seed in or recovery test r=kev-cao a=msbutler Informs #154345 Release note: none Co-authored-by: Michael Butler <[email protected]>
4 parents a9fe917 + f27c66a + 06ce54c + 12d84c9 commit c4b6ef8

File tree

3 files changed

+19
-18
lines changed

3 files changed

+19
-18
lines changed

pkg/backup/full_cluster_backup_restore_test.go

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1189,29 +1189,29 @@ func TestFullClusterRestoreWithUserIDs(t *testing.T) {
11891189
sqlDB.Exec(t, `CREATE USER test2`)
11901190
sqlDB.Exec(t, `BACKUP INTO $1`, localFoo)
11911191

1192-
sqlDB.CheckQueryResults(t, `SELECT username, "hashedPassword", "isRole", user_id FROM system.users ORDER BY user_id`, [][]string{
1193-
{"root", "", "false", "1"},
1194-
{"admin", "", "true", "2"},
1195-
{"test1", "NULL", "false", "100"},
1196-
{"test2", "NULL", "false", "101"},
1192+
sqlDB.CheckQueryResults(t, `SELECT username, "hashedPassword", "isRole" FROM system.users ORDER BY user_id`, [][]string{
1193+
{"root", "", "false"},
1194+
{"admin", "", "true"},
1195+
{"test1", "NULL", "false"},
1196+
{"test2", "NULL", "false"},
11971197
})
11981198
// Ensure that the new backup succeeds.
11991199
sqlDBRestore.Exec(t, `RESTORE FROM LATEST IN $1`, localFoo)
12001200

1201-
sqlDBRestore.CheckQueryResults(t, `SELECT username, "hashedPassword", "isRole", user_id FROM system.users ORDER BY user_id`, [][]string{
1202-
{"root", "", "false", "1"},
1203-
{"admin", "", "true", "2"},
1204-
{"test1", "NULL", "false", "100"},
1205-
{"test2", "NULL", "false", "101"},
1201+
sqlDBRestore.CheckQueryResults(t, `SELECT username, "hashedPassword", "isRole" FROM system.users ORDER BY user_id`, [][]string{
1202+
{"root", "", "false"},
1203+
{"admin", "", "true"},
1204+
{"test1", "NULL", "false"},
1205+
{"test2", "NULL", "false"},
12061206
})
12071207

12081208
sqlDBRestore.Exec(t, `CREATE USER test3`)
12091209

1210-
sqlDBRestore.CheckQueryResults(t, `SELECT username, "hashedPassword", "isRole", user_id FROM system.users ORDER BY user_id`, [][]string{
1211-
{"root", "", "false", "1"},
1212-
{"admin", "", "true", "2"},
1213-
{"test1", "NULL", "false", "100"},
1214-
{"test2", "NULL", "false", "101"},
1215-
{"test3", "NULL", "false", "102"},
1210+
sqlDBRestore.CheckQueryResults(t, `SELECT username, "hashedPassword", "isRole" FROM system.users ORDER BY user_id`, [][]string{
1211+
{"root", "", "false"},
1212+
{"admin", "", "true"},
1213+
{"test1", "NULL", "false"},
1214+
{"test2", "NULL", "false"},
1215+
{"test3", "NULL", "false"},
12161216
})
12171217
}

pkg/cloud/externalconn/record.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ func LoadExternalConnection(
9494
// `SELECT` on the system table.
9595
row, cols, err := txn.QueryRowExWithCols(ctx, "lookup-schedule", txn.KV(),
9696
sessiondata.NodeUserSessionDataOverride,
97-
fmt.Sprintf("SELECT * FROM system.external_connections WHERE connection_name = '%s'", name))
97+
"SELECT * FROM system.external_connections WHERE connection_name = $1", name)
9898

9999
if err != nil {
100100
return nil, errors.CombineErrors(err, &ExternalConnectionNotFoundError{connectionName: name})

pkg/cmd/roachtest/tests/backup_restore_roundtrip.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,7 @@ func createDriversForBackupRestore(
448448

449449
func testOnlineRestoreRecovery(ctx context.Context, t test.Test, c cluster.Cluster) {
450450
testRNG, seed := randutil.NewLockedPseudoRand()
451+
workloadSeed := testRNG.Int63()
451452
t.L().Printf("random seed: %d", seed)
452453

453454
c.Start(ctx, t.L(), roachtestutil.MaybeUseMemoryBudget(t, 50), install.MakeClusterSettings(), c.CRDBNodes())
@@ -466,7 +467,7 @@ func testOnlineRestoreRecovery(ctx context.Context, t test.Test, c cluster.Clust
466467

467468
dbs := []string{"bank", "tpcc", schemaChangeDB}
468469
d, runBackgroundWorkload, _, err := createDriversForBackupRestore(
469-
ctx, t, c, m, testRNG, seed, testUtils, dbs,
470+
ctx, t, c, m, testRNG, workloadSeed, testUtils, dbs,
470471
)
471472
if err != nil {
472473
return err

0 commit comments

Comments
 (0)