feat: initial parser exposing walker/writer

2026-02-07 09:38:33 +00:00 · 2020-06-30 17:43:04 +02:00
parent 7207a5d59e
commit 6ea3fc1963
10 changed files with 513 additions and 0 deletions
--- a/pkg/message/parser/walker_test.go
+++ b/pkg/message/parser/walker_test.go
@ -0,0 +1,64 @@
+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(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().
+		WithContentTypeHandler("text/html", 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().
+		WithContentDispositionHandler("attachment", func(p *Part, hdl PartHandler) (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)
+}