80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
|
package encoding_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"apigo.cc/go/encoding"
|
||
|
|
)
|
||
|
|
|
||
|
|
// --- Hex ---
|
||
|
|
func TestHex(t *testing.T) {
|
||
|
|
data := []byte("hello go")
|
||
|
|
encoded := encoding.HexToString(data)
|
||
|
|
decoded, err := encoding.UnHex([]byte(encoded))
|
||
|
|
if err != nil || !bytes.Equal(data, decoded) {
|
||
|
|
t.Fatal("Hex roundtrip failed")
|
||
|
|
}
|
||
|
|
|
||
|
|
if !bytes.Equal(encoding.MustUnHexFromString(encoded), data) {
|
||
|
|
t.Error("MustUnHexFromString failed")
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(encoding.MustUnHexFromString("!@#$")) != 0 {
|
||
|
|
t.Error("MustUnHexFromString should return empty for invalid hex chars")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// --- Base64 ---
|
||
|
|
func TestBase64(t *testing.T) {
|
||
|
|
data := []byte("hello world")
|
||
|
|
|
||
|
|
// Standard
|
||
|
|
enc := encoding.Base64ToString(data)
|
||
|
|
dec, err := encoding.UnBase64([]byte(enc))
|
||
|
|
if err != nil || !bytes.Equal(data, dec) {
|
||
|
|
t.Error("Base64 roundtrip failed")
|
||
|
|
}
|
||
|
|
|
||
|
|
if !bytes.Equal(encoding.MustUnBase64FromString(enc), data) {
|
||
|
|
t.Error("MustUnBase64FromString failed")
|
||
|
|
}
|
||
|
|
|
||
|
|
// URL
|
||
|
|
uEnc := encoding.UrlBase64ToString([]byte("hello/world+"))
|
||
|
|
uDec, _ := encoding.UnUrlBase64([]byte(uEnc))
|
||
|
|
if !bytes.Equal([]byte("hello/world+"), uDec) {
|
||
|
|
t.Error("UrlBase64 roundtrip failed")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// --- Web ---
|
||
|
|
func TestWebEncoding(t *testing.T) {
|
||
|
|
// URL
|
||
|
|
data := []byte("a b+c")
|
||
|
|
enc := encoding.UrlEncode(data)
|
||
|
|
if !bytes.Equal(encoding.MustUnUrlEncode(enc), data) {
|
||
|
|
t.Error("UrlEncode roundtrip failed")
|
||
|
|
}
|
||
|
|
if len(encoding.MustUnUrlEncode("%ZZ")) != 0 {
|
||
|
|
t.Error("MustUnUrlEncode should return empty for invalid input")
|
||
|
|
}
|
||
|
|
|
||
|
|
// HTML
|
||
|
|
htmlData := []byte("<script>")
|
||
|
|
escaped := encoding.HtmlEscape(htmlData)
|
||
|
|
if encoding.MustUnHtmlEscape(escaped) != string(htmlData) {
|
||
|
|
t.Error("HtmlEscape roundtrip failed")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// --- Utf8 ---
|
||
|
|
func TestUtf8(t *testing.T) {
|
||
|
|
if !encoding.Utf8Valid([]byte("你好")) {
|
||
|
|
t.Error("Valid UTF-8 should pass")
|
||
|
|
}
|
||
|
|
if encoding.Utf8Valid([]byte{0xff, 0xff}) {
|
||
|
|
t.Error("Invalid UTF-8 should fail")
|
||
|
|
}
|
||
|
|
}
|