mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-17 15:46:44 +00:00
87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package parser
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWalker(t *testing.T) {
|
|
p := newTestParser(t, "text_html_octet_attachment.eml")
|
|
|
|
allBodies := [][]byte{}
|
|
|
|
walker := p.
|
|
NewWalker().
|
|
WithDefaultHandler(NewPartHandler().OnEnter(func(p *Part) (err error) {
|
|
if p.Body != nil {
|
|
allBodies = append(allBodies, p.Body)
|
|
}
|
|
return
|
|
}))
|
|
|
|
assert.NoError(t, walker.Walk())
|
|
assert.ElementsMatch(t, [][]byte{
|
|
[]byte("<html><body>This is body of <b>HTML mail</b> with attachment</body></html>"),
|
|
[]byte("if you are reading this, hi!"),
|
|
}, allBodies)
|
|
}
|
|
|
|
func TestWalkerTypeHandler(t *testing.T) {
|
|
p := newTestParser(t, "text_html_octet_attachment.eml")
|
|
|
|
html := [][]byte{}
|
|
|
|
walker := p.NewWalker()
|
|
|
|
walker.RegisterContentTypeHandler("text/html").
|
|
OnEnter(func(p *Part) (err error) {
|
|
html = append(html, p.Body)
|
|
return
|
|
})
|
|
|
|
assert.NoError(t, walker.Walk())
|
|
assert.ElementsMatch(t, [][]byte{
|
|
[]byte("<html><body>This is body of <b>HTML mail</b> with attachment</body></html>"),
|
|
}, html)
|
|
}
|
|
|
|
func TestWalkerDispositionHandler(t *testing.T) {
|
|
p := newTestParser(t, "text_html_octet_attachment.eml")
|
|
|
|
attachments := [][]byte{}
|
|
|
|
walker := p.NewWalker()
|
|
|
|
walker.RegisterContentDispositionHandler("attachment").
|
|
OnEnter(func(p *Part, hdl PartHandlerFunc) (err error) {
|
|
attachments = append(attachments, p.Body)
|
|
return
|
|
})
|
|
|
|
assert.NoError(t, walker.Walk())
|
|
assert.ElementsMatch(t, [][]byte{
|
|
[]byte("if you are reading this, hi!"),
|
|
}, attachments)
|
|
}
|
|
|
|
func TestWalkerDispositionAndTypeHandler(t *testing.T) {
|
|
p := newTestParser(t, "text_html_octet_attachment.eml")
|
|
|
|
walker := p.NewWalker()
|
|
|
|
var enter, exit int
|
|
|
|
walker.RegisterContentTypeHandler("application/octet-stream").
|
|
OnEnter(func(p *Part) (err error) { enter++; return }).
|
|
OnExit(func(p *Part) (err error) { exit--; return })
|
|
|
|
walker.RegisterContentDispositionHandler("attachment").
|
|
OnEnter(func(p *Part, hdl PartHandlerFunc) (err error) { _ = hdl(p); _ = hdl(p); return }).
|
|
OnExit(func(p *Part, hdl PartHandlerFunc) (err error) { _ = hdl(p); _ = hdl(p); return })
|
|
|
|
assert.NoError(t, walker.Walk())
|
|
assert.Equal(t, 2, enter)
|
|
assert.Equal(t, -2, exit)
|
|
}
|