如何在 Go 中有效模拟文件系统交互以进行单元测试？

使用 Mock 测试 Go 中的文件系统交互

在对与文件系统交互的函数进行单元测试时，模拟文件系统非常有价值。为了演示其用法，让我们检查基于接口的方法，在该方法中我们创建一个封装文件系统操作的自定义文件系统接口。

var fs fileSystem = osFS{}

type fileSystem interface {
    Open(name string) (file, error)
    Stat(name string) (os.FileInfo, error)
}

type file interface {
    io.Closer
    io.Reader
    io.ReaderAt
    io.Seeker
    Stat() (os.FileInfo, error)
}

// osFS implements fileSystem using the local disk.
type osFS struct{}

func (osFS) Open(name string) (file, error)        { return os.Open(name) }
func (osFS) Stat(name string) (os.FileInfo, error) { return os.Stat(name) }

模拟实现：

为了模拟文件系统，我们可以创建文件系统接口的模拟版本，称为mockedFS。这使我们能够在测试期间控制文件系统操作的行为。

示例函数：

让我们考虑一个使用 filesystem.Stat 返回文件大小的函数 getSize方法：

func getSize(name string) (int64, error) {
    stat, err := fs.Stat(name)
    if err != nil {
        return 0, err
    }
    return stat.Size(), nil
}

设置模拟：

在测试 getSize 之前，我们需要将全局 fs 变量替换为我们的模拟实例：

func TestGetSize(t *testing.T) {
    // Save the original filesystem
    oldFs := fs

    // Replace with mocked filesystem
    mfs := &mockedFS{}
    fs = mfs

    // Restore original filesystem after test
    defer func() { fs = oldFs }()
}

测试执行：

下面的测试代码验证了 getSize 函数在各种错误和文件大小下的行为条件：

type mockedFS struct {
    reportErr  bool
    reportSize int64
}

func (mfs) Stat(name string) (os.FileInfo, error) {
    if mfs.reportErr {
        return nil, os.ErrNotExist
    }
    return &mockedFileInfo{size: mfs.reportSize}, nil
}

type mockedFileInfo struct {
    size int64
}

func (mfi mockedFileInfo) Size() int64 { return mfi.size }

// Test Error Case
func TestGetSize(t *testing.T) {
    ...
    mfs.reportErr = true
    if _, err := getSize("hello.go"); err == nil { ... }

// Test Non-Error Case
func TestGetSize(t *testing.T) {
    ...
    mfs.reportSize = 123
    if size, err := getSize("hello.go"); ... }

此示例演示如何创建自定义文件系统接口，模拟其实现，并使用它来测试与文件系统交互的函数，确保 Go 中的受控测试环境。

以上是如何在 Go 中有效模拟文件系统交互以进行单元测试？的详细内容。更多信息请关注PHP中文网其他相关文章！