// Copyright 2011 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. package mail import ( "bytes" "io/ioutil" "reflect" "strings" "testing" "time" ) var parseTests = []struct { in string header Header body string }{ { // RFC 5322, Appendix A.1.1 in: `From: John Doe To: Mary Smith Subject: Saying Hello Date: Fri, 21 Nov 1997 09:55:06 -0600 Message-ID: <1234@local.machine.example> This is a message just to say hello. So, "Hello". `, header: Header{ "From": []string{"John Doe "}, "To": []string{"Mary Smith "}, "Subject": []string{"Saying Hello"}, "Date": []string{"Fri, 21 Nov 1997 09:55:06 -0600"}, "Message-Id": []string{"<1234@local.machine.example>"}, }, body: "This is a message just to say hello.\nSo, \"Hello\".\n", }, } func TestParsing(t *testing.T) { for i, test := range parseTests { msg, err := ReadMessage(bytes.NewBuffer([]byte(test.in))) if err != nil { t.Errorf("test #%d: Failed parsing message: %v", i, err) continue } if !headerEq(msg.Header, test.header) { t.Errorf("test #%d: Incorrectly parsed message header.\nGot:\n%+v\nWant:\n%+v", i, msg.Header, test.header) } body, err := ioutil.ReadAll(msg.Body) if err != nil { t.Errorf("test #%d: Failed reading body: %v", i, err) continue } bodyStr := string(body) if bodyStr != test.body { t.Errorf("test #%d: Incorrectly parsed message body.\nGot:\n%+v\nWant:\n%+v", i, bodyStr, test.body) } } } func headerEq(a, b Header) bool { if len(a) != len(b) { return false } for k, as := range a { bs, ok := b[k] if !ok { return false } if !reflect.DeepEqual(as, bs) { return false } } return true } func TestDateParsing(t *testing.T) { tests := []struct { dateStr string exp time.Time }{ // RFC 5322, Appendix A.1.1 { "Fri, 21 Nov 1997 09:55:06 -0600", time.Date(1997, 11, 21, 9, 55, 6, 0, time.FixedZone("", -6*60*60)), }, // RFC5322, Appendix A.6.2 // Obsolete date. { "21 Nov 97 09:55:06 GMT", time.Date(1997, 11, 21, 9, 55, 6, 0, time.FixedZone("GMT", 0)), }, // Commonly found format not specified by RFC 5322. { "Fri, 21 Nov 1997 09:55:06 -0600 (MDT)", time.Date(1997, 11, 21, 9, 55, 6, 0, time.FixedZone("", -6*60*60)), }, } for _, test := range tests { hdr := Header{ "Date": []string{test.dateStr}, } date, err := hdr.Date() if err != nil { t.Errorf("Failed parsing %q: %v", test.dateStr, err) continue } if !date.Equal(test.exp) { t.Errorf("Parse of %q: got %+v, want %+v", test.dateStr, date, test.exp) } } } func TestAddressParsingError(t *testing.T) { const txt = "=?iso-8859-2?Q?Bogl=E1rka_Tak=E1cs?= " _, err := ParseAddress(txt) if err == nil || !strings.Contains(err.Error(), "charset not supported") { t.Errorf(`mail.ParseAddress(%q) err: %q, want ".*charset not supported.*"`, txt, err) } } func TestAddressParsing(t *testing.T) { tests := []struct { addrsStr string exp []*Address }{ // Bare address { `jdoe@machine.example`, []*Address{{ Address: "jdoe@machine.example", }}, }, // RFC 5322, Appendix A.1.1 { `John Doe `, []*Address{{ Name: "John Doe", Address: "jdoe@machine.example", }}, }, // RFC 5322, Appendix A.1.2 { `"Joe Q. Public" `, []*Address{{ Name: "Joe Q. Public", Address: "john.q.public@example.com", }}, }, { `Mary Smith , jdoe@example.org, Who? `, []*Address{ { Name: "Mary Smith", Address: "mary@x.test", }, { Address: "jdoe@example.org", }, { Name: "Who?", Address: "one@y.test", }, }, }, { `, "Giant; \"Big\" Box" `, []*Address{ { Address: "boss@nil.test", }, { Name: `Giant; "Big" Box`, Address: "sysservices@example.net", }, }, }, // RFC 5322, Appendix A.1.3 // TODO(dsymonds): Group addresses. // RFC 2047 "Q"-encoded ISO-8859-1 address. { `=?iso-8859-1?q?J=F6rg_Doe?= `, []*Address{ { Name: `Jörg Doe`, Address: "joerg@example.com", }, }, }, // RFC 2047 "Q"-encoded US-ASCII address. Dumb but legal. { `=?us-ascii?q?J=6Frg_Doe?= `, []*Address{ { Name: `Jorg Doe`, Address: "joerg@example.com", }, }, }, // RFC 2047 "Q"-encoded UTF-8 address. { `=?utf-8?q?J=C3=B6rg_Doe?= `, []*Address{ { Name: `Jörg Doe`, Address: "joerg@example.com", }, }, }, // RFC 2047, Section 8. { `=?ISO-8859-1?Q?Andr=E9?= Pirard `, []*Address{ { Name: `André Pirard`, Address: "PIRARD@vm1.ulg.ac.be", }, }, }, // Custom example of RFC 2047 "B"-encoded ISO-8859-1 address. { `=?ISO-8859-1?B?SvZyZw==?= `, []*Address{ { Name: `Jörg`, Address: "joerg@example.com", }, }, }, // Custom example of RFC 2047 "B"-encoded UTF-8 address. { `=?UTF-8?B?SsO2cmc=?= `, []*Address{ { Name: `Jörg`, Address: "joerg@example.com", }, }, }, // Custom example with "." in name. For issue 4938 { `Asem H. `, []*Address{ { Name: `Asem H.`, Address: "noreply@example.com", }, }, }, } for _, test := range tests { if len(test.exp) == 1 { addr, err := ParseAddress(test.addrsStr) if err != nil { t.Errorf("Failed parsing (single) %q: %v", test.addrsStr, err) continue } if !reflect.DeepEqual([]*Address{addr}, test.exp) { t.Errorf("Parse (single) of %q: got %+v, want %+v", test.addrsStr, addr, test.exp) } } addrs, err := ParseAddressList(test.addrsStr) if err != nil { t.Errorf("Failed parsing (list) %q: %v", test.addrsStr, err) continue } if !reflect.DeepEqual(addrs, test.exp) { t.Errorf("Parse (list) of %q: got %+v, want %+v", test.addrsStr, addrs, test.exp) } } } func TestAddressFormatting(t *testing.T) { tests := []struct { addr *Address exp string }{ { &Address{Address: "bob@example.com"}, "", }, { &Address{Name: "Bob", Address: "bob@example.com"}, `"Bob" `, }, { // note the ö (o with an umlaut) &Address{Name: "Böb", Address: "bob@example.com"}, `=?utf-8?q?B=C3=B6b?= `, }, { &Address{Name: "Bob Jane", Address: "bob@example.com"}, `"Bob Jane" `, }, { &Address{Name: "Böb Jacöb", Address: "bob@example.com"}, `=?utf-8?q?B=C3=B6b_Jac=C3=B6b?= `, }, } for _, test := range tests { s := test.addr.String() if s != test.exp { t.Errorf("Address%+v.String() = %v, want %v", *test.addr, s, test.exp) } } }