69 lines
2.2 KiB
Go
69 lines
2.2 KiB
Go
|
package web
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"net/http"
|
|||
|
|
"net/http/httptest"
|
|||
|
|
"net/url"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func sybPostContext(t *testing.T, 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, "/syb/sync", strings.NewReader(values.Encode()))
|
|||
|
|
request.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|||
|
|
context.Request = request
|
|||
|
|
return context, recorder
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestSybRedirectLogin_保留指定日期并打开登录弹窗(t *testing.T) {
|
|||
|
|
context, recorder := sybPostContext(t, url.Values{
|
|||
|
|
"order_no": {"ORDER-1"},
|
|||
|
|
"date_from": {"2026-08-01"},
|
|||
|
|
"date_to": {"2026-08-09"},
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
(&Handler{}).sybRedirectLogin(context, "")
|
|||
|
|
if context.Writer.Status() != http.StatusSeeOther {
|
|||
|
|
t.Fatalf("状态码 = %d，期望 %d", context.Writer.Status(), http.StatusSeeOther)
|
|||
|
|
}
|
|||
|
|
location, err := url.Parse(recorder.Header().Get("Location"))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("解析 Location 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
query := location.Query()
|
|||
|
|
for name, want := range map[string]string{
|
|||
|
|
"login": "1", "order_no": "ORDER-1",
|
|||
|
|
"date_from": "2026-08-01", "date_to": "2026-08-09",
|
|||
|
|
} {
|
|||
|
|
if got := query.Get(name); got != want {
|
|||
|
|
t.Errorf("%s = %q，期望 %q；Location=%s", name, got, want, location.String())
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestSybRedirectRange_保留输入并重新打开日期弹窗(t *testing.T) {
|
|||
|
|
context, recorder := sybPostContext(t, url.Values{
|
|||
|
|
"date_from": {"2026-08-09"},
|
|||
|
|
"date_to": {"2026-08-01"},
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
(&Handler{}).sybRedirectRange(context, "开始日期不能晚于结束日期")
|
|||
|
|
location, err := url.Parse(recorder.Header().Get("Location"))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("解析 Location 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
query := location.Query()
|
|||
|
|
if query.Get("range_sync") != "1" || query.Get("range_error") == "" {
|
|||
|
|
t.Fatalf("日期错误后应该重开弹窗并显示错误，Location=%s", location.String())
|
|||
|
|
}
|
|||
|
|
if query.Get("date_from") != "2026-08-09" || query.Get("date_to") != "2026-08-01" {
|
|||
|
|
t.Fatalf("日期错误后应该保留两端输入，Location=%s", location.String())
|
|||
|
|
}
|
|||
|
|
}
|