parser_accept_test.go

package mimeheader_test

import (
	"fmt"
	"reflect"
	"testing"

	"github.com/aohorodnyk/mimeheader"
)

func ExampleParseAcceptHeader() {
	header := "image/*; q=0.9; s=4, application/json; q=0.9; b=3;, text/plain,image/png;q=0.9, image/jpeg,image/svg;q=0.8"
	ah := mimeheader.ParseAcceptHeader(header)

	fmt.Println(ah.Negotiate([]string{"application/xml", "image/tiff"}, "text/javascript"))
	fmt.Println(ah.Negotiate([]string{"application/xml", "image/png"}, "text/javascript"))
	fmt.Println(ah.Negotiate([]string{"application/xml", "image/svg"}, "text/javascript"))
	fmt.Println(ah.Negotiate([]string{"text/dart", "application/dart"}, "text/javascript"))

	fmt.Println(ah.Match("image/svg"))
	fmt.Println(ah.Match("text/javascript"))
	// Output:
	// image/* image/tiff true
	// image/png image/png true
	// image/* image/svg true
	//  text/javascript false
	// true
	// false
}

func TestParseAcceptHeader(t *testing.T) {
	t.Parallel()

	for _, prov := range providerParseAcceptHeader() {
		prov := prov
		t.Run(prov.name, func(t *testing.T) {
			t.Parallel()

			act := mimeheader.ParseAcceptHeader(prov.header)
			if !reflect.DeepEqual(prov.exp, act) {
				t.Fatalf("AcceptHeaders are not equal.\nExpected: %+v\nActual: %+v", prov.exp, act)
			}
		})
	}
}

type parseAcceptHeader struct {
	name   string
	header string
	exp    mimeheader.AcceptHeader
}

func providerParseAcceptHeader() []parseAcceptHeader {
	return []parseAcceptHeader{
		{
			name:   "Empty",
			header: "",
			exp:    mimeheader.NewAcceptHeaderPlain([]mimeheader.MimeHeader{}),
		},
		{
			name:   "Empty object",
			header: "{}",
			exp:    mimeheader.NewAcceptHeaderPlain([]mimeheader.MimeHeader{}),
		},
		{
			name:   "Full wildcard",
			header: "*/*",
			exp: mimeheader.NewAcceptHeaderPlain([]mimeheader.MimeHeader{
				{
					MimeType: mimeheader.MimeType{
						Type:    "*",
						Subtype: "*",
						Params:  map[string]string{},
					},
					Quality: 1.0,
				},
			}),
		},
		{
			name:   "Header with sorting",
			header: "*/*; q=0.9; s=1, image/*; q=0.9; s=4, application/json; q=0.9; b=3;, text/plain",
			exp: mimeheader.NewAcceptHeaderPlain([]mimeheader.MimeHeader{
				{
					MimeType: mimeheader.MimeType{
						Type:    "text",
						Subtype: "plain",
						Params:  map[string]string{},
					},
					Quality: 1.0,
				},
				{
					MimeType: mimeheader.MimeType{
						Type:    "application",
						Subtype: "json",
						Params:  map[string]string{"q": "0.9", "b": "3"},
					},
					Quality: 0.9,
				},
				{
					MimeType: mimeheader.MimeType{
						Type:    "image",
						Subtype: "*",
						Params:  map[string]string{"q": "0.9", "s": "4"},
					},
					Quality: 0.9,
				},
				{
					MimeType: mimeheader.MimeType{
						Type:    "*",
						Subtype: "*",
						Params:  map[string]string{"q": "0.9", "s": "1"},
					},
					Quality: 0.9,
				},
			}),
		},
		{
			name:   "Lost wildcard",
			header: "*/* q=0.9; s=1, image/*; q=0.9; s=4, application/json; q=0.9; b=3;, text/plain",
			exp: mimeheader.NewAcceptHeaderPlain([]mimeheader.MimeHeader{
				{
					MimeType: mimeheader.MimeType{
						Type:    "text",
						Subtype: "plain",
						Params:  map[string]string{},
					},
					Quality: 1.0,
				},
				{
					MimeType: mimeheader.MimeType{
						Type:    "application",
						Subtype: "json",
						Params:  map[string]string{"q": "0.9", "b": "3"},
					},
					Quality: 0.9,
				},
				{
					MimeType: mimeheader.MimeType{
						Type:    "image",
						Subtype: "*",
						Params:  map[string]string{"q": "0.9", "s": "4"},
					},
					Quality: 0.9,
				},
			}),
		},
	}
}