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()) } }