Golang单元测试文件IO操作示例-Golang-PHP中文网

Golang单元测试文件IO操作示例

P粉602998670

发布： 2025-10-24 20:25:02

原创

503人浏览过

通过接口抽象文件操作并使用mock实现，可有效解耦IO依赖，提升Go单元测试的可靠性与速度。

golang单元测试文件io操作示例

在Go语言中进行单元测试时，如果遇到文件IO操作，直接读写真实文件会带来依赖问题，影响测试的可重复性和速度。最佳做法是通过接口抽象文件操作，并在测试中使用模拟（mock）或内存中的数据替代真实IO。下面是一个实用示例，展示如何对涉及文件读写的函数进行单元测试。

定义文件操作接口

为了便于测试，先将文件操作抽象成一个接口：

type FileReader interface {
    ReadFile(filename string) ([]byte, error)
}

// 实现真实文件读取
type RealFileReader struct{}

func (r RealFileReader) ReadFile(filename string) ([]byte, error) {
    return os.ReadFile(filename)
}

登录后复制

假设我们有一个函数，它依赖读取JSON配置文件并返回结构体：

type Config struct {
    Host string `json:"host"`
    Port int    `json:"port"`
}

func LoadConfig(reader FileReader, filename string) (*Config, error) {
    data, err := reader.ReadFile(filename)
    if err != nil {
        return nil, err
    }
    var config Config
    if err := json.Unmarshal(data, &config); err != nil {
        return nil, err
    }
    return &config, nil
}
</font>

<H3>编写模拟实现用于测试</H3>
<p>在测试中，我们不希望真正读取磁盘文件，可以创建一个模拟的 <strong>FileReader</strong>：</p><p><span>立即学习</span>“<a href="https://pan.quark.cn/s/00968c3c2c15" style="text-decoration: underline !important; color: blue; font-weight: bolder;" rel="nofollow" target="_blank">go语言免费学习笔记（深入）</a>”；</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/%E9%9D%92%E6%9F%9A%E9%9D%A2%E8%AF%95">
                            <img src="https://img.php.cn/upload/ai_manual/001/246/273/68b6cab553c77389.png" alt="青柚面试">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/%E9%9D%92%E6%9F%9A%E9%9D%A2%E8%AF%95">青柚面试</a>
                            <p>简单好用的日语面试辅助工具</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="青柚面试">
                                <span>57</span>
                            </div>
                        </div>
                        <a href="/ai/%E9%9D%92%E6%9F%9A%E9%9D%A2%E8%AF%95" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="青柚面试">
                        </a>
                    </div>
                
<font face="Courier New">
<pre class="brush:php;toolbar:false;">
type MockFileReader struct {
    Data []byte
    Err  error
}

func (m MockFileReader) ReadFile(filename string) ([]byte, error) {
    return m.Data, m.Err
}

登录后复制

编写单元测试

使用 mock 来测试 LoadConfig 函数的各种情况：

func TestLoadConfig_Success(t *testing.T) {
    jsonData := `{"host": "localhost", "port": 8080}`
    mockReader := MockFileReader{Data: []byte(jsonData)}

    config, err := LoadConfig(mockReader, "config.json") // 文件名仅作占位

    if err != nil {
        t.Fatalf("Expected no error, got %v", err)
    }
    if config.Host != "localhost" || config.Port != 8080 {
        t.Errorf("Expected localhost:8080, got %s:%d", config.Host, config.Port)
    }
}

func TestLoadConfig_FileNotFound(t *testing.T) {
    mockReader := MockFileReader{Err: os.ErrNotExist}

    _, err := LoadConfig(mockReader, "missing.json")

    if err == nil {
        t.Fatal("Expected error, got nil")
    }
    if !errors.Is(err, os.ErrNotExist) {
        t.Errorf("Expected os.ErrNotExist, got %v", err)
    }
}

func TestLoadConfig_InvalidJSON(t *testing.T) {
    mockReader := MockFileReader{Data: []byte("{invalid json}")}

    _, err := LoadConfig(mockReader, "bad.json")

    if err == nil {
        t.Fatal("Expected unmarshal error")
    }
}

登录后复制

这样就完全解耦了文件IO和业务逻辑，测试快速、可靠，无需准备真实文件或清理临时目录。