67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package file
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"testing"
|
||
)
|
||
|
||
type TestStruct struct {
|
||
Name string `json:"name" yaml:"name"`
|
||
Age int `json:"age" yaml:"age"`
|
||
}
|
||
|
||
type SnakeStruct struct {
|
||
UserName string `json:"user_name" yaml:"user_name"`
|
||
}
|
||
|
||
func TestUnmarshalFile(t *testing.T) {
|
||
tmpDir, _ := os.MkdirTemp("", "test_object")
|
||
defer os.RemoveAll(tmpDir)
|
||
|
||
jsonFile := filepath.Join(tmpDir, "test.json")
|
||
// 测试智能映射：输入 user_name，目标 UserName
|
||
jsonData := `{"user_name": "Alice"}`
|
||
os.WriteFile(jsonFile, []byte(jsonData), 0644)
|
||
|
||
t.Run("SmartMapping", func(t *testing.T) {
|
||
var ts SnakeStruct
|
||
err := UnmarshalFile(jsonFile, &ts)
|
||
if err != nil {
|
||
t.Errorf("UnmarshalFile failed: %v", err)
|
||
}
|
||
if ts.UserName != "Alice" {
|
||
t.Errorf("Expected Alice, got %s", ts.UserName)
|
||
}
|
||
})
|
||
}
|
||
|
||
func TestMarshalFile(t *testing.T) {
|
||
tmpDir, _ := os.MkdirTemp("", "test_marshal")
|
||
defer os.RemoveAll(tmpDir)
|
||
|
||
data := TestStruct{Name: "Bob", Age: 25}
|
||
|
||
t.Run("Pretty", func(t *testing.T) {
|
||
filename := filepath.Join(tmpDir, "pretty.json")
|
||
err := MarshalFilePretty(filename, data)
|
||
if err != nil {
|
||
t.Fatalf("MarshalFilePretty failed: %v", err)
|
||
}
|
||
})
|
||
}
|
||
|
||
func TestPatchFile(t *testing.T) {
|
||
tmpDir, _ := os.MkdirTemp("", "test_patch")
|
||
defer os.RemoveAll(tmpDir)
|
||
filename := filepath.Join(tmpDir, "config.json")
|
||
|
||
t.Run("Update", func(t *testing.T) {
|
||
patch := map[string]any{"name": "Initial"}
|
||
err := PatchFile(filename, patch)
|
||
if err != nil {
|
||
t.Fatalf("PatchFile failed: %v", err)
|
||
}
|
||
})
|
||
}
|