mirror of
https://github.com/Jinnrry/PMail.git
synced 2025-10-04 23:12:51 +08:00
150 lines
3.2 KiB
Go
150 lines
3.2 KiB
Go
package parsemail
|
||
|
||
import (
|
||
"bytes"
|
||
"fmt"
|
||
"github.com/emersion/go-message"
|
||
log "github.com/sirupsen/logrus"
|
||
"io"
|
||
"os"
|
||
"pmail/config"
|
||
"pmail/db"
|
||
"pmail/session"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func testInit() {
|
||
// 设置日志格式为json格式
|
||
//log.SetFormatter(&log.JSONFormatter{})
|
||
|
||
log.SetReportCaller(true)
|
||
log.SetFormatter(&log.TextFormatter{
|
||
//以下设置只是为了使输出更美观
|
||
DisableColors: true,
|
||
TimestampFormat: "2006-01-02 15:03:04",
|
||
})
|
||
|
||
// 设置将日志输出到标准输出（默认的输出为stderr,标准错误）
|
||
// 日志消息输出可以是任意的io.writer类型
|
||
log.SetOutput(os.Stdout)
|
||
|
||
// 设置日志级别为warn以上
|
||
log.SetLevel(log.TraceLevel)
|
||
|
||
var cst, _ = time.LoadLocation("Asia/Shanghai")
|
||
time.Local = cst
|
||
|
||
config.Init()
|
||
Init()
|
||
db.Init()
|
||
session.Init()
|
||
|
||
}
|
||
func TestEmail_domainMatch(t *testing.T) {
|
||
//e := &Email{}
|
||
//dnsNames := []string{
|
||
// "*.mail.qq.com",
|
||
// "993.dav.qq.com",
|
||
// "993.eas.qq.com",
|
||
// "993.imap.qq.com",
|
||
// "993.pop.qq.com",
|
||
// "993.smtp.qq.com",
|
||
// "imap.qq.com",
|
||
// "mx1.qq.com",
|
||
// "mx2.qq.com",
|
||
// "mx3.qq.com",
|
||
// "pop.qq.com",
|
||
// "smtp.qq.com",
|
||
// "mail.qq.com",
|
||
//}
|
||
//
|
||
//fmt.Println(e.domainMatch("", dnsNames))
|
||
//fmt.Println(e.domainMatch("xjiangwei.cn", dnsNames))
|
||
//fmt.Println(e.domainMatch("qq.com", dnsNames))
|
||
//fmt.Println(e.domainMatch("test.aaa.mail.qq.com", dnsNames))
|
||
//fmt.Println(e.domainMatch("smtp.qq.com", dnsNames))
|
||
//fmt.Println(e.domainMatch("pop.qq.com", dnsNames))
|
||
//fmt.Println(e.domainMatch("test.mail.qq.com", dnsNames))
|
||
|
||
}
|
||
|
||
func Test_buildUser(t *testing.T) {
|
||
u := buildUser("Jinnrry N <jiangwei1995910@gmail.com>")
|
||
if u.EmailAddress != "jiangwei1995910@gmail.com" {
|
||
t.Error("error")
|
||
}
|
||
if u.Name != "Jinnrry N" {
|
||
t.Error("error")
|
||
}
|
||
|
||
u = buildUser("=?UTF-8?B?YWRtaW5AamlubnJyeS5jb20=?=<admin@jinnrry.com>")
|
||
if u.EmailAddress != "admin@jinnrry.com" {
|
||
t.Error("error")
|
||
}
|
||
if u.Name != "admin@jinnrry.com" {
|
||
t.Error("error")
|
||
}
|
||
|
||
u = buildUser("\"admin@jinnrry.com\" <admin@jinnrry.com>")
|
||
if u.EmailAddress != "admin@jinnrry.com" {
|
||
t.Error("error")
|
||
}
|
||
if u.Name != "admin@jinnrry.com" {
|
||
t.Error("error")
|
||
}
|
||
}
|
||
|
||
func TestEmailBuidlers(t *testing.T) {
|
||
var b bytes.Buffer
|
||
|
||
var h message.Header
|
||
h.SetContentType("multipart/alternative", nil)
|
||
w, err := message.CreateWriter(&b, h)
|
||
if err != nil {
|
||
}
|
||
|
||
var h1 message.Header
|
||
h1.SetContentType("text/html", nil)
|
||
w1, err := w.CreatePart(h1)
|
||
if err != nil {
|
||
}
|
||
io.WriteString(w1, "<h1>Hello World!</h1><p>This is an HTML part.</p>")
|
||
w1.Close()
|
||
|
||
var h2 message.Header
|
||
h2.SetContentType("text/plain", nil)
|
||
w2, err := w.CreatePart(h2)
|
||
if err != nil {
|
||
}
|
||
io.WriteString(w2, "Hello World!\n\nThis is a text part.")
|
||
w2.Close()
|
||
|
||
w.Close()
|
||
|
||
fmt.Println(b.String())
|
||
}
|
||
|
||
func TestEmail_builder(t *testing.T) {
|
||
testInit()
|
||
|
||
e := Email{
|
||
From: buildUser("i@test.com"),
|
||
To: buildUsers([]string{"to@test.com"}),
|
||
Subject: "Title",
|
||
HTML: []byte("Html"),
|
||
Text: []byte("Text"),
|
||
Attachments: []*Attachment{
|
||
{
|
||
Filename: "a.png",
|
||
ContentType: "image/jpeg",
|
||
Content: []byte("aaa"),
|
||
ContentID: "1",
|
||
},
|
||
},
|
||
}
|
||
|
||
rest := e.BuildBytes(nil, false)
|
||
fmt.Println(string(rest))
|
||
}
|