roachtest: create required test schemas for SQLAlchemy test suite

Release note: None

Author: rafiss

pkg/cmd/roachtest/tests/sqlalchemy.go

@@ -128,6 +128,22 @@ func runSQLAlchemy(ctx context.Context, t test.Test, c cluster.Cluster) {
 		t.Fatal(err)
 	}
 
+	// The test suite requires two schemas to be created ahead of time.
+	// See https://github.com/sqlalchemy/sqlalchemy/blob/b3feb9cd3e87960ee65968b712e73d8464385a4a/README.unittests.rst?plain=1#L139.
+	db, err := c.ConnE(ctx, t.L(), node[0], option.DBName("defaultdb"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+	for _, cmd := range []string{
+		`CREATE SCHEMA IF NOT EXISTS test_schema`,
+		`CREATE SCHEMA IF NOT EXISTS test_schema_2`,
+	} {
+		if _, err := db.ExecContext(ctx, cmd); err != nil {
+			t.Fatal(err)
+		}
+	}
+
 	blocklistName, expectedFailures := "sqlAlchemyBlocklist", sqlAlchemyBlocklist
 	ignoredlistName, ignoredlist := "sqlAlchemyIgnoreList", sqlAlchemyIgnoreList
 	t.L().Printf("Running cockroach version %s, using blocklist %s, using ignoredlist %s",