package matcher_test import ( "fmt" "testing" "git.max-richter.dev/max/marka/parser/matcher" "git.max-richter.dev/max/marka/parser/utils" "git.max-richter.dev/max/marka/registry" "git.max-richter.dev/max/marka/template" ) func TestFuzzyFindAll(t *testing.T) { recipeMd := utils.ReadTestDataFile(t, "baguette.md") tests := []struct { Needle string Start, End, StartIndex int }{ {StartIndex: 0, Needle: "# Ingredients\n", Start: 77, End: 91}, {StartIndex: 0, Needle: "# Ingrdients\n", Start: 77, End: 91}, {StartIndex: 0, Needle: "# Inrdients\n", Start: 77, End: 91}, {StartIndex: 0, Needle: "---\n", Start: 0, End: 4}, {StartIndex: 4, Needle: "---\n", Start: 29, End: 33}, {StartIndex: 0, Needle: "# Steps\n", Start: 116, End: 124}, {StartIndex: 0, Needle: "# Stps\n", Start: 116, End: 124}, {StartIndex: 0, Needle: "# Step\n", Start: 116, End: 124}, } for _, test := range tests { start, end := matcher.FuzzyFind(recipeMd, test.StartIndex, test.Needle, 0.3) // allow 50% error if start != test.Start || end != test.End { t.Errorf("Start or end do not match: Needle=%q Start=%d/%d End=%d/%d", test.Needle, test.Start, start, test.End, end) } } } func TestFuzzyBlockMatch(t *testing.T) { recipeMd := utils.ReadTestDataFile(t, "baguette.md") schemaMd, err := registry.GetTemplate("Recipe") if err != nil { t.Errorf("Failed to load template: %s", err.Error()) t.FailNow() } blocks, err := template.CompileTemplate(schemaMd) if err != nil { t.Errorf("Failed to compile template: %s", err.Error()) t.FailNow() } for _, b := range blocks { fmt.Printf("block: %#v\n", b) } matches := matcher.MatchBlocksFuzzy(recipeMd, blocks, 0.3) expected := []struct { value string }{ { value: "author.name: Max Richter", }, { value: "Baguette", }, { value: "My favourite baguette recipe", }, { value: "- Flour\n- Water\n- Salt", }, { value: "1. Mix Flour Water and Salt\n2. Bake the bread\n", }, } for i, m := range matches { if i > len(expected)-1 { t.Errorf("No expected result for match: %d -> %q", i, m.GetContent()) t.FailNow() } if expected[i].value != m.GetContent() { t.Errorf("Match %d did not match expected: %q", i, m.GetContent()) } } }