Golang如何测试JSON文件操作_Golang JSON文件读写测试实践

type Person struct {
    Name  string `json:"name"`
    Age   int    `json:"age"`
    Email string `json:"email,omitempty"`
}

{
    "name": "Alice",
    "age": 25,
    "email": "alice@example.com"
}

func ReadJSON(filename string, v interface{}) error {
    data, err := os.ReadFile(filename)
    if err != nil {
        return err
    }
    return json.Unmarshal(data, v)
}

func WriteJSON(filename string, v interface{}) error {
    data, err := json.MarshalIndent(v, "", "  ")
    if err != nil {
        return err
    }
    return os.WriteFile(filename, data, 0644)
}

func TestReadJSON(t *testing.T) {
    var p Person
    err := ReadJSON("testdata/person.json", &p)
    if err != nil {
        t.Fatalf("读取JSON失败: %v", err)
    }

    if p.Name != "Alice" {
        t.Errorf("期望名字为Alice，实际为 %s", p.Name)
    }
    if p.Age != 25 {
        t.Errorf("期望年龄为25，实际为 %d", p.Age)
    }
    if p.Email != "alice@example.com" {
        t.Errorf("期望邮箱为alice@example.com，实际为 %s", p.Email)
    }
}

func TestWriteJSON(t *testing.T) {
    tempFile := "testdata/output.json"
    person := Person{
        Name:  "Bob",
        Age:   30,
        Email: "bob@example.com",
    }

    err := WriteJSON(tempFile, person)
    if err != nil {
        t.Fatalf("写入JSON失败: %v", err)
    }
    defer os.Remove(tempFile) // 清理临时文件

    // 验证写入后能否正确读回
    var p Person
    err = ReadJSON(tempFile, &p)
    if err != nil {
        t.Fatalf("读取写入的文件失败: %v", err)
    }

    if p.Name != "Bob" || p.Age != 30 || p.Email != "bob@example.com" {
        t.Errorf("读回的数据不匹配: %+v", p)
    }
}

tempFile, err := os.CreateTemp("testdata", "temp_*.json")
if err != nil {
    t.Fatal(err)
}
defer os.Remove(tempFile.Name())