qheader_test.go

// SPDX-FileCopyrightText: 2019-2024 caixw
//
// SPDX-License-Identifier: MIT

package qheader

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/issue9/assert/v4"
)

func TestAccept(t *testing.T) {
	a := assert.New(t, false)

	r := httptest.NewRequest(http.MethodGet, "/path", nil)
	r.Header.Add("Accept", "text/json;q=0.5,text/xml;q=0.8,application/xml;q=0.8,zero;q=0")
	qh := Accept(r)
	a.NotNil(qh).Length(qh.Items, 4)
	a.Equal(qh.Items[0].Value, "text/xml")
	a.Equal(qh.Items[1].Value, "application/xml")
	a.Equal(qh.Items[2].Value, "text/json")
	a.Equal(qh.Items[3].Value, "zero").Equal(qh.Items[3].Q, 0.0)

	qh.Destroy()
}

func TestAcceptLanguage(t *testing.T) {
	a := assert.New(t, false)

	r := httptest.NewRequest(http.MethodGet, "/path", nil)
	r.Header.Add("Accept-Language", "zh-tw;q=0.5,zh-cn;q=0.8,en;q=0.8")
	qh := AcceptLanguage(r)
	a.Equal(3, len(qh.Items))
	a.Equal(qh.Items[0].Value, "zh-cn")
	a.Equal(qh.Items[1].Value, "en")
	a.Equal(qh.Items[2].Value, "zh-tw")
}

func TestAcceptEncoding(t *testing.T) {
	a := assert.New(t, false)

	r := httptest.NewRequest(http.MethodGet, "/path", nil)
	r.Header.Add("Accept-Encoding", "gzip;q=0.5,compress;q=0.8,*;q=0.6,br")
	qh := AcceptEncoding(r)
	a.Equal(4, len(qh.Items))
	a.Equal(qh.Items[0].Value, "br")
	a.Equal(qh.Items[1].Value, "compress")
	a.Equal(qh.Items[2].Value, "*")
	a.Equal(qh.Items[3].Value, "gzip")
}

func TestAcceptCharset(t *testing.T) {
	a := assert.New(t, false)

	r := httptest.NewRequest(http.MethodGet, "/path", nil)
	r.Header.Add("Accept-Charset", "utf8;q=0.5,abc;q=0.5,defg;q=0.5,*;q=0.5,cp936,utf32;q=0.4")
	qh := AcceptCharset(r)
	a.Equal(len(qh.Items), 6)
	a.Equal(qh.Items[0].Value, "cp936")
	a.Equal(qh.Items[1].Value, "utf8")
	a.Equal(qh.Items[2].Value, "abc")
	a.Equal(qh.Items[3].Value, "defg")
	a.Equal(qh.Items[4].Value, "*")
	a.Equal(qh.Items[5].Value, "utf32")
}

func TestParse(t *testing.T) {
	a := assert.New(t, false)

	a.Panic(func() {
		Parse(",a1", "not-allow")
	})

	qh := Parse("a0,a1,a2,a3;q=0.5,a4,a5;q=0.9,a6;a61;q=0.8", "*/*")
	a.NotEmpty(qh)
	a.Equal(len(qh.Items), 7)
	// 确定排序是否正常
	a.Equal(qh.Items[0].Q, float32(1.0))
	a.Equal(qh.Value, qh.Items[0].Value)
	a.Equal(qh.Items[5].Q, float32(.5))

	qh = Parse(",a1,a2,a3;q=0.5,a4,a5;q=0.9,a6;a61;q=0.0", "*/*")
	a.NotEmpty(qh)
	a.Equal(len(qh.Items), 6)
	a.Equal(qh.Items[0].Q, float32(1.0))

	// xx/* 的权限低于相同 Q 值的其它权限
	qh = Parse("x/*;q=0.1,b/*;q=0.1,a/*;q=0.1,t/*;q=0.1,text/plain;q=0.1", "*/*")
	a.NotEmpty(qh)
	a.Equal(len(qh.Items), 5)
	a.Equal(qh.Items[0].Value, "text/plain").Equal(qh.Items[0].Q, float32(0.1))
	a.Equal(qh.Items[1].Value, "x/*").Equal(qh.Items[1].Q, float32(0.1))
	a.Equal(qh.Items[2].Value, "b/*").Equal(qh.Items[2].Q, float32(0.1))
	a.Equal(qh.Items[3].Value, "a/*").Equal(qh.Items[3].Q, float32(0.1))
	a.Equal(qh.Items[4].Value, "t/*").Equal(qh.Items[4].Q, float32(0.1))

	// xx/* 的权限低于相同 Q 值的其它权限
	qh = Parse("text/*;q=0.1,xx/*;q=0.1,text/html;q=0.1", "*/*")
	a.NotEmpty(qh)
	a.Equal(len(qh.Items), 3)
	a.Equal(qh.Items[0].Value, "text/html").Equal(qh.Items[0].Q, float32(0.1))
	a.Equal(qh.Items[1].Value, "text/*").Equal(qh.Items[1].Q, float32(0.1))

	// */* 的权限最底
	qh = Parse("text/html;q=0.1,text/*;q=0.1,xx/*;q=0.1,*/*;q=0.1", "*/*")
	a.NotEmpty(qh)
	a.Equal(len(qh.Items), 4)
	a.Equal(qh.Items[0].Value, "text/html").Equal(qh.Items[0].Q, float32(0.1))
	a.Equal(qh.Items[1].Value, "text/*").Equal(qh.Items[1].Q, float32(0.1))

	qh = Parse("utf-8;q=x.9,gbk;q=0.8", "*/*")
	a.Equal(2, len(qh.Items)).
		Equal(qh.Items[0].Value, "gbk").NotError(qh.Items[0].Err).
		Equal(qh.Items[1].Value, "utf-8").Error(qh.Items[1].Err)
}