feat(markdown): custom replacer

1 files changed, 21 insertions, 6 deletions

diff --git a/markdown/lexer_test.go b/markdown/lexer_test.go
index e994142..d8ec3ba 100644
--- a/markdown/lexer_test.go
+++ b/markdown/lexer_test.go
@@ -3,28 +3,43 @@ package markdown
 import "testing"
 
 func TestLex(t *testing.T) {
-	lxs := lex("bonjour les gens")
+	opt := new(Option)
+	lxs := lex("bonjour les gens", opt)
 	if lxs.String() != "Lexers[literal(bonjour les gens) ]" {
 		t.Errorf("invalid lex, got %s", lxs)
 	}
-	lxs = lex("# bonjour les gens")
+	lxs = lex("# bonjour les gens", opt)
 	if lxs.String() != "Lexers[header(#) literal( bonjour les gens) ]" {
 		t.Errorf("invalid lex, got %s", lxs)
 	}
-	lxs = lex("# bonjour les gens\nComment ça va ?")
+	lxs = lex("# bonjour les gens\nComment ça va ?", opt)
 	if lxs.String() != `Lexers[header(#) literal( bonjour les gens) break({\n}) literal(Comment ça va ?) ]` {
 		t.Errorf("invalid lex, got %s", lxs)
 	}
-	lxs = lex("***hey***, what's up?")
+	lxs = lex("***hey***, what's up?", opt)
 	if lxs.String() != "Lexers[modifier(***) literal(hey) modifier(***) literal(, what's up?) ]" {
 		t.Errorf("invalid lex, got %s", lxs)
 	}
-	lxs = lex(`Xxx\_DarkEmperor\_xxX`)
+	lxs = lex(`Xxx\_DarkEmperor\_xxX`, opt)
 	if lxs.String() != `Lexers[literal(Xxx_DarkEmperor_xxX) ]` {
 		t.Errorf("invalid lex, got %s", lxs)
 	}
-	lxs = lex(`* list`)
+	lxs = lex(`* list`, opt)
 	if lxs.String() != `Lexers[list(*) literal( list) ]` {
 		t.Errorf("invalid lex, got %s", lxs)
 	}
 }
+
+func TestLex_Replacer(t *testing.T) {
+	opt := &Option{
+		Replaces: map[rune]string{'~': " "},
+	}
+	lxs := lex("bonjour les gens", opt)
+	if lxs.String() != "Lexers[literal(bonjour les gens) ]" {
+		t.Errorf("invalid lex, got %s", lxs)
+	}
+	lxs = lex("bonjour les gens~!", opt)
+	if lxs.String() != "Lexers[literal(bonjour les gens) replace(~) external(!) ]" {
+		t.Errorf("invalid lex, got %s", lxs)
+	}
+}