yesBad/proton-bridge
1
0
Fork 0
forked from Silverfish/proton-bridge
Code Actions Activity

feat: better handling of multipart messages

Browse Source
This commit is contained in:
James Houlahan
2020-07-30 14:51:34 +02:00
parent a7da66ccbc
commit c6b18b45b5
10 changed files with 502 additions and 214 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/message/parser/walker_test.go
View File

@ -34,8 +34,7 @@ func TestWalkerTypeHandler(t *testing.T) {
walker := p.NewWalker()
walker.
RegisterContentTypeHandler("text/html").
walker.RegisterContentTypeHandler("text/html").
OnEnter(func(p *Part) (err error) {
html = append(html, p.Body)
return
@ -54,8 +53,7 @@ func TestWalkerDispositionHandler(t *testing.T) {
walker := p.NewWalker()
walker.
RegisterContentDispositionHandler("attachment").
walker.RegisterContentDispositionHandler("attachment").
OnEnter(func(p *Part, hdl PartHandlerFunc) (err error) {
attachments = append(attachments, p.Body)
return
@ -74,13 +72,11 @@ func TestWalkerDispositionAndTypeHandler(t *testing.T) {
var enter, exit int
walker.
RegisterContentTypeHandler("application/octet-stream").
walker.RegisterContentTypeHandler("application/octet-stream").
OnEnter(func(p *Part) (err error) { enter++; return }).
OnExit(func(p *Part) (err error) { exit--; return })
walker.
RegisterContentDispositionHandler("attachment").
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 })