172 lines
4.2 KiB
Go
172 lines
4.2 KiB
Go
package parse // import "github.com/tdewolff/parse"
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"mime"
|
|
"testing"
|
|
|
|
"github.com/tdewolff/test"
|
|
)
|
|
|
|
func TestParseNumber(t *testing.T) {
|
|
var numberTests = []struct {
|
|
number string
|
|
expected int
|
|
}{
|
|
{"5", 1},
|
|
{"0.51", 4},
|
|
{"0.5e-99", 7},
|
|
{"0.5e-", 3},
|
|
{"+50.0", 5},
|
|
{".0", 2},
|
|
{"0.", 1},
|
|
{"", 0},
|
|
{"+", 0},
|
|
{".", 0},
|
|
{"a", 0},
|
|
}
|
|
for _, tt := range numberTests {
|
|
t.Run(tt.number, func(t *testing.T) {
|
|
n := Number([]byte(tt.number))
|
|
test.T(t, n, tt.expected)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseDimension(t *testing.T) {
|
|
var dimensionTests = []struct {
|
|
dimension string
|
|
expectedNum int
|
|
expectedUnit int
|
|
}{
|
|
{"5px", 1, 2},
|
|
{"5px ", 1, 2},
|
|
{"5%", 1, 1},
|
|
{"5em", 1, 2},
|
|
{"px", 0, 0},
|
|
{"1", 1, 0},
|
|
{"1~", 1, 0},
|
|
}
|
|
for _, tt := range dimensionTests {
|
|
t.Run(tt.dimension, func(t *testing.T) {
|
|
num, unit := Dimension([]byte(tt.dimension))
|
|
test.T(t, num, tt.expectedNum, "number")
|
|
test.T(t, unit, tt.expectedUnit, "unit")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMediatype(t *testing.T) {
|
|
var mediatypeTests = []struct {
|
|
mediatype string
|
|
expectedMimetype string
|
|
expectedParams map[string]string
|
|
}{
|
|
{"text/plain", "text/plain", nil},
|
|
{"text/plain;charset=US-ASCII", "text/plain", map[string]string{"charset": "US-ASCII"}},
|
|
{" text/plain ; charset = US-ASCII ", "text/plain", map[string]string{"charset": "US-ASCII"}},
|
|
{" text/plain a", "text/plain", nil},
|
|
{"text/plain;base64", "text/plain", map[string]string{"base64": ""}},
|
|
{"text/plain;inline=;base64", "text/plain", map[string]string{"inline": "", "base64": ""}},
|
|
}
|
|
for _, tt := range mediatypeTests {
|
|
t.Run(tt.mediatype, func(t *testing.T) {
|
|
mimetype, _ := Mediatype([]byte(tt.mediatype))
|
|
test.String(t, string(mimetype), tt.expectedMimetype, "mimetype")
|
|
//test.T(t, params, tt.expectedParams, "parameters") // TODO
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseDataURI(t *testing.T) {
|
|
var dataURITests = []struct {
|
|
dataURI string
|
|
expectedMimetype string
|
|
expectedData string
|
|
expectedErr error
|
|
}{
|
|
{"www.domain.com", "", "", ErrBadDataURI},
|
|
{"data:,", "text/plain", "", nil},
|
|
{"data:text/xml,", "text/xml", "", nil},
|
|
{"data:,text", "text/plain", "text", nil},
|
|
{"data:;base64,dGV4dA==", "text/plain", "text", nil},
|
|
{"data:image/svg+xml,", "image/svg+xml", "", nil},
|
|
{"data:;base64,()", "", "", base64.CorruptInputError(0)},
|
|
}
|
|
for _, tt := range dataURITests {
|
|
t.Run(tt.dataURI, func(t *testing.T) {
|
|
mimetype, data, err := DataURI([]byte(tt.dataURI))
|
|
test.T(t, err, tt.expectedErr)
|
|
test.String(t, string(mimetype), tt.expectedMimetype, "mimetype")
|
|
test.String(t, string(data), tt.expectedData, "data")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseQuoteEntity(t *testing.T) {
|
|
var quoteEntityTests = []struct {
|
|
quoteEntity string
|
|
expectedQuote byte
|
|
expectedN int
|
|
}{
|
|
{""", '"', 5},
|
|
{"'", '\'', 6},
|
|
{""", '"', 8},
|
|
{"'", '\'', 6},
|
|
{""", '"', 6},
|
|
{"'", '\'', 6},
|
|
{">", 0x00, 0},
|
|
{"&", 0x00, 0},
|
|
}
|
|
for _, tt := range quoteEntityTests {
|
|
t.Run(tt.quoteEntity, func(t *testing.T) {
|
|
quote, n := QuoteEntity([]byte(tt.quoteEntity))
|
|
test.T(t, quote, tt.expectedQuote, "quote")
|
|
test.T(t, n, tt.expectedN, "quote length")
|
|
})
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
|
|
func BenchmarkParseMediatypeStd(b *testing.B) {
|
|
mediatype := "text/plain"
|
|
for i := 0; i < b.N; i++ {
|
|
mime.ParseMediaType(mediatype)
|
|
}
|
|
}
|
|
|
|
func BenchmarkParseMediatypeParamStd(b *testing.B) {
|
|
mediatype := "text/plain;inline=1"
|
|
for i := 0; i < b.N; i++ {
|
|
mime.ParseMediaType(mediatype)
|
|
}
|
|
}
|
|
|
|
func BenchmarkParseMediatypeParamsStd(b *testing.B) {
|
|
mediatype := "text/plain;charset=US-ASCII;language=US-EN;compression=gzip;base64"
|
|
for i := 0; i < b.N; i++ {
|
|
mime.ParseMediaType(mediatype)
|
|
}
|
|
}
|
|
|
|
func BenchmarkParseMediatypeParse(b *testing.B) {
|
|
mediatype := []byte("text/plain")
|
|
for i := 0; i < b.N; i++ {
|
|
Mediatype(mediatype)
|
|
}
|
|
}
|
|
|
|
func BenchmarkParseMediatypeParamParse(b *testing.B) {
|
|
mediatype := []byte("text/plain;inline=1")
|
|
for i := 0; i < b.N; i++ {
|
|
Mediatype(mediatype)
|
|
}
|
|
}
|
|
|
|
func BenchmarkParseMediatypeParamsParse(b *testing.B) {
|
|
mediatype := []byte("text/plain;charset=US-ASCII;language=US-EN;compression=gzip;base64")
|
|
for i := 0; i < b.N; i++ {
|
|
Mediatype(mediatype)
|
|
}
|
|
}
|