如何在 GoLang 中取消设置在命令行上访问的标志以进行测试

Question

我正在尝试运行每次使用不同参数多次调用同一函数的测试。这是一个接受不同命令行标志的应用程序。如果未提供命令行标志，则使用默认值。flagset = make(map[string]bool)flagset["flag1"] = falseflagset["flag2"] = falseflagset["flag3"] = falseflagset["flag4"] = falsefunc LoadCommandLineArguments(args []string) error{        err := flag.CommandLine.Parse(args)        /*Do error handling         */       flag.Visit(func(f *flag.Flag) { flagset[f.Name] = true })       // Iterate through all the flags and set their default values whatever was not passed on the commandline.}func resetFlags(){/* Reset flag variables to their original default values and set map values to false */flagset["flag1"] = falseflagset["flag2"] = falseflagset["flag3"] = falseflagset["flag4"] = false}我有不同的测试函数，每个函数都为测试提供不同的标志。例如：这就是我的测试文件的样子func TestFlag1(t *testing.T) {   resetFlags()   err := LoadCommandLineArguments([]string{"-flag1=somevalue1"})}func TestFlag2(t *testing.T) {resetFlags()err := LoadCommandLineArguments([]string{"-flag2=somevalue2"})}func TestFlag3(t *testing.T) {resetFlags()err := LoadCommandLineArguments([]string{"-flag3=somevalue3"})}func TestFlag4(t *testing.T) {resetFlags()err := LoadCommandLineArguments([]string{""})}每当我单独运行测试文件和每个单元测试时，它们似乎都工作正常。但是当我一起运行整个文件测试文件时，当它到达 testFlag4 时，在 LoadCommandLineArguments 中，它会访问所有标志并认为所有标志都在命令行上传递，即使我在每个测试函数开始时重置它们。在 TestFlag1 中，它认为 flag1 已通过。在 TestFlag2 中，它认为 flag1 和 flag2 已通过。在TestFlag3中，它认为flag1、flag2、flag3通过了。同样，在TestFlag4中，它认为flag1也通过了，flag2也通过了，flag3也通过了，而我什么也没通过。我希望这个测试没有通过任何命令行标志。flags.Visit 函数对于任何标志都不应该为 true。我知道通过将我的地图值设置为 false，我实际上并没有取消设置传递的命令行值，但是如何做到这一点呢？有没有办法在 Golang 中取消访问/取消设置命令行标志？任何帮助将不胜感激。谢谢。

Answer 1

根据标志的源代码，这应该可以工作（您需要将操作系统添加到导入中）：

flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError) //flags are now reset

---

另外，就风格而言，还有一些事情：

当未找到键时，映射会返回默认值。所以所有这些初始化/重置行都可以删除。

flagset = make(map[string]bool)

// the initialization lines are not needed; bool already initalizes to false:

//flagset["flag1"] = false

//...

func LoadCommandLineArguments(args []string) error{

   //...

}

func resetFlags(){

   /* Reset flag variables */

   flagset = make(map[string]bool)//just make a new map

}