-
Notifications
You must be signed in to change notification settings - Fork 0
/
email2pushover_test.go
65 lines (50 loc) · 1.66 KB
/
email2pushover_test.go
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
package main
import (
"gopkg.in/stretchr/testify.v1/assert"
"os"
"testing"
)
func TestCapitalize(t *testing.T) {
assert := assert.New(t)
assert.Equal("Hello", Capitalize("hello"))
assert.Equal("Hello", Capitalize("Hello"))
assert.Equal("Hello world", Capitalize("hello world"))
assert.Equal("Èllo", Capitalize("èllo"))
}
func TestParseHeaderList(t *testing.T) {
assert := assert.New(t)
assert.Equal(
[]string{"From", "To", "Subject"},
ParseHeaderList(" from,to , subject"))
}
func TestExtractHeadersFromMail(t *testing.T) {
assert := assert.New(t)
email, err := os.Open("test.eml")
if err != nil {
t.Fatal(err)
}
expected := make(map[string]string)
expected["From"] = "Fictious sender <sender@example.com>"
expected["To"] = "\"Fictious recipient\" <recipient@example.com>"
expected["Subject"] = "Example subject"
expected["No_such_header"] = ""
headerlist := ParseHeaderList("from,to,subject,no_such_header")
headers, err := extractHeadersFromMail(email, headerlist)
assert.Nil(err)
assert.Equal(expected, headers)
}
func TestConstructMessageBody(t *testing.T) {
assert := assert.New(t)
headerlist := []string{"From", "To", "Subject"}
headers := make(map[string]string)
headers["From"] = "Fictious sender <sender@example.com>"
headers["To"] = "\"Fictious recipient\" <recipient@example.com>"
headers["Subject"] = "Example subject"
expected_body := "From: Fictious sender <sender@example.com>\n" +
"To: \"Fictious recipient\" <recipient@example.com>\n" +
"Subject: Example subject"
assert.Equal(expected_body, constructMessageBody(headers, headerlist))
assert.Panics(func() {
constructMessageBody(headers, []string{"Missing key"})
})
}