textedit/escape_test.go

package textedit_test

import (
	"bytes"
	"code.squareroundforest.org/arpio/textedit"
	"testing"
)

func TestEscape(t *testing.T) {
	t.Run("basic", func(t *testing.T) {
		var b bytes.Buffer
		w := textedit.New(&b, textedit.Escape('\\', '\n', '\t', '"'))
		w.Write([]byte("foo\nbar\t\"baz\""))
		w.Flush()
		if b.String() != "foo\\\nbar\\\t\\\"baz\\\"" {
			t.Fatal(b.String())
		}
	})
}