如何对返回通道类型的Golang函数进行测试？

大家好，今天本人给大家带来文章《如何对返回通道类型的Golang函数进行测试？》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我尝试测试 startp 函数，

预计 start() 应该被调用 1 次，done() 应该被调用 1 次

但是我遇到了问题，运行此步骤时测试将被阻止<-ps.done()

我期望 <-ps.done() 返回 nil

如何测试返回 chan 类型的函数？

// production code

func (s *vservice) startp(ctx context.context, reason string) error {

    ps, err := s.factory.createvservice(ctx)
    if err != nil {
        return err
    }
    ps.start(reason)

    err = <-ps.done()   // code stop here to wait ? how can i test ?

    if err != nil {
        return err
    }
    return nil
}

// test code

func test_startp(t *testing.t) {
    mockctrl := gomock.newcontroller(t)
    defer mockctrl.finish()

    mockpservice := mockpservice.newmockpinterface(mockctrl)

    vservice := &vservice {
                      factory: &servicefactory.fakeservicefactory{
                                 mockpservice: mockpservice
                               }
                    }

    mockpservice.expect().start("reason").times(1).return()
    mockpservice.expect().done().times(1).doandreturn(func() chan error {
        return nil
    })

    err := vservice.startp(context.background(), "reason")
    assert.equal(t, nil, err)
}

我使用 gomock 来模拟 pserviceinterface

// interface

type pserviceinterface interface {
    start(reason string)
    done() <-chan error
}

gomock 生成这个函数

func (m *mockprovisionserviceinterface) done() <-chan error {
        m.ctrl.t.helper()
        ret := m.ctrl.call(m, "done")
        ret0, _ := ret[0].(<-chan error)
        fmt.println(ret0,".....mock done()")
        return ret0
}

//我也试试这个

mockProvisionService.EXPECT().Done().Times(1).DoAndReturn( func() chan error {
        fmt.Println("DoAndReturn...err nil")

        ch := make(chan error, 1)
        ch <- nil
        return ch
    })

正确答案

我找到了答案，根本原因是 doandreturn 出了问题。

func 类型应为 <-chan error，而不是 chan error

mockprovisionservice.expect().done().times(1).doandreturn( func() <-chan error{
            fmt.println("doandreturn...err nil")
            ch := make(chan error, 1)
            ch <- nil
            return ch
        })

我认为，下面显示了实现您的测试目标的最少代码。

它不使用任何模拟框架，因为根据我的经验，它们往往会混淆测试意图，要求团队中的每个人学习如何使用它们，但至少在 go 中是不需要的。人们还可能想知道测试实际上在测试什么......

首先，让我们添加一些缺失的生产代码：

type factoryinterface interface {
    createvservice(ctx context.context) (pserviceinterface, error)
}

type vservice struct {
    factory factoryinterface
}

现在是测试代码，分为三个部分：工厂、模拟和测试。

测试工厂：

type testfactory struct {
    mock pserviceinterface
}

func (f *testfactory) createvservice(ctx context.context) (pserviceinterface, error) {
    return f.mock, nil
}

模拟：

type servicemock struct {
    records []string
}

func (sm *servicemock) start(reason string) {
    sm.records = append(sm.records, "start")
}

func (sm *servicemock) done() <-chan error {
    sm.records = append(sm.records, "done")
    ch := make(chan error)
    close(ch)
    return ch
}

最后是测试：

func TestWithMock(t *testing.T) {
    mock := ServiceMock{}
    sut := &vService{factory: &testFactory{&mock}}

    err := sut.StartP(context.Background(), "banana")
    if err != nil {
        t.Fatalf("StartP: have: %s; want: no error", err)
    }

    if have, want := len(mock.records), 2; have != want {
        t.Fatalf("number of mock calls: have: %v; want: %v", have, want)
    }

    if have, want := mock.records[0], "start"; have != want {
        t.Fatalf("mock call 1: have: %v; want: %v", have, want)
    }

    if have, want := mock.records[1], "done"; have != want {
        t.Fatalf("mock call 2: have: %v; want: %v", have, want)
    }
}

如果使用测试库，例如优秀的 assert 包，则模拟调用序列上的三个断言可以合并为一个，直接比较切片 []string{"start", "done"} https://github.com/gotestyourself/gotest.tools

终于介绍完啦！小伙伴们，这篇关于《如何对返回通道类型的Golang函数进行测试？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！