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

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

Mary-Kate Olsen

发布： 2024-12-16 20:30:12

原创

227 人浏览过

How to Effectively Mock Filesystem Interactions for Go Unit Testing?

Go 中的模拟文件系统测试

简介

与文件系统交互的单元测试函数通常需要模拟文件系统以提供受控输入并验证预期输出。

实现细节

模拟接口和文件类型

要模拟文件系统，请定义一个接口 fileSystem，它声明文件操作（如打开和访问文件信息）的方法，并创建一个具体类型 osFS，它实现该接口以进行实际的文件系统交互。

测试函数

举个例子，考虑 getSize(name string)，它返回给定名称的文件的大小fs.Stat(name) 获取文件信息。

模拟文件系统设置

在测试代码中，将全局 fs 变量替换为继承自 fileSystem 并提供受控行为的模拟mockedFS。您可以指定是否报告错误以及Stat返回的大小。

测试代码

使用mockedFS测试不同的场景，例如错误报告和大小检索。

示例测试代码

type mockedFS struct {
    osFS
    reportErr  bool
    reportSize int64
}

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

func TestGetSize(t *testing.T) {
    oldFs := fs
    mfs := &mockedFS{}
    fs = mfs
    defer func() { fs = oldFs }()

    mfs.reportErr = true
    if _, err := getSize("hello.go"); err == nil {
        t.Error("Expected error, but err is nil!")
    }

    mfs.reportErr = false
    mfs.reportSize = 123
    if size, err := getSize("hello.go"); err != nil {
        t.Errorf("Expected no error, got: %v", err)
    } else if size != 123 {
        t.Errorf("Expected size %d, got: %d", 123, size)
    }
}

登录后复制