go - 如何在 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 函数对于任何标志都不能为真。
我知道通过将我的地图值设置为 false,我实际上并没有取消设置传递的命令行值,但是这样做的方法是什么?
有没有办法在Golang 中取消访问/取消设置命令行标志?任何帮助将不胜感激。
谢谢。
解决方案
根据flag 的源代码,这应该可以工作(您需要将 os 添加到您的导入中):
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
}
推荐阅读
- javascript - 使用 NodeJS 在后端读取并提取音频音量强度作为数组
- mysql - MySQL - 从大表中的多个表中删除行
- fpga - 如何启动 FPGA 的 DDR 内存?
- python - 如何使用恒定的查找时间创建间隔到数字的映射?(如果这是可能的?)
- c# - 函数中的绝对数之和
- amazon-web-services - 使用 terraform 和 bitbucket 管道访问 S3
- android - 如何检查用户是第一次在谷歌登录android
- css - Ant Design中具有多种图像大小的轮播
- python - 从基本矩阵估计两个未校准相机之间的相对旋转
- neo4j - Neo4j 中的模式匹配