Repository navigation
Expand file tree
/
Copy pathpop3_test.go
More file actions
202 lines (179 loc) · 4.69 KB
/
Copy pathpop3_test.go
File metadata and controls
202 lines (179 loc) · 4.69 KB
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
package pop3
import (
"bytes"
"fmt"
"io"
"log"
"net/smtp"
"strings"
"testing"
"github.com/emersion/go-message"
"github.com/emersion/go-message/mail"
)
const MSG = `Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do.`
// n represents number of messages to add to the testuser's inbox
func add_messages(n int) error {
to := []string{"recipient@example.net"}
msgs := make([][]byte, 0)
for i := 0; i < 5; i++ {
to := "To: recipient@example.net\r\n"
subject := fmt.Sprintf("Subject: Subject %d\r\n", i)
mime := "MIME-version: 1.0;\nContent-Type: text/plain; charset=\"UTF-8\";\n\n"
body := fmt.Sprintf("Message %d.\r\n"+MSG+"\r\n", i)
msg := []byte(to + subject + mime + body)
msgs = append(msgs, msg)
}
for _, msg := range msgs {
err := smtp.SendMail("localhost:2500", nil, "sender@example.org", to, msg)
if err != nil {
log.Fatal(err)
}
}
return nil
}
func getConnection() (*Conn, error) {
p := New(Opt{
Host: "localhost",
Port: 1100,
TLSEnabled: false,
})
c, err := p.NewConn()
if err != nil {
return nil, err
}
return c, nil
}
func readAndCompareMessageBody(m *message.Entity, msg string) error {
mr := mail.NewReader(m)
if mr != nil {
// This is a multipart message
for {
p, err := mr.NextPart()
if err == io.EOF {
break
} else if err != nil {
return err
}
8000
b, err := io.ReadAll(p.Body)
if err != nil {
return err
}
if !strings.EqualFold(string(b), msg) {
return fmt.Errorf("expected message body:\n%sreceived:\n%s", msg, string(b))
}
}
return nil
} else {
t, _, _ := m.Header.ContentType()
log.Println("This is a non-multipart message with type", t)
return nil
}
}
func TestAll(t *testing.T) {
c, err := getConnection()
if err != nil {
t.Fatal("error establishing connection to pop3 server ", err)
}
err = add_messages(5)
if err != nil {
t.Fatal("unable to send messages to the mail server", err)
}
// testing Auth
if err := c.Auth("recipient", "password"); err != nil {
t.Fatal(err)
}
// testing Stat
count, size, err := c.Stat()
if err != nil {
t.Fatal("error using Stat", err)
}
log.Printf("count: %d, size: %d\n", count, size)
// testing Uidl
msgIds, err := c.Uidl(0)
if err != nil {
t.Fatal("error using Uidl(0)", err)
}
if len(msgIds) != count {
t.Fatalf("Uidl returned: %d number of messages, but actually there are %d messages\n", len(msgIds), 5)
}
msgId, err := c.Uidl(msgIds[0].ID)
if err != nil {
t.Fatal("error using Uidl for positive message ID", err)
}
if len(msgId) != 1 {
t.Fatalf("Uidl returns a list of (message ID, message UID) pairs. If the optional msgID is > 0, then only that particular message is listed but it returned %d pair\n", len(msgId))
}
// testing List
msgs, err := c.List(0)
if err != nil {
t.Fatal("error using List(0)", err)
}
if len(msgs) != 5 {
t.Fatalf("List(0) returned incorrect number of messages got: %d actual: %d\n", len(msgs), 5)
}
msgId, err = c.List(msgs[1].ID)
if err != nil {
t.Fatal("error using List for positive message ID", err)
}
if len(msgId) != 1 {
t.Fatalf("List returns a list of (message ID, message UID) pairs. If the optional msgID is > 0, then only that particular message is listed but it returned %d pair\n", len(msgId))
}
// testing Retr
m, err := c.Retr(msgs[0].ID)
if err != nil {
t.Fatal("error using Retr", err)
}
if m.Header.Get("subject") != "Subject 0" {
t.Fatalf("Retr returned wrong subject returned: %s, expected: Subject 0 ", m.Header.Get("subject"))
}
err = readAndCompareMessageBody(m, "Message 0.\r\n"+MSG+"\r\n")
if err != nil {
t.Fatal(err)
}
// testing RetrRaw
mb, err := c.RetrRaw(msgs[0].ID)
if err != nil {
t.Fatal("error using RetrRaw", err)
}
b := mb.Bytes()
if !bytes.Contains(b, []byte("Message 0.\r\n"+MSG+"\r\n")) {
t.Fatalf("expected message body:\n%s, received:\n%s", "Message 0.\r\n"+MSG+"\r\n", string(b))
}
// testing Top
m, err = c.Top(msgs[0].ID, 1)
if err != nil {
t.Fatal("error using Top", err)
}
err = readAndCompareMessageBody(m, "Message 0.\r\n")
if err != nil {
t.Fatal(err)
}
// testing Noop
err = c.Noop()
if err != nil {
t.Fatal("error in using Noop", err)
}
// testing Dele
err = c.Dele([]int{1, 2}...)
if err != nil {
t.Fatal("error using Dele", err)
}
msgs, _ = c.List(0)
if len(msgs) != 3 {
t.Fatalf("after deleting 2 messages number of messages in inbox should be 3 but got %d", len(msgs))
}
// testing Rset, list
err = c.Rset()
if err != nil {
t.Fatal("error using Rset", err)
}
msgs, _ = c.List(0)
if len(msgs) != 5 {
t.Fatalf("after Rseting number of messages in inbox should be 5 but got %d", len(msgs))
}
// testing Quit
err = c.Quit()
if err != nil {
t.Fatal("error using Quit method", err)
}
}