Function TestMain

dgraph/cmd/alpha/run_test.go:1678–1728

(m *testing.M)

}

func TestMain(m *testing.M) {
	conf := dgraphtest.NewClusterConfig().WithNumAlphas(1).WithNumZeros(1).WithReplicas(1).WithACL(time.Hour)
	c, err := dgraphtest.NewLocalCluster(conf)

	x.Panic(err)
	x.Panic(c.Start())
	var cleanup func()
	dg, cleanup, err = c.Client()
	x.Panic(err)
	defer cleanup()

	x.Panic(dg.Login(context.Background(), dgraphapi.DefaultUser, dgraphapi.DefaultPassword))

	alphaGrpcPort, err := c.GetAlphaGrpcPublicPort(0)
	x.Panic(err)

	alphaSockAdd = "0.0.0.0:" + alphaGrpcPort
	alphaSockAddHttp, err := c.GetAlphaHttpPublicPort(0)
	x.Panic(err)
	addr = "http://0.0.0.0:" + alphaSockAddHttp
	zeroSockAdd, err := c.GetZeroGrpcPublicPort(0)
	x.Panic(err)

	// Increment lease, so that mutations work.
	conn, err := grpc.NewClient("0.0.0.0:"+zeroSockAdd, grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		log.Fatal(err)
	}
	zc := pb.NewZeroClient(conn)
	if _, err := zc.AssignIds(context.Background(),
		&pb.Num{Val: 1e6, Type: pb.Num_UID}); err != nil {
		log.Fatal(err)
	}
	hc, err = c.HTTPClient()
	x.Panic(err)

	x.Panic(hc.LoginIntoNamespace(dgraphapi.DefaultUser, dgraphapi.DefaultPassword, 0))

	httpToken := hc.HttpToken
	token = &Token{
		token:   httpToken,
		RWMutex: sync.RWMutex{},
	}
	code := m.Run()
	if code != 0 {
		c.Cleanup(true)
	} else {
		c.Cleanup(false)
	}
	os.Exit(code)
}