package quotedprintable import ( "bytes" "errors" "fmt" "io" "io/ioutil" "strings" "testing" ) func ExampleWordEncoder_Encode() { fmt.Println(QEncoding.Encode("utf-8", "¡Hola, señor!")) fmt.Println(QEncoding.Encode("utf-8", "Hello!")) fmt.Println(BEncoding.Encode("UTF-8", "¡Hola, señor!")) fmt.Println(QEncoding.Encode("ISO-8859-1", "Caf\xE9")) // Output: // =?utf-8?q?=C2=A1Hola,_se=C3=B1or!?= // Hello! // =?UTF-8?b?wqFIb2xhLCBzZcOxb3Ih?= // =?ISO-8859-1?q?Caf=E9?= } func ExampleWordDecoder_Decode() { dec := new(WordDecoder) header, err := dec.Decode("=?utf-8?q?=C2=A1Hola,_se=C3=B1or!?=") if err != nil { panic(err) } fmt.Println(header) dec.CharsetReader = func(charset string, input io.Reader) (io.Reader, error) { switch charset { case "x-case": // Fake character set for example. // Real use would integrate with packages such // as code.google.com/p/go-charset content, err := ioutil.ReadAll(input) if err != nil { return nil, err } return bytes.NewReader(bytes.ToUpper(content)), nil } return nil, fmt.Errorf("unhandled charset %q", charset) } header, err = dec.Decode("=?x-case?q?hello!?=") if err != nil { panic(err) } fmt.Println(header) // Output: // ¡Hola, señor! // HELLO! } func ExampleWordDecoder_DecodeHeader() { dec := new(WordDecoder) header, err := dec.DecodeHeader("=?utf-8?q?=C3=89ric?= , =?utf-8?q?Ana=C3=AFs?= ") if err != nil { panic(err) } fmt.Println(header) header, err = dec.DecodeHeader("=?utf-8?q?=C2=A1Hola,?= =?utf-8?q?_se=C3=B1or!?=") if err != nil { panic(err) } fmt.Println(header) dec.CharsetReader = func(charset string, input io.Reader) (io.Reader, error) { switch charset { case "x-case": // Fake character set for example. // Real use would integrate with packages such // as code.google.com/p/go-charset content, err := ioutil.ReadAll(input) if err != nil { return nil, err } return bytes.NewReader(bytes.ToUpper(content)), nil } return nil, fmt.Errorf("unhandled charset %q", charset) } header, err = dec.DecodeHeader("=?x-case?q?hello_?= =?x-case?q?world!?=") if err != nil { panic(err) } fmt.Println(header) // Output: // Éric , Anaïs // ¡Hola, señor! // HELLO WORLD! } func TestEncodeWord(t *testing.T) { utf8, iso88591 := "utf-8", "iso-8859-1" tests := []struct { enc WordEncoder charset string src, exp string }{ {QEncoding, utf8, "François-Jérôme", "=?utf-8?q?Fran=C3=A7ois-J=C3=A9r=C3=B4me?="}, {BEncoding, utf8, "Café", "=?utf-8?b?Q2Fmw6k=?="}, {QEncoding, iso88591, "La Seleção", "=?iso-8859-1?q?La_Sele=C3=A7=C3=A3o?="}, {QEncoding, utf8, "", ""}, {QEncoding, utf8, "A", "A"}, {QEncoding, iso88591, "a", "a"}, {QEncoding, utf8, "123 456", "123 456"}, {QEncoding, utf8, "\t !\"#$%&'()*+,-./ :;<>?@[\\]^_`{|}~", "\t !\"#$%&'()*+,-./ :;<>?@[\\]^_`{|}~"}, } for _, test := range tests { if s := test.enc.Encode(test.charset, test.src); s != test.exp { t.Errorf("Encode(%q) = %q, want %q", test.src, s, test.exp) } } } func TestDecodeWord(t *testing.T) { tests := []struct { src, exp string hasErr bool }{ {"=?UTF-8?Q?=C2=A1Hola,_se=C3=B1or!?=", "¡Hola, señor!", false}, {"=?UTF-8?Q?Fran=C3=A7ois-J=C3=A9r=C3=B4me?=", "François-Jérôme", false}, {"=?UTF-8?q?ascii?=", "ascii", false}, {"=?utf-8?B?QW5kcsOp?=", "André", false}, {"=?ISO-8859-1?Q?Rapha=EBl_Dupont?=", "Raphaël Dupont", false}, {"=?utf-8?b?IkFudG9uaW8gSm9zw6kiIDxqb3NlQGV4YW1wbGUub3JnPg==?=", `"Antonio José" `, false}, {"=?UTF-8?A?Test?=", "", true}, {"=?UTF-8?Q?A=B?=", "", true}, {"=?UTF-8?Q?=A?=", "", true}, {"=?UTF-8?A?A?=", "", true}, } for _, test := range tests { dec := new(WordDecoder) s, err := dec.Decode(test.src) if test.hasErr && err == nil { t.Errorf("Decode(%q) should return an error", test.src) continue } if !test.hasErr && err != nil { t.Errorf("Decode(%q): %v", test.src, err) continue } if s != test.exp { t.Errorf("Decode(%q) = %q, want %q", test.src, s, test.exp) } } } func TestDecodeHeader(t *testing.T) { tests := []struct { src, exp string }{ {"=?UTF-8?Q?=C2=A1Hola,_se=C3=B1or!?=", "¡Hola, señor!"}, {"=?UTF-8?Q?Fran=C3=A7ois-J=C3=A9r=C3=B4me?=", "François-Jérôme"}, {"=?UTF-8?q?ascii?=", "ascii"}, {"=?utf-8?B?QW5kcsOp?=", "André"}, {"=?ISO-8859-1?Q?Rapha=EBl_Dupont?=", "Raphaël Dupont"}, {"Jean", "Jean"}, {"=?utf-8?b?IkFudG9uaW8gSm9zw6kiIDxqb3NlQGV4YW1wbGUub3JnPg==?=", `"Antonio José" `}, {"=?UTF-8?A?Test?=", "=?UTF-8?A?Test?="}, {"=?UTF-8?Q?A=B?=", "=?UTF-8?Q?A=B?="}, {"=?UTF-8?Q?=A?=", "=?UTF-8?Q?=A?="}, {"=?UTF-8?A?A?=", "=?UTF-8?A?A?="}, // Incomplete words {"=?", "=?"}, {"=?UTF-8?", "=?UTF-8?"}, {"=?UTF-8?=", "=?UTF-8?="}, {"=?UTF-8?Q", "=?UTF-8?Q"}, {"=?UTF-8?Q?", "=?UTF-8?Q?"}, {"=?UTF-8?Q?=", "=?UTF-8?Q?="}, {"=?UTF-8?Q?A", "=?UTF-8?Q?A"}, {"=?UTF-8?Q?A?", "=?UTF-8?Q?A?"}, // Tests from RFC 2047 {"=?ISO-8859-1?Q?a?=", "a"}, {"=?ISO-8859-1?Q?a?= b", "a b"}, {"=?ISO-8859-1?Q?a?= =?ISO-8859-1?Q?b?=", "ab"}, {"=?ISO-8859-1?Q?a?= =?ISO-8859-1?Q?b?=", "ab"}, {"=?ISO-8859-1?Q?a?= \r\n\t =?ISO-8859-1?Q?b?=", "ab"}, {"=?ISO-8859-1?Q?a_b?=", "a b"}, } for _, test := range tests { dec := new(WordDecoder) s, err := dec.DecodeHeader(test.src) if err != nil { t.Errorf("DecodeHeader(%q): %v", test.src, err) } if s != test.exp { t.Errorf("DecodeHeader(%q) = %q, want %q", test.src, s, test.exp) } } } func TestCharsetDecoder(t *testing.T) { tests := []struct { src string want string charsets []string content []string }{ {"=?utf-8?b?Q2Fmw6k=?=", "Café", nil, nil}, {"=?ISO-8859-1?Q?caf=E9?=", "café", nil, nil}, {"=?US-ASCII?Q?foo_bar?=", "foo bar", nil, nil}, {"=?utf-8?Q?=?=", "=?utf-8?Q?=?=", nil, nil}, {"=?utf-8?Q?=A?=", "=?utf-8?Q?=A?=", nil, nil}, { "=?ISO-8859-15?Q?f=F5=F6?= =?windows-1252?Q?b=E0r?=", "f\xf5\xf6b\xe0r", []string{"iso-8859-15", "windows-1252"}, []string{"f\xf5\xf6", "b\xe0r"}, }, } for _, test := range tests { i := 0 dec := &WordDecoder{ CharsetReader: func(charset string, input io.Reader) (io.Reader, error) { if charset != test.charsets[i] { t.Errorf("DecodeHeader(%q), got charset %q, want %q", test.src, charset, test.charsets[i]) } content, err := ioutil.ReadAll(input) if err != nil { t.Errorf("DecodeHeader(%q), error in reader: %v", test.src, err) } got := string(content) if got != test.content[i] { t.Errorf("DecodeHeader(%q), got content %q, want %q", test.src, got, test.content[i]) } i++ return strings.NewReader(got), nil }, } got, err := dec.DecodeHeader(test.src) if err != nil { t.Errorf("DecodeHeader(%q): %v", test.src, err) } if got != test.want { t.Errorf("DecodeHeader(%q) = %q, want %q", test.src, got, test.want) } } } func TestCharsetDecoderError(t *testing.T) { dec := &WordDecoder{ CharsetReader: func(charset string, input io.Reader) (io.Reader, error) { return nil, errors.New("Test error") }, } if _, err := dec.DecodeHeader("=?charset?Q?foo?="); err == nil { t.Error("DecodeHeader should return an error") } } func BenchmarkQEncodeWord(b *testing.B) { for i := 0; i < b.N; i++ { QEncoding.Encode("UTF-8", "¡Hola, señor!") } } func BenchmarkQDecodeWord(b *testing.B) { dec := new(WordDecoder) for i := 0; i < b.N; i++ { dec.Decode("=?utf-8?q?=C2=A1Hola,_se=C3=B1or!?=") } } func BenchmarkQDecodeHeader(b *testing.B) { dec := new(WordDecoder) for i := 0; i < b.N; i++ { dec.Decode("=?utf-8?q?=C2=A1Hola,_se=C3=B1or!?=") } }