谷歌云函数 Golang 单元测试与 init()

本篇文章向大家介绍《谷歌云函数 Golang 单元测试与 init()》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

我在 golang 中创建云函数并为其创建单元测试时遇到问题。具体来说，我想要在测试阶段注入的环境变量无法识别，因为 init() 函数始终在测试之前运行。此 init() 函数使用环境变量来创建客户端。有什么办法可以解决这个问题吗？

google 建议将客户端等全局变量放入 init() 函数中，以便在将来的调用中重用这些对象。这减少了启动时间（来源）。

目录结构

/function
  do_something.go
  do_something_test.go

do_something.go

package main

import (
    "context"
    "log"
    "os"

    "cloud.google.com/go/firestore"
)

var (
    projectid = os.getenv("gcloud_project")
    client    *firestore.client
)

func init() {
    ctx := context.background()
    var err error
    client, err = firestore.newclient(ctx, projectid)
    if err != nil {
        log.fatalf("firestore: %v", err)
    }
}

func main(w http.responsewriter, r *http.request) {
    // do something
}

do_something_test.go

package main

import (
    "os"
    "testing"
)

func init() {
    os.setenv("gcloud_project", "my-test-project") // this does not work because of lexigraphical order, init() in do_something.go before this init().
}

func testmain(t *testing.t) {
    os.setenv("gcloud_project", "my-test-project") // this does not work because init() is run before testmain.
}

可能的解决方案

1. 请勿在 do_something.go 中使用 init()。但是，这会降低云功能性能，由于启动时间延长而导致延迟增加。
2. 通过配置文件注入变量。但是，我认为这会带来同样的问题。
3. 在运行 go test 之前设置环境变量。这是可行的，但我发现这是一个次优的解决方案，因为我不想在进行实际测试之前运行额外的参数。
4. 始终默认为开发环境，除非环境变量指定代码在生产环境中运行：

projectID = GetEnv("GCLOUD_PROJECT", "my-test-project")

func GetEnv(key string, defaultVal string) string {
    if value, exists := os.LookupEnv(key); exists {
        return value
    }
    return defaultVal
}

正确答案

您应该在项目级别公开一个环境变量，说明代码是否在 test 设置或 prod 设置中执行。基于此，您的 init 函数将设置变量的值。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。