使用Golang进行嵌套结构的Mock方法进行单元测试

在IT行业这个发展更新速度很快的行业，只有不停止的学习，才不会被行业所淘汰。如果你是Golang学习者，那么本文《使用Golang进行嵌套结构的Mock方法进行单元测试》就很适合你！本篇内容主要包括##content_title##，希望对大家的知识积累有所帮助，助力实战开发！

我想模拟嵌套结构的方法。我尝试定义一个接口并让 mock 实现它，但我无法让它工作。

这是我要测试的结构：

type outerthing struct {
    innerthing *innerthing
}

func (a *outerthing) dolotsofstuff() {
    println("i am doing")
    u, err := a.innerthing.dostuff("lots of stuff")
    if err != nil {
        println("ran into an error, also doing some logic")
    }
    println("and more", u)
}

我想模拟其 dostuff() 函数的嵌套结构如下所示：

type innerthing struct {
    name string
}

func (b *innerthing) dostuff(x string) (uint64, error) {
    println("i want to mock this method")
    return 0, nil
}

小问题：我无法更改这些结构及其方法的代码。

为了让我的观点更清楚一些，我编写了这个测试：

func TestDoLotsOfStuff(t *testing.T) {
    testCases := []struct {
        name            string
        structUnderTest *OuterThing
    }{
        {
            name: "happy case",
            structUnderTest: &OuterThing{
                innerThing: &InnerThing{name: "I am the inner thing."},
            },
        },
        {
            name: "error case",
            structUnderTest: &OuterThing{
                innerThing: &InnerThing{name: "i should be a mock with a mocked DoStuff function"},
            },
        },
    }

    for _, testCase := range testCases {
        t.Run(testCase.name, func(t *testing.T) {
            testCase.structUnderTest.doLotsOfStuff()
            // assertions
        })
    }
}

我非常熟悉 java 和 mockito，这将是一个非常简单的任务。我知道 go 的隐式接口和没有类等方面有很多差异，但这真的是一个不常见的用例吗？

正确答案

您可以通过使用接口和依赖注入来做到这一点。 假设您有一个名为 inner 的接口，其中包含 dostuff() 方法。 outerthing 结构应包含此接口，而不是结构 innerthing。现在，您可以使用依赖项注入将 inner 接口传递给此 outerthing 结构。

type inner interface {
    dostuff(string) (uint64, error)
}

type outerthing struct {
    innerthing inner
}

func newouterthing(in inner) *outerthing {
    return &outerthing{
        innerthing: in,
    }
}

现在，在测试中，您可以创建 mockinnerthing 结构体并实现模拟 dostuff() 方法，并将 mockinnerthing 传递给 outerthing 结构体。

type mockinnerthing struct {
    name string
}

func newmockinnerthing(name string) *mockinnerthing {
    return &mockinnerthing{
        name: name,
    }
}

func (b *mockinnerthing) dostuff(x string) (uint64, error) {
    println("mock!")
    return 0, nil
}

func TestDoLotsOfStuff(t *testing.T) {
    mit := NewMockInnerThing("Test")
    ot := NewOuterThing(mit)
    ot.doLotsOfStuff()
}

今天关于《使用Golang进行嵌套结构的Mock方法进行单元测试》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！