Golang 测试中的夹具

Question

来自 python 世界，fixtures 非常有用（Fixtures 为可重用状态/支持逻辑定义了一个 Python 契约，主要用于单元测试）。我想知道 Golang 是否有类似的支持，它可以让我使用一些预定义的装置来运行我的测试，比如设置服务器，拆除它，每次运行测试时都做一些重复的任务？有人可以指出我在 Golang 中做同样事情的一些例子吗？

Answer 1

如果你想使用标准的 Go 测试工具，你可以定义一个带有签名的函数TestMain(m *testing.M)并将你的夹具代码放在那里。

从测试包维基：

有时，测试程序需要在测试之前或之后进行额外的设置或拆卸。有时也需要测试来控制哪些代码在主线程上运行。为了支持这些和其他情况，如果测试文件包含一个函数：

func TestMain(m *testing.M)

那么生成的测试将调用 TestMain(m) 而不是直接运行测试。TestMain 运行在主 goroutine 中，并且可以围绕对 m.Run 的调用进行任何必要的设置和拆卸。然后它应该使用 m.Run 的结果调用 os.Exit。调用 TestMain 时，flag.Parse 尚未运行。如果 TestMain 依赖于命令行标志，包括测试包的那些，它应该显式调用 flag.Parse。

TestMain 的一个简单实现是：

func TestMain(m *testing.M) {

    flag.Parse()

    os.Exit(m.Run())

}

Answer 2

我知道这是一个老问题，但这仍然出现在搜索结果中，所以我想我会给出一个可能的答案。

您可以将代码隔离到辅助函数中，这些辅助函数返回一个“拆卸”函数以在其自身之后进行清理。这是启动服务器并在测试用例结束时关闭它的一种可能方法。

func setUpServer() (string, func()) {

    h := func(w http.ResponseWriter, r *http.Request) {

        code := http.StatusTeapot

        http.Error(w, http.StatusText(code), code)

    }

    ts := httptest.NewServer(http.HandlerFunc(h))

    return ts.URL, ts.Close

}

func TestWithServer(t *testing.T) {

    u, close := setUpServer()

    defer close()

    rsp, err := http.Get(u)

    assert.Nil(t, err)

    assert.Equal(t, http.StatusTeapot, rsp.StatusCode)

}

这将启动一个服务器net/http/httptest并返回它的 URL 以及一个充当“拆卸”的函数。这个函数被添加到 defer 堆栈中，因此无论测试用例如何退出，它总是被调用。

或者，*testing.T如果您有更复杂的设置并且需要处理错误，您可以传入。此示例显示设置函数返回一个*url.URL而不是 URL 格式的字符串，并且解析可能会返回一个错误。

func setUpServer(t *testing.T) (*url.URL, func()) {

    h := func(w http.ResponseWriter, r *http.Request) {

        code := http.StatusTeapot

        http.Error(w, http.StatusText(code), code)

    }

    ts := httptest.NewServer(http.HandlerFunc(h))

    u, err := url.Parse(ts.URL)

    assert.Nil(t, err)

    return u, ts.Close

}

func TestWithServer(t *testing.T) {

    u, close := setUpServer(t)

    defer close()

    u.Path = "/a/b/c/d"

    rsp, err := http.Get(u.String())

    assert.Nil(t, err)

    assert.Equal(t, http.StatusTeapot, rsp.StatusCode)

}

Answer 3

我写了golang引擎来使用类似于pytest的fixtures：https : //github.com/rekby/fixenv

用法示例：

package example

// db create database abd db struct, cached per package - call

// once and same db shared with all tests

func db(e Env)*DB{...}

// DbCustomer - create customer with random personal data

// but fixed name. Fixture result shared by test and subtests, 

// then mean many calls Customer with same name will return same

// customer object.

// Call Customer with other name will create new customer

// and resurn other object.

func DbCustomer(e Env, name string) Customer {

    // ... create customer

    db(e).CustomerStore(cust)

    // ...

    return cust

}

// DbAccount create bank account for customer with given name.

func DbAccount(e Env, customerName, accountName string)Account{

    cust := DbCustomer(e, customerName)

    // ... create account

    db(e).AccountStore(acc)

    // ...

    return acc

}

func TestFirstOwnAccounts(t *testing.T){

    e := NewEnv(t)

    // background:

    // create database

    // create customer bob 

    // create account from

    accFrom := DbAccount(e, "bob", "from")

    

    // get existed db, get existed bob, create account to

    accTo := DbAccount(e, "bob", "to")

    

    PutMoney(accFrom, 100)

    SendMoney(accFrom, accTo, 20)

    if accFrom != 80 {

        t.Error()

    }

    if accTo != 20 {

        t.Error()   

    }

    

    // background:

    // delete account to

    // delete account from

    // delete customer bob

}

func TestSecondTransferBetweenCustomers(t *testing.T){

    e := NewEnv(t)

    

    // background:

    // get db, existed from prev test

    // create customer bob

    // create account main for bob

    accFrom := DbAccount(e, "bob", "main")

    

    // background:

    // get existed db

    // create customer alice

    // create account main for alice

    accTo := DbAccount(e, "alice", "main")

    PutMoney(accFrom, 100)

    SendMoney(accFrom, accTo, 20)

    if accFrom != 80 {

        t.Error()

    }

    if accTo != 20 {

        t.Error()

    }

    

    // background:

    // remove account of alice

    // remove customer alice

    // remove account of bob

    // remove customer bob

}

// background:

// after all test finished drop database