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

从现在开始，我们要努力学习啦！今天我给大家带来《如何在 GoLang 中取消设置在命令行上访问的标志以进行测试》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

我正在尝试运行测试，每次使用不同的参数多次调用同一函数。这是一个接受不同命令行标志的应用程序。如果未提供命令行标志，则使用默认值。

flagset = make(map[string]bool)
flagset["flag1"] = false
flagset["flag2"] = false
flagset["flag3"] = false
flagset["flag4"] = false
func 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"] = false
flagset["flag2"] = false
flagset["flag3"] = false
flagset["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 中取消访问/取消设置命令行标志？ 任何帮助将不胜感激。

谢谢。

解决方案

基于 source code for flag，这应该可以工作（您需要将操作系统添加到导入中）：

flag.commandline = flag.newflagset(os.args[0], flag.exitonerror) //flags are now reset

此外，还有一些风格问题：

当未找到键时，映射会返回类型的零值。类型的零值为 int 的 0、"" 的 string、false 的 bool 等。

因此所有这些初始化/重置行都可以删除：

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
}

今天关于《如何在 GoLang 中取消设置在命令行上访问的标志以进行测试》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！