-
Notifications
You must be signed in to change notification settings - Fork 1
/
snippet_types_test.go
126 lines (105 loc) · 2.91 KB
/
snippet_types_test.go
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package codegen
import (
"bytes"
"reflect"
"testing"
"github.com/stretchr/testify/require"
)
func TestSnippetTypeOf(t *testing.T) {
tt := require.New(t)
tt.Equal("*bytes.Buffer", Stringify(TypeOf(reflect.TypeOf(&bytes.Buffer{}))))
tt.Equal(`chan struct {
Name string `+"`"+`json:"name"`+"`"+`
}`, Stringify(TypeOf(reflect.TypeOf(make(chan struct {
Name string `json:"name"`
})))))
tt.Equal(`chan struct {
bytes.Buffer
Name string `+"`"+`json:"name"`+"`"+`
}`, Stringify(TypeOf(reflect.TypeOf(make(chan struct {
bytes.Buffer
Name string `json:"name"`
})))))
tt.Equal("*bytes.Buffer", Stringify(TypeOf(reflect.TypeOf(&bytes.Buffer{}))))
tt.Equal("[]string", Stringify(TypeOf(reflect.TypeOf([]string{}))))
tt.Equal("[0]string", Stringify(TypeOf(reflect.TypeOf([0]string{}))))
tt.Equal("map[string]string", Stringify(TypeOf(reflect.TypeOf(map[string]string{}))))
}
func TestSnippetType(t *testing.T) {
tt := require.New(t)
tt.Equal("bool", Stringify(Bool))
tt.Equal("Name", Stringify(Type("Name")))
}
func TestSnippetType_ModifiedOrComposedTypes(t *testing.T) {
tt := require.New(t)
tt.Equal("map[bool]bool", Stringify(Map(Bool, Bool)))
tt.Equal("[]bool", Stringify(Slice(Bool)))
tt.Equal("[0]bool", Stringify(Array(Bool, 0)))
tt.Equal("chan *bool", Stringify(Chan(Star(Bool))))
tt.Equal("chan bool", Stringify(Chan(Bool)))
tt.Equal("chan Name", Stringify(Chan(Type("Name"))))
}
func TestSnippetType_InterfaceType(t *testing.T) {
tt := require.New(t)
tt.Equal("interface {}", Stringify(Interface()))
tt.Equal(`interface {
Type()
String() (string)
}`, Stringify(Interface(
Func().Named("Type"),
Func().Return(Var(String)).Named("String"),
)))
tt.Equal(`interface {
Type
String() (string)
}`, Stringify(Interface(
Type("Type"),
Func().Return(Var(String)).Named("String"),
)))
}
func TestSnippetType_FuncType(t *testing.T) {
tt := require.New(t)
tt.Equal("func ()", Stringify(Func()))
tt.Equal("func Fn()", Stringify(Func().Named("Fn")))
tt.Equal("func Fn(a, b string)", Stringify(Func(
Var(String, "a", "b"),
).Named("Fn")))
tt.Equal("func Fn(a string, list ...string)", Stringify(Func(
Var(String, "a"),
Var(Ellipsis(String), "list"),
).Named("Fn")))
tt.Equal("func Fn(a string, b string) (string, error)", Stringify(
Func(
Var(String, "a"),
Var(String, "b"),
).
Return(
Var(String),
Var(Error),
).
Named("Fn"),
))
tt.Equal("func (r R) Fn(a, b interface {}) (error)", Stringify(
Func(
Var(Interface(), "a", "b"),
).
Return(Var(Error)).
MethodOf(Var(Type("R"), "r")).
Named("Fn"),
))
}
func TestSnippetType_StructType(t *testing.T) {
tt := require.New(t)
tt.Equal(`struct {
Embed
Key bool `+"`"+`json:"key,omitempty" validate:"@string"`+"`"+`
KeyA, KeyA1 bool
}`, Stringify(Struct(
Var(Type("Embed")),
Var(Bool, "Key").WithTags(map[string][]string{
"json": {"key", "omitempty"},
"validate": {"@string"},
}),
Var(Bool, "KeyA", "KeyA1"),
)))
}