/
options_test.go
123 lines (104 loc) · 3.51 KB
/
options_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
package httpreq_test
import (
"testing"
"github.com/gookit/goutil/dump"
"github.com/gookit/goutil/fsutil"
"github.com/gookit/goutil/jsonutil"
"github.com/gookit/goutil/netutil/httpctype"
"github.com/gookit/goutil/netutil/httpreq"
"github.com/gookit/goutil/testutil"
"github.com/gookit/goutil/testutil/assert"
)
func TestOptions_build(t *testing.T) {
opt := httpreq.OptOrNew(nil)
opt.BytesBody([]byte("name=inhere"))
assert.Equal(t, []byte("name=inhere"), fsutil.ReadAll(opt.Body))
opt.StringBody("name=inhere")
assert.Equal(t, []byte("name=inhere"), fsutil.ReadAll(opt.Body))
// FormBody
opt.FormBody(map[string]string{"name": "inhere"})
assert.Equal(t, []byte("name=inhere"), fsutil.ReadAll(opt.Body))
// WithJSON
opt.WithJSON(map[string]string{"name": "inhere"})
assert.Equal(t, httpctype.JSON, opt.ContentType)
assert.Equal(t, `{"name":"inhere"}
`, fsutil.ReadString(opt.Body))
}
func TestOption_Send(t *testing.T) {
opt := httpreq.OptOrNew(nil)
assert.NotNil(t, opt)
opt = httpreq.NewOpt(httpreq.WithJSONType)
assert.NotNil(t, opt)
assert.Eq(t, httpctype.JSON, opt.ContentType)
resp, err := httpreq.New(testSrvAddr).
StringBody(`{"name": "inhere"}`).
WithContentType(httpctype.JSON).
WithHeaderMap(map[string]string{"coustom1": "value1"}).
Send("POST", "/json")
assert.NoErr(t, err)
sc := resp.StatusCode
assert.True(t, httpreq.IsOK(sc))
assert.True(t, httpreq.IsSuccessful(sc))
assert.False(t, httpreq.IsRedirect(sc))
assert.False(t, httpreq.IsForbidden(sc))
assert.False(t, httpreq.IsNotFound(sc))
assert.False(t, httpreq.IsClientError(sc))
assert.False(t, httpreq.IsServerError(sc))
retMp := make(map[string]any)
err = jsonutil.DecodeReader(resp.Body, &retMp)
assert.NoErr(t, err)
dump.P(retMp)
}
func TestOptions_REST(t *testing.T) {
opt := httpreq.New(testSrvAddr).
WithOption().
WithContentType(httpctype.Form).
WithHeader("custom1", "value1").
WithHeaderMap(map[string]string{
"custom2": "value2",
})
t.Run("Get", func(t *testing.T) {
resp, err := opt.Copy().Get("/get", httpreq.WithData("name=inhere&age=18"))
assert.NoErr(t, err)
sc := resp.StatusCode
assert.True(t, httpreq.IsOK(sc))
assert.True(t, httpreq.IsSuccessful(sc))
rr := testutil.ParseRespToReply(resp)
assert.Equal(t, "GET", rr.Method)
assert.Equal(t, "value1", rr.Headers["Custom1"])
assert.Equal(t, "value2", rr.Headers["Custom2"])
})
t.Run("Post", func(t *testing.T) {
resp, err := opt.Copy().Post("/post", nil, httpreq.WithData("name=inhere&age=18"))
assert.NoErr(t, err)
sc := resp.StatusCode
assert.True(t, httpreq.IsOK(sc))
assert.True(t, httpreq.IsSuccessful(sc))
rr := testutil.ParseRespToReply(resp)
assert.Equal(t, "POST", rr.Method)
assert.Equal(t, "value1", rr.Headers["Custom1"])
assert.Equal(t, "value2", rr.Headers["Custom2"])
dump.P(rr)
})
t.Run("Put", func(t *testing.T) {
resp, err := opt.Copy().WithData("name=inhere&age=18").Put("/put", nil)
assert.NoErr(t, err)
sc := resp.StatusCode
assert.True(t, httpreq.IsOK(sc))
assert.True(t, httpreq.IsSuccessful(sc))
rr := testutil.ParseRespToReply(resp)
// dump.P(rr)
assert.Equal(t, "PUT", rr.Method)
assert.Equal(t, "value1", rr.Headers["Custom1"])
assert.Equal(t, "value2", rr.Headers["Custom2"])
assert.NotEmpty(t, rr.Form)
assert.NotEmpty(t, rr.Body)
})
t.Run("Delete", func(t *testing.T) {
resp, err := opt.Copy().Delete("/delete", httpreq.WithData("name=inhere&age=18"))
assert.NoErr(t, err)
sc := resp.StatusCode
assert.True(t, httpreq.IsOK(sc))
assert.True(t, httpreq.IsSuccessful(sc))
})
}