-
-
Notifications
You must be signed in to change notification settings - Fork 357
Expand file tree
/
Copy pathgenerator_incremental_test.go
More file actions
84 lines (68 loc) · 2.32 KB
/
Copy pathgenerator_incremental_test.go
File metadata and controls
84 lines (68 loc) · 2.32 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package gen
import (
"os"
"path/filepath"
"strings"
"sync"
"testing"
"gorm.io/gen/internal/generate"
)
func TestOutputWithManifest_IncrementalSkipDoesNotOverwrite(t *testing.T) {
tmp := t.TempDir()
g := NewGenerator(Config{OutPath: tmp})
g.Incremental = true
m := &genManifest{Version: 1, Files: map[string]string{}}
var mu sync.Mutex
fileName := filepath.Join(tmp, "x.go")
content := []byte("package p\n\nfunc A() {}\n")
if err := g.outputWithManifest(fileName, content, m, filepath.Base(fileName), &mu); err != nil {
t.Fatalf("first output: %v", err)
}
if err := os.WriteFile(fileName, []byte("package p\n\nfunc B() {}\n"), 0640); err != nil {
t.Fatalf("tamper file: %v", err)
}
if err := g.outputWithManifest(fileName, content, m, filepath.Base(fileName), &mu); err != nil {
t.Fatalf("second output: %v", err)
}
b, err := os.ReadFile(fileName)
if err != nil {
t.Fatalf("read file: %v", err)
}
if !strings.Contains(string(b), "func B") {
t.Fatalf("expected file to keep tampered content, got:\n%s", string(b))
}
}
func TestBuildMergedQueryData_MergeKeepsPreviousTables(t *testing.T) {
tmp := t.TempDir()
g := NewGenerator(Config{OutPath: tmp})
g.MergeQuery = true
g.Data = map[string]*genInfo{
"UserB": {QueryStructMeta: &generate.QueryStructMeta{ModelStructName: "UserB", QueryStructName: "userB", FileName: "user_b"}},
}
if err := os.WriteFile(filepath.Join(tmp, "user_a.gen.go"), []byte("package query\n"), 0640); err != nil {
t.Fatalf("write user_a.gen.go: %v", err)
}
if err := os.WriteFile(filepath.Join(tmp, "user_b.gen.go"), []byte("package query\n"), 0640); err != nil {
t.Fatalf("write user_b.gen.go: %v", err)
}
manifest := &genManifest{
Version: 1,
Tables: map[string]genManifestTable{
"UserA": {ModelStructName: "UserA", QueryStructName: "userA", FileName: "user_a"},
},
Files: map[string]string{},
}
mergedTables, dataForGenGo := g.buildMergedQueryData(manifest)
if _, ok := mergedTables["UserA"]; !ok {
t.Fatalf("expected merged tables to contain UserA")
}
if _, ok := mergedTables["UserB"]; !ok {
t.Fatalf("expected merged tables to contain UserB")
}
if _, ok := dataForGenGo["UserA"]; !ok {
t.Fatalf("expected merged data to contain placeholder UserA")
}
if _, ok := dataForGenGo["UserB"]; !ok {
t.Fatalf("expected merged data to contain current UserB")
}
}