-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodule_test.go
95 lines (76 loc) · 2.02 KB
/
module_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
package cacher_test
import (
"encoding/json"
"io"
"net/http/httptest"
"testing"
"time"
"github.com/stretchr/testify/require"
"github.com/tinh-tinh/cacher"
"github.com/tinh-tinh/tinhtinh/common"
"github.com/tinh-tinh/tinhtinh/core"
)
func Test_Module(t *testing.T) {
userController := func(module *core.DynamicModule) *core.DynamicController {
cache := cacher.Inject[[]byte](module)
ctrl := module.NewController("users")
ctrl.Get("", func(ctx core.Ctx) error {
data, err := cache.Get("users")
if err != nil {
return common.InternalServerException(ctx.Res(), err.Error())
}
return ctx.JSON(core.Map{
"data": string(data),
})
})
ctrl.Post("", func(ctx core.Ctx) error {
cache.Set("users", []byte("John"))
return ctx.JSON(core.Map{
"data": "ok",
})
})
return ctrl
}
userModule := func(module *core.DynamicModule) *core.DynamicModule {
return module.New(core.NewModuleOptions{
Controllers: []core.Controller{
userController,
},
})
}
appModule := func() *core.DynamicModule {
module := core.NewModule(core.NewModuleOptions{
Imports: []core.Module{
cacher.Register(cacher.Options[[]byte]{
Ttl: 15 * time.Minute,
}),
userModule,
},
})
return module
}
app := core.CreateFactory(appModule)
app.SetGlobalPrefix("api")
testServer := httptest.NewServer(app.PrepareBeforeListen())
defer testServer.Close()
testClient := testServer.Client()
_, err := testClient.Post(testServer.URL+"/api/users", "application/json", nil)
require.Nil(t, err)
resp, err := testClient.Get(testServer.URL + "/api/users")
require.Nil(t, err)
require.Equal(t, 200, resp.StatusCode)
data, err := io.ReadAll(resp.Body)
require.Nil(t, err)
type Response struct {
Data string `json:"data"`
}
var response Response
err = json.Unmarshal(data, &response)
require.Nil(t, err)
require.Equal(t, "John", response.Data)
}
func Test_Nil(t *testing.T) {
module := core.NewModule(core.NewModuleOptions{})
cache := cacher.Inject[[]byte](module)
require.Nil(t, cache)
}