Function TestMain

client/candy_test.go:44–101

(main *testing.M)

}

func TestMain(main *testing.M) {
	userNames = make(map[string]int64)
	passwd = make(map[string]string)

	debug := flag.Bool("V", false, "set log level:debug")
	flag.Parse()
	if *debug {
		log.SetLevel(log.LOG_DEBUG)
	} else {
		log.SetLevel(log.LOG_ERROR)
	}

	exes := []string{
		"../bin/master",
		"../bin/notice",
		"../bin/store",
		"../bin/gate",
	}

	cmds := []*exec.Cmd{}

	for _, exe := range exes {
		cmd := exec.Command(exe)
		if err := cmd.Start(); err != nil {
			panic(err.Error())
		}
		cmds = append(cmds, cmd)
	}

	time.Sleep(time.Second * 3)

	client = NewCandyClient("0.0.0.0:9000", &cmdClient{})
	if err := client.Start(); err != nil {
		panic(err.Error())
	}

	for i := 0; i < 10; i++ {
		name := fmt.Sprintf("testuser_%v_%d@qq.com", time.Now().Unix(), i)
		pass := fmt.Sprintf("testpass_%v_%d", time.Now().Unix(), i)
		id, err := client.Register(name, pass)
		if err != nil {
			for _, cmd := range cmds {
				cmd.Process.Kill()
			}
			panic("Register error:" + err.Error())
		}
		userNames[name] = id
		passwd[name] = pass
	}

	ret := main.Run()

	for _, cmd := range cmds {
		cmd.Process.Kill()
	}

	os.Exit(ret)
}