78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package web
|
||
|
||
import (
|
||
"net/http"
|
||
"net/http/httptest"
|
||
"net/url"
|
||
"strings"
|
||
"testing"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
func listPostContext(t *testing.T, path string, values url.Values) (*gin.Context, *httptest.ResponseRecorder) {
|
||
t.Helper()
|
||
gin.SetMode(gin.TestMode)
|
||
recorder := httptest.NewRecorder()
|
||
context, _ := gin.CreateTestContext(recorder)
|
||
request := httptest.NewRequest(http.MethodPost, path, strings.NewReader(values.Encode()))
|
||
request.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||
context.Request = request
|
||
return context, recorder
|
||
}
|
||
|
||
func assertRedirectQuery(t *testing.T, recorder *httptest.ResponseRecorder, wantPath string, want map[string]string) {
|
||
t.Helper()
|
||
location, err := url.Parse(recorder.Header().Get("Location"))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if location.Path != wantPath {
|
||
t.Fatalf("跳转路径=%s，期望 %s", location.Path, wantPath)
|
||
}
|
||
for key, value := range want {
|
||
if got := location.Query().Get(key); got != value {
|
||
t.Fatalf("%s=%q，期望 %q；Location=%s", key, got, value, location.String())
|
||
}
|
||
}
|
||
}
|
||
|
||
func Test列表写操作跳转保留筛选和页码(t *testing.T) {
|
||
t.Run("PDD", func(t *testing.T) {
|
||
context, recorder := listPostContext(t, "/pdd/delete", url.Values{
|
||
"q": {"737"}, "status": {"pending"}, "page": {"3"},
|
||
})
|
||
(&Handler{}).pddRedirect(context, "完成")
|
||
assertRedirectQuery(t, recorder, "/pdd", map[string]string{
|
||
"q": "737", "status": "pending", "page": "3", "msg": "完成",
|
||
})
|
||
})
|
||
t.Run("任务", func(t *testing.T) {
|
||
context, recorder := listPostContext(t, "/tasks/delete", url.Values{
|
||
"q": {"TASK"}, "type": {"collect"}, "status": {"pending"}, "page": {"2"},
|
||
})
|
||
(&Handler{}).taskRedirect(context, "完成")
|
||
assertRedirectQuery(t, recorder, "/tasks", map[string]string{
|
||
"q": "TASK", "type": "collect", "status": "pending", "page": "2", "msg": "完成",
|
||
})
|
||
})
|
||
t.Run("客户端", func(t *testing.T) {
|
||
context, recorder := listPostContext(t, "/clients/assign", url.Values{
|
||
"name": {"仓库"}, "page": {"4"},
|
||
})
|
||
redirectClients(context, "完成", "")
|
||
assertRedirectQuery(t, recorder, "/clients", map[string]string{
|
||
"name": "仓库", "page": "4", "msg": "完成",
|
||
})
|
||
})
|
||
t.Run("用户", func(t *testing.T) {
|
||
context, recorder := listPostContext(t, "/users/status", url.Values{
|
||
"q": {"buyer"}, "list_status": {"active"}, "page": {"5"},
|
||
})
|
||
redirectUsers(context, "完成", "")
|
||
assertRedirectQuery(t, recorder, "/users", map[string]string{
|
||
"q": "buyer", "status": "active", "page": "5", "msg": "完成",
|
||
})
|
||
})
|
||
}
|