1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
package rfc822
import (
"io"
"os"
"path/filepath"
"testing"
"time"
"git.sr.ht/~rjarry/aerc/models"
"github.com/emersion/go-message/mail"
"github.com/stretchr/testify/require"
)
func TestMessageInfoParser(t *testing.T) {
rootDir := "testdata/message/valid"
msgFiles, err := os.ReadDir(rootDir)
die(err)
for _, fi := range msgFiles {
if fi.IsDir() {
continue
}
p := fi.Name()
t.Run(p, func(t *testing.T) {
m := newMockRawMessageFromPath(filepath.Join(rootDir, p))
mi, err := MessageInfo(m)
if err != nil {
t.Fatal("Failed to create MessageInfo with:", err)
}
if perr := mi.Error; perr != nil {
t.Fatal("Expected no parsing error, but got:", mi.Error)
}
})
}
}
func TestMessageInfoHandledError(t *testing.T) {
rootDir := "testdata/message/invalid"
msgFiles, err := os.ReadDir(rootDir)
die(err)
for _, fi := range msgFiles {
if fi.IsDir() {
continue
}
p := fi.Name()
t.Run(p, func(t *testing.T) {
m := newMockRawMessageFromPath(filepath.Join(rootDir, p))
mi, err := MessageInfo(m)
if err != nil {
t.Fatal(err)
}
if perr := mi.Error; perr == nil {
t.Fatal("Expected MessageInfo.Error, got none")
}
})
}
}
func TestParseMessageDate(t *testing.T) {
// we use different times for "Date" and "Received" fields to make sure the right one is parsed
tests := []struct {
date string
received string
utc time.Time
}{
{
date: "Fri, 22 Dec 2023 11:19:01 +0000",
received: "from aaa.bbb.com for <user@host.com>; Fri, 22 Dec 2023 06:19:02 -0500 (EST)",
utc: time.Date(2023, time.December, 22, 11, 19, 1, 0, time.UTC),
},
{
date: "Fri, 29 Dec 2023 14:06:37 +0100",
received: "from somewhere.com for a@b.c; Fri, 30 Dec 2023 4:06:43 +1300",
utc: time.Date(2023, time.December, 29, 13, 6, 37, 0, time.UTC),
},
{
date: "Fri, 29 Dec 2023 00:51:00 EST",
received: "by hostname.com; Fri, 29 Dec 2023 00:51:33 -0500 (EST)",
utc: time.Date(2023, time.December, 29, 5, 51, 33, 0, time.UTC),
},
}
for _, test := range tests {
h := mail.Header{}
h.SetText("Date", test.date)
h.SetText("Received", test.received)
res, err := parseDate(&h)
require.Nil(t, err)
require.Equal(t, test.utc, res.UTC())
}
}
type mockRawMessage struct {
path string
}
func newMockRawMessageFromPath(p string) *mockRawMessage {
return &mockRawMessage{
path: p,
}
}
func (m *mockRawMessage) NewReader() (io.ReadCloser, error) {
return os.Open(m.path)
}
func (m *mockRawMessage) ModelFlags() (models.Flags, error) { return 0, nil }
func (m *mockRawMessage) Labels() ([]string, error) { return nil, nil }
func (m *mockRawMessage) UID() uint32 { return 0 }
func die(err error) {
if err != nil {
panic(err)
}
}
|
