Add more unit-tests

Author: gaby

ctx.go

@@ -400,6 +400,7 @@ func (c *DefaultCtx) Cookie(cookie *Cookie) {
 	if cookie.Path == "" {
 		cookie.Path = "/"
 	}
+
 	if cookie.SessionOnly {
 		cookie.MaxAge = 0
 		cookie.Expires = time.Time{}
@@ -431,7 +432,9 @@ func (c *DefaultCtx) Cookie(cookie *Cookie) {
 		SameSite:    sameSite,
 		Partitioned: cookie.Partitioned,
 	}
+
 	if err := hc.Valid(); err != nil {
+		// invalid cookies are ignored, same approach as net/http
 		return
 	}
 
@@ -441,12 +444,15 @@ func (c *DefaultCtx) Cookie(cookie *Cookie) {
 	fcookie.SetValue(hc.Value)
 	fcookie.SetPath(hc.Path)
 	fcookie.SetDomain(hc.Domain)
+
 	if !cookie.SessionOnly {
 		fcookie.SetMaxAge(hc.MaxAge)
 		fcookie.SetExpire(hc.Expires)
 	}
+
 	fcookie.SetSecure(hc.Secure)
 	fcookie.SetHTTPOnly(hc.HttpOnly)
+
 	switch sameSite {
 	case http.SameSiteLaxMode:
 		fcookie.SetSameSite(fasthttp.CookieSameSiteLaxMode)
@@ -459,7 +465,10 @@ func (c *DefaultCtx) Cookie(cookie *Cookie) {
 	default:
 		fcookie.SetSameSite(fasthttp.CookieSameSiteDisabled)
 	}
+
 	fcookie.SetPartitioned(hc.Partitioned)
+
+	// Set resp header
 	c.fasthttp.Response.Header.SetCookie(fcookie)
 	fasthttp.ReleaseCookie(fcookie)
 }

ctx_test.go

@@ -941,7 +941,6 @@ func Test_Ctx_Cookie_PartitionedSecure(t *testing.T) {
 func Test_Ctx_Cookie_Invalid(t *testing.T) {
 	t.Parallel()
 	app := New()
-	c := app.AcquireCtx(&fasthttp.RequestCtx{})
 
 	cases := []*Cookie{
 		{Name: "", Value: "a"},                                                        // empty name
@@ -956,10 +955,71 @@ func Test_Ctx_Cookie_Invalid(t *testing.T) {
 	}
 
 	for _, invalid := range cases {
+		c := app.AcquireCtx(&fasthttp.RequestCtx{})
 		c.Res().Cookie(invalid)
 		require.Empty(t, c.Res().Get(HeaderSetCookie))
 		c.Response().Header.Reset()
+		app.ReleaseCtx(c)
+	}
+}
+
+// go test -run Test_Ctx_Cookie_DefaultPath
+func Test_Ctx_Cookie_DefaultPath(t *testing.T) {
+	t.Parallel()
+	app := New()
+	c := app.AcquireCtx(&fasthttp.RequestCtx{})
+
+	ck := &Cookie{
+		Name:  "p",
+		Value: "v",
+		// Path intentionally empty to verify defaulting
+	}
+
+	c.Res().Cookie(ck)
+	require.Equal(t,
+		"p=v; path=/; SameSite=Lax",
+		c.Res().Get(HeaderSetCookie),
+	)
+}
+
+// go test -run Test_Ctx_Cookie_MaxAgeOnly
+func Test_Ctx_Cookie_MaxAgeOnly(t *testing.T) {
+	t.Parallel()
+	app := New()
+	c := app.AcquireCtx(&fasthttp.RequestCtx{})
+
+	ck := &Cookie{
+		Name:   "ttl",
+		Value:  "v",
+		MaxAge: 3600,
 	}
+	c.Res().Cookie(ck)
+
+	require.Equal(t,
+		"ttl=v; Max-Age=3600; path=/; SameSite=Lax",
+		c.Res().Get(HeaderSetCookie),
+	)
+}
+
+// go test -run Test_Ctx_Cookie_StrictPartitioned
+func Test_Ctx_Cookie_StrictPartitioned(t *testing.T) {
+	t.Parallel()
+	app := New()
+	c := app.AcquireCtx(&fasthttp.RequestCtx{})
+
+	ck := &Cookie{
+		Name:        "sp",
+		Value:       "v",
+		Secure:      true,
+		SameSite:    CookieSameSiteStrictMode,
+		Partitioned: true,
+	}
+	c.Res().Cookie(ck)
+
+	require.Equal(t,
+		"sp=v; path=/; secure; SameSite=Strict; Partitioned",
+		c.Res().Get(HeaderSetCookie),
+	)
 }
 
 // go test -v -run=^$ -bench=Benchmark_Ctx_Cookie -benchmem -count=4