Пример #1
0
func init() {
	runtime.GOMAXPROCS(runtime.NumCPU())
	flag.Int("vv", 0, "vv")
	var fs = flag.NewFlagSet("deblocus", flag.ErrorHandling(0xffff))
	var null, _ = os.Open(os.DevNull)
	defer null.Close()
	fs.SetOutput(null)
	var v int
	fs.IntVar(&v, "vv", 0, "log verbose")
	var deblocusArgs []string
	deblocusArgs = append(deblocusArgs, os.Args[1:]...)
	fs.Parse(deblocusArgs)

	log.SetLogOutput("")
	log.SetLogVerbose(v)
	cltAddr += strconv.FormatInt(randomRange(1, 1<<13)+3e4, 10)
	svrAddr += strconv.FormatInt(randomRange(1, 1<<13)+3e4, 10)
	dstAddr += strconv.FormatInt(randomRange(1, 1<<13)+3e4, 10)
	fmt.Println("=== deblocus TEST ===")
	printArgLine("logV", v)
	printArgLine("cltAddr", cltAddr)
	printArgLine("svrAddr", svrAddr)
	printArgLine("dstAddr", dstAddr)
	fmt.Println("\n")
}
Пример #2
0
func initFlagSets(_flagsets interface{}) {
	flagsets := _flagsets.([]interface{})
	for i, _flagset := range flagsets {
		flagset := _flagset.(map[string]interface{})

		errorHandling := flagset["errorHandling"].(float64)
		name := flagset["name"].(string)

		if errorHandling < 0 || errorHandling > 2 {
			log.Fatalf("Unable to parse flagset at index \"%d\"\n", i)
		}
		if len(name) == 0 {
			log.Fatalf("Unable to parse flagset at index %d\n", i)
		}

		newFlagSet := flag.NewFlagSet(name, flag.ErrorHandling(errorHandling))
		flagvars.m[name] = make(map[string]interface{})
		err := initFlags(flagset["flags"], name, newFlagSet)
		if err != nil {
			log.Fatalf("Unable to parse flagset at index %d: %s\n", i, err)
		}
		if index := getArgIndex(name); index > -1 {
			newFlagSet.Parse(os.Args[index:])
		}
	}
}
Пример #3
0
// NewCmdVar defines a cmd with specified name, and explain string.
// The argument cmd points to a Cmd variable in which to store the flags of the cmd.
func (c *CmdSet) NewCmdVar(cmd *Cmd, name string, explain string) {
	cmd.Name = name
	cmd.Explain = explain
	cmd.Init(name, flag.ErrorHandling(c.errorHandling))
	cmd.SetOutput(c.output)
	c.cmds[name] = cmd
	if len(name) > c.maxCmdLen {
		c.maxCmdLen = len(name)
	}
}
Пример #4
0
func GetModule(key string) (Module, bool) {
	lock.Lock()
	defer lock.Unlock()

	cf, has := modules[key]
	if has {
		v, p := cf()
		policies[key] = flag.ErrorHandling(p)
		return v, true
	}
	return nil, false
}
Пример #5
0
func TestBadFlags(t *testing.T) {

	// Test that this line *does* break:
	teststring := []string{"-bad"}
	const ContinueOnError = true
	devnull, _ := os.OpenFile(os.DevNull, os.O_WRONLY, 0700)
	flag.CommandLine.SetOutput(devnull)
	flag.Usage = func() {}
	flag.CommandLine.Init("testflags", flag.ErrorHandling(0))
	e := flag.CommandLine.Parse(teststring)
	if e == nil {
		fmt.Println("\tExpected Error. Got nil.")
		t.FailNow()
	} else if e.Error() == "flag provided but not defined: -bad" {
		//
	} else {
		fmt.Println("\tExpected a different error:", e)
	}
	return
}
Пример #6
0
// Test that this line does not break:
// 'cosgo -gpg /etc/aerth.asc -fastcgi -log cosgo.log -port 8089'
// Lots of flags: bind config debug fastcgi gpg log mbox new nolog port quiet sg title to
// No more breaking crontabs on cosgo upgrades!
// These flag tests need to be at the bottom of the test suite
func TestFlags(t *testing.T) {

	// This causes some races so...
	if !*debug {
		return
	}
	const PanicOnError = true
	devnull, _ := os.OpenFile(os.DevNull, os.O_WRONLY, 0700)
	flag.CommandLine.SetOutput(devnull)
	flag.CommandLine.Init("testflags", flag.ErrorHandling(2))
	flag.Usage = func() {}
	orig := os.Args
	os.Args = []string{"cosgo", "-gpg", "/etc/aerth.asc", "-fastcgi", "-log", "cosgo.log", "-port", "8089"}
	flag.Parse()
	os.Args = []string{"cosgo", "-mbox", "test.mbox", "-bind", "0.0.0.0", "-debug"}
	flag.Parse()
	os.Args = []string{"cosgo", "-title", "Test Page", "-fastcgi", "-log", "cosgo.log", "-port", "8089"}
	flag.Parse()
	os.Args = []string{"cosgo", "-config", "/tmp/config.cosgo", "-nolog", "-sg", "sgkey123", "-to", "*****@*****.**"}
	flag.Parse()
	os.Args = orig
	flag.Parse()
	return
}