在 Go 测试中模拟对象方法的调用

你在学习Golang相关的知识吗？本文《在 Go 测试中模拟对象方法的调用》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

我希望在测试中模拟结构的方法以提高代码覆盖率。 有一些关于此的帖子，但没有一个对我有用。我可能完全错了。

main/file1.go

type application struct {
    name string
 }

func (app *application) find() error {
    // perform function logic
    return nil
}

func main() {
    app := &application{
        name:   "main application",
    }

    err := app.find()
    if err != nil {
        fmt.printf("error in find call: %s\n", err)
    }
}

我需要能够模拟测试 find() 并返回 error （我不想生成可能导致错误的测试用例，因为这不在我的控制范围内，我不知道如何通过传递可接受的参数生成一个）。 我尝试遵循这篇文章的第二个答案，但编译器不喜欢它。

main/file1_test.go

func Test_application_find(t *testing.T) {
    tests := []struct {
        name           string
        app            *application
        wantErr        string
    }{
        {
            name: "generate error",
            app: &application{
                Name: "Mock Application",
            },
            wantErr: true,
        },
    }
    for _, tt := range tests {
        mockCaller := tt.app.find // this works fine
        tt.app.find = func() error { // this assignment errors out
            return fmt.Errorf("Mock Error Message")
        }
        defer func() {
            tt.app.find = mockCaller // this assignment errors out
        }()

        t.Run(tt.name, func(t *testing.T) {
            if err := tt.app.find(); (err != nil) && (err.Error() != "Mock Error Message") {
                    t.Errorf("error = %s, wantErr %s", err.Error(), tt.wantErr)
            }
        } 
    }
}

我在这里做错了什么？请提出建议。

解决方案

模拟方法有多种方法。他们都有适合的情况。

组成

对于测试，您在原始结构之上创建一个新结构。

struct testapp struct{
    application
}

现在 teststruct 的行为方式与 app 几乎相同，但您可以覆盖 find 函数：

func (app *testapp) find() error {
    return errors.new("some error")
}

在测试中，您使用给定的正确 app 来初始化 testapp：

app := testapp{
    application: &application{
        name: "mock application",
    }
}

现在您可以在 app 变量上编写测试。

请注意，这并不适用于所有用例，因为组合不是继承。在原始 application 上调用函数不会调用 testapp 中新的 find 函数。

界面

您还可以模拟整个应用程序结构/测试所需的部分。如果您想测试依赖于 application 的另一个结构，这通常是有意义的。

要测试的结构不应直接使用 application ，而应使用其所需功能的接口。 （这可能意味着要重构您的代码才能做到这一点。）

然后使用模拟的 find 函数创建一个模拟，并将该模拟传递给要测试的结构：

type app interface{
    find() error
}

type mockapp struct{
    name string
}

(s *mockapp) find() error {
    return errors.new("some error")
}

结构体字段

您还可以将 find 函数设为结构体字段。然后您可以覆盖它以进行测试。然而，这样做的缺点是必须更改代码。

type application struct {
    name string
    find func() error
}

编写一个创建者，以便仍然能够以合理的方式初始化结构：

func newapplication() *application {
    return &application{
        name: "main application",
        find: func() error {
            // perform function logic
            return nil
        }
    }
}

现在您可以在测试中覆盖 find 方法（实际上它现在是一个字段）：

app := newApplication()
app.find = func() error {
    return errors.New("some error")
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~