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("This is body of HTML mail with attachment"),
[]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("This is body of HTML mail with attachment"),
}, 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)
}