0

我正在尝试存根os.Stat并ioutil.ReadFile(path)使用下面的代码，或者如果您喜欢在 go playground [1]

package main

import (
    "fmt"
    "io/ioutil"
    "os"
    "strings"
)

func AssignFileValueFrom(path string, val *string) {
    var (
        tempValue []byte
        err       error
    )

    if _, err = os.Stat(path); err == nil {
        if err != nil {
            fmt.Println("There was a os stat error:", err)
        }

        tempValue, err = ioutil.ReadFile(path)
        if err != nil {
            fmt.Println("There was an io read error:", err)
        }
        *val = strings.TrimSpace(string(tempValue))
    }
}

我已经使用了 testify 并尝试按照此处的示例 [2]

package main

import (
    "testing"

    "github.com/stretchr/testify/mock"
)

type osMock struct {
    mock.Mock
}

func (o osMock) Stat(path string) (interface{}, error) {
    return nil, nil
}

func TestAssignFileValueFrom(t *testing.T) {
    var test string
    osm := new(osMock)
    osm.On(`Stat`, `./.test`).Return([]byte(`1`), nil)

    AssignFileValueFrom(`./.test`, &test)
    // assert.Equal(t, `1`, test)
    osm.AssertExpectations(t)
}

我做错了什么？

[1] https://play.golang.org/p/xcbdMkMwoBN

[2] https://github.com/stretchr/testify#mock-package

4

2 回答 2

1

您的代码osMock不会影响AssignFileValueFrom函数的执行。有一个直接调用，os.Stat它不会因为你在osMock某处声明而被替换。
要进行实际测试，您应该使用interfaces 和依赖注入来测试您的代码。
首先，os.Stat调用必须替换为对您的结构的调用，该结构实现了定义了相同方法的接口。并且您需要创建此接口的至少 2 个实现：1 - 是要使用的实际工作代码，2 - 模拟作为您osMock在测试中使用的结构。你需要将它注入或传递给它AssignFileValueFrom，然后使用它来调用Stat它的方法。

于 2018-03-29T07:05:42.430 回答
0

谢谢你们的投入，我已经重写了我的测试如下..

package main

import (
    "io/ioutil"
    "os"
    "testing"

    "github.com/stretchr/testify/assert"
)

var (
    err      error
    testFile *os.File
    test string
)

const (
    TestPrefix  = `test_file_prefix`
    FileContent = `1234`
)

func init() {
    testFile, err = ioutil.TempFile(os.TempDir(), TestPrefix)
    if err != nil {
        panic(err)
    }

    err = ioutil.WriteFile(testFile.Name(), []byte(FileContent), 0644)
    if err != nil {
        panic(err)
    }
}

func TestAssignFileValueFrom(t *testing.T) {
    AssignFileValueFrom(testFile.Name(), &test)

    assert.Equal(t, test, FileContent)
}
于 2018-04-01T12:55:16.973 回答