mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-11 13:16:53 +00:00
feat: recreate message with parser's writer
This commit is contained in:
@ -18,7 +18,6 @@
|
||||
package parser
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
|
||||
@ -30,10 +29,10 @@ type Parser struct {
|
||||
root *Part
|
||||
}
|
||||
|
||||
func New(b []byte) (*Parser, error) {
|
||||
func New(r io.Reader) (*Parser, error) {
|
||||
p := new(Parser)
|
||||
|
||||
entity, err := message.Read(bytes.NewReader(b))
|
||||
entity, err := message.Read(r)
|
||||
if err != nil && !message.IsUnknownCharset(err) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@ -18,7 +18,6 @@
|
||||
package parser
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
@ -29,15 +28,7 @@ import (
|
||||
)
|
||||
|
||||
func newTestParser(t *testing.T, msg string) *Parser {
|
||||
r := f(msg)
|
||||
|
||||
buf := new(bytes.Buffer)
|
||||
|
||||
if _, err := buf.ReadFrom(r); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
p, err := New(buf.Bytes())
|
||||
p, err := New(f(msg))
|
||||
require.NoError(t, err)
|
||||
|
||||
return p
|
||||
|
||||
Reference in New Issue
Block a user