Question

我正在尝试使用下面的代码存根os.Stat和ioutil.ReadFile(path)，或者如果您喜欢这里的游乐场[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

Answer 1

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

Answer 2

谢谢，伙计们为你的输入重写我的测试如下..

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)
}

在golang中调用另一个函数

2 个答案: