如何测试涉及大量环境配置和操作系统参数的函数？

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《如何测试涉及大量环境配置和操作系统参数的函数？》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

我已经编写了一个 go 应用程序，并且所有包都具有完整的测试覆盖率。我正在编写 main 包 - 它将处理 main() 函数中应用程序的所有初始设置 - 该函数当前读取 14 个环境变量，然后在应用程序中设置相关变量。代码的简单概述是：

func main() {
    mystruct1 := privatepackage.mystructtype{}
    mystruct2 := publicpackage.otherstructtype{}


    if config1 := os.getenv("config_foo"); config1 != "" {
        mystruct1.attribute1 = config1
    }

    // ....

    if config14 := os.getenv("config_bar"); config14 != "" {
        mystruct2.attribute5 = config14
    }
}

当我测试单元环境变量/操作系统参数时，我通常只是直接在测试函数中设置环境变量 - 类似于：

我几乎总是使用表驱动的测试，因此上面的代码片段是一个简化的示例。

问题是我的 main() 函数接受 14 个（并且不断增长）环境变量，而某些环境变量本质上是枚举（因此有少量有效选项 - 例如有少量数据库驱动程序）可供选择），其他环境变量实际上具有无限的潜在值。那么我怎样才能有效地覆盖潜在配置的所有（或足够的）排列呢？

编辑：部署此应用程序后，它将进入 k8s 集群。其中一些变量是将从安全存储中提取的秘密。使用 json 文件是不可行的，因为某些值需要轻松加密/更改。 此外，使用 json 文件需要我存储该文件并在数百/数千个正在运行的 pod 之间共享它 - 该存储将成为故障点。 需要澄清的是，这个问题与环境变量与配置文件无关；而是关于环境变量与配置文件的问题。这个问题是关于当存在大量可配置变量时进行测试的最佳方法 - 每个变量都有大量潜在值 - 导致数千种可能的配置排列。在这种情况下如何保证足够的测试覆盖率？

正确答案

@steven penny 是对的：使用 json

• 并且使用reflect可以使代码更加简单：

package main

import (
    "encoding/json"
    "fmt"
    "os"
    "reflect"
    "strconv"
)

type MyStructType struct {
    Attribute1 string `json:"CONFIG_FOO"`
    Attribute2 string `json:"CONFIG_BAZ"`
    Attribute3 int `json:"CONFIG_BAR"`
}

func NewMyStructTypeFormEnv() *MyStructType {
    myStructType := MyStructType{}
    ReflectMyStructType(&myStructType)
    fmt.Println("myStructType is now", myStructType)
    return &myStructType
}

func NewMyStructTypeFormJson() *MyStructType {
    myStructType := MyStructType{}
    f, e := os.Open("file.json")
    if e != nil {
        panic(e)
    }
    defer f.Close()
    json.NewDecoder(f).Decode(&myStructType)
    fmt.Println("myStructType is now", myStructType)
    return &myStructType
}

func ReflectMyStructType(ptr interface{}){
    v := reflect.ValueOf(ptr).Elem()
    fmt.Printf("%v\n", v.Type())
    for i := 0; i < v.NumField(); i++ {
        env_str := v.Type().Field(i).Tag.Get("json")
        if(env_str == ""){continue}
        if config := os.Getenv(env_str); config != "" {
            if v.Field(i).Kind() == reflect.String{
                v.Field(i).SetString(config)
            }else if v.Field(i).Kind() == reflect.Int{
                iConfig,_ := strconv.Atoi(config)
                v.Field(i).SetInt(int64(iConfig))
            }

        }
    }
}

func main() {
    NewMyStructTypeFormJson()
    
    os.Setenv("CONFIG_FOO", "apple")
    os.Setenv("CONFIG_BAZ", "apple")
    os.Setenv("CONFIG_BAR", "1")
    NewMyStructTypeFormEnv()
}

到这里，我们也就讲完了《如何测试涉及大量环境配置和操作系统参数的函数？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！