使用模拟测试替代实际调用 NewManager()

golang学习网今天将给大家带来《使用模拟测试替代实际调用 NewManager()》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

我有一个用 go 编写的自定义 kubernetes 运算符。在编写单元测试时，我发现需要初始化一个假管理器。管理器是使用“sigs.k8s.io/controller-runtime”包的 newmanager() 函数创建的。在本地运行时这不是问题，但在管道中运行时却是问题。

import (
    ctrl "sigs.k8s.io/controller-runtime"
    // other imports
)

var _ = Describe("ProvideNewController", func() {
    var (
        reconciler  reconcile.Reconciler
        customUtils controllers.CustomUtils // local utils package
        mockCC      *controllers.MockCommandClient
        errResult   error
    )
    When("ProvideNewController is called", func() {
        BeforeEach(func() {
            logger := logr.Discard()
            recorder := record.NewFakeRecorder(100)
            mockCC = &controllers.MockCommandClient{}
            customUtils = controllers.CustomUtils{
                CommandClient: mockCC,
            }
            scheme := scheme.Scheme
            scheme.AddKnownTypes(operatorV1alpha1.GroupVersion, &operatorV1alpha1.Custom{})
            managerOptions := ctrl.Options{
                Scheme:           scheme,
                Logger:           ctrl.Log.WithName("controllers").WithName("custom"),
                WebhookServer:    webhook.NewServer(webhook.Options{Port: 9443}),
                LeaderElectionID: "someValue",
            }
            fakeClient := fake.NewClientBuilder().WithScheme(scheme).WithRuntimeObjects().Build()
            reconciler = NewReconciler(fakeClient, customUtils, logger, scheme, recorder)
            mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), managerOptions)
            if err != nil {
                common.FailTestSetup(err)
            }
            errResult = ProvideNewController(mgr, reconciler)
        })
        It("should set up a controller with the specified resource types", func() {
            Expect(errResult).ToNot(HaveOccurred())
        })
    })
})

现在， mgr, err := ctrl.newmanager(ctrl.getconfigordie(), manageroptions) 行使测试完全崩溃，因为 ctrl.getconfigordie() 失败读取 kubeconfig 文件，这是预期的，因为它确实不存在于 ci/cd 运行程序的 pod 上。

我是 go 新手，所以我不知道所有选项。从我所看到的来看，如果我要模拟整个结构并将其模拟为 newmanager 的参数，我将不得不模拟结构实现的所有方法，并且我不认为这是一个好方法。

有没有办法让我模拟配置文件或以编程方式提供它，这样我就不必编辑管道并从那里提供它？

正确答案

您可以使用 sigs.k8s.io 控制器运行时存储库的 envtest 包。 请参阅configuring envtest for integration测试 文档页面下的 writing test 部分：

import sigs.k8s.io/controller-runtime/pkg/envtest

//specify testEnv configuration
testEnv = &envtest.Environment{
    CRDDirectoryPaths: []string{filepath.Join("..", "config", "crd", "bases")},
}

//start testEnv
cfg, err = testEnv.Start()

//write test logic

//stop testEnv
err = testEnv.Stop()

到这里，我们也就讲完了《使用模拟测试替代实际调用 NewManager()》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！