// Starts up a cluster made of up `nodes` in-memory testing servers, // creates database `name and returns open gosql.DB connections to each // node (to the named db), as well as a cleanup func that stops and // cleans up all nodes and connections. func SetupMultinodeTestCluster( t testing.TB, nodes int, name string, ) (MultinodeTestCluster, []*gosql.DB, *stop.Stopper) { if nodes < 1 { t.Fatal("invalid cluster size: ", nodes) } stopper := stop.NewStopper() // Force all ranges to be replicated everywhere. This is needed until #7297 is // fixed, otherwise starting a cluster takes forever. cfg := config.DefaultZoneConfig() cfg.ReplicaAttrs = make([]roachpb.Attributes, nodes) fn := config.TestingSetDefaultZoneConfig(cfg) stopper.AddCloser(stop.CloserFn(fn)) var servers []serverutils.TestServerInterface var conns []*gosql.DB args := base.TestServerArgs{ Stopper: stopper, PartOfCluster: true, UseDatabase: name, } first, conn, _ := serverutils.StartServer(t, args) servers = append(servers, first) conns = append(conns, conn) args.JoinAddr = first.ServingAddr() for i := 1; i < nodes; i++ { s, conn, _ := serverutils.StartServer(t, args) servers = append(servers, s) conns = append(conns, conn) } if _, err := conns[0].Exec(fmt.Sprintf(`CREATE DATABASE %s`, name)); err != nil { t.Fatal(err) } testCluster := MultinodeTestCluster{Servers: servers} return testCluster, conns, first.Stopper() }