func TestMain(m *testing.M) {
security.SetReadFileFn(securitytest.Asset)
randutil.SeedForTests()
serverutils.InitTestServerFactory(server.TestServerFactory)
serverutils.InitTestClusterFactory(testcluster.TestClusterFactory)
os.Exit(m.Run())
}
func main() {
// Seed the random number generator for non-determinism across
// multiple runs.
randutil.SeedForTests()
if f := flag.Lookup("alsologtostderr"); f != nil {
fmt.Println("Starting simulation. Add -alsologtostderr to see progress.")
}
flag.Parse()
dirName, err := ioutil.TempDir("", "gossip-simulation-")
if err != nil {
log.Fatalf(context.TODO(), "could not create temporary directory for gossip simulation output: %s", err)
}
// Simulation callbacks to run the simulation for cycleCount
// cycles. At each cycle % outputEvery, a dot file showing the
// state of the network graph is output.
nodeCount := 3
switch *size {
case "tiny":
// Use default parameters.
case "small":
nodeCount = 10
case "medium":
nodeCount = 25
case "large":
nodeCount = 50
case "huge":
nodeCount = 100
case "ginormous":
nodeCount = 250
default:
log.Fatalf(context.TODO(), "unknown simulation size: %s", *size)
}
edgeSet := make(map[string]edge)
stopper := stop.NewStopper()
defer stopper.Stop()
n := simulation.NewNetwork(stopper, nodeCount, true)
n.SimulateNetwork(
func(cycle int, network *simulation.Network) bool {
// Output dot graph.
dotFN := fmt.Sprintf("%s/sim-cycle-%03d.dot", dirName, cycle)
_, quiescent := outputDotFile(dotFN, cycle, network, edgeSet)
// Run until network has quiesced.
return !quiescent
},
)
// Output instructions for viewing graphs.
fmt.Printf("To view simulation graph output run (you must install graphviz):\n\nfor f in %s/*.dot ; do circo $f -Tpng -o $f.png ; echo $f.png ; done\n", dirName)
}
func TestMain(m *testing.M) {
randutil.SeedForTests()
serverutils.InitTestServerFactory(server.TestServerFactory)
// Create a set of all protos we believe to be marshalled downstream of raft.
// After the tests are run, we'll subtract the encountered protos from this
// set.
notBelowRaftProtos := make(map[reflect.Type]struct{}, len(belowRaftGoldenProtos))
for typ := range belowRaftGoldenProtos {
notBelowRaftProtos[typ] = struct{}{}
}
// Before running the tests, enable instrumentation that tracks protos which
// are marshalled downstream of raft.
stopTrackingAndGetTypes := storage.TrackRaftProtos()
code := m.Run()
// Only do this verification if the associated test was run. Without this
// condition, the verification here would spuriously fail when running a
// small subset of tests e.g. as we often do with `stress`.
if verifyBelowRaftProtos {
failed := false
// Retrieve all the observed downstream-of-raft protos and confirm that they
// are all present in our expected set.
for _, typ := range stopTrackingAndGetTypes() {
if _, ok := belowRaftGoldenProtos[typ]; ok {
delete(notBelowRaftProtos, typ)
} else {
failed = true
fmt.Printf("%s: missing fixture!\n", typ)
}
}
// Confirm that our expected set is now empty; we don't want to cement any
// protos needlessly.
for typ := range notBelowRaftProtos {
failed = true
fmt.Printf("%s: not observed below raft!\n", typ)
}
// Make sure our error messages make it out.
if failed && code == 0 {
code = 1
}
}
os.Exit(code)
}
func TestMain(m *testing.M) {
randutil.SeedForTests()
go func() {
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt)
<-sig
select {
case <-stopper:
default:
// There is a very tiny race here: the cluster might be closing
// the stopper simultaneously.
close(stopper)
}
}()
os.Exit(m.Run())
}
func runTests(m *testing.M) {
randutil.SeedForTests()
go func() {
// Shut down tests when interrupted (for example CTRL+C).
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt)
<-sig
select {
case <-stopper:
default:
// There is a very tiny race here: the cluster might be closing
// the stopper simultaneously.
close(stopper)
}
}()
os.Exit(m.Run())
}
func TestMain(m *testing.M) {
randutil.SeedForTests()
leaktest.TestMainWithLeakCheck(m)
}
func TestMain(m *testing.M) {
randutil.SeedForTests()
os.Exit(m.Run())
}
func TestMain(m *testing.M) {
randutil.SeedForTests()
m.Run()
}