package pcre_test import ( "strings" "sync" "testing" "go.arsenm.dev/pcre" ) func TestCompileError(t *testing.T) { r, err := pcre.Compile("(") if err == nil { t.Error("expected error, got nil") } defer r.Close() } func TestMatch(t *testing.T) { r := pcre.MustCompile(`\d+ (?=USD)`) defer r.Close() matches := r.MatchString("9000 USD") if !matches { t.Error("expected 9000 USD to match") } matches = r.MatchString("9000 RUB") if matches { t.Error("expected 9000 RUB not to match") } matches = r.MatchString("800 USD") if !matches { t.Error("expected 800 USD to match") } matches = r.MatchString("700 CAD") if matches { t.Error("expected 700 CAD not to match") } matches = r.Match([]byte("8 USD")) if !matches { t.Error("expected 8 USD to match") } } func TestMatchUngreedy(t *testing.T) { r := pcre.MustCompileOpts(`Hello, (.+)\.`, pcre.Ungreedy) defer r.Close() submatches := r.FindAllStringSubmatch("Hello, World. Hello, pcre2.", 1) if submatches[0][1] != "World" { t.Errorf("expected World, got %s", submatches[0][1]) } matches := r.MatchString("hello, world.") if matches { t.Error("expected lowercase 'hello, world' not to match") } } func TestReplace(t *testing.T) { r := pcre.MustCompile(`(\d+)\.\d+`) defer r.Close() testStr := "123.54321 Test" newStr := r.ReplaceAllString(testStr, "${1}.12345") if newStr != "123.12345 Test" { t.Errorf(`expected "123.12345 Test", got "%s"`, newStr) } newStr = r.ReplaceAllString(testStr, "${9}.12345") if newStr != ".12345 Test" { t.Errorf(`expected ".12345 Test", got "%s"`, newStr) } newStr = r.ReplaceAllString(testStr, "${hi}.12345") if newStr != ".12345 Test" { t.Errorf(`expected ".12345 Test", got "%s"`, newStr) } newStr = r.ReplaceAllLiteralString(testStr, "${1}.12345") if newStr != "${1}.12345 Test" { t.Errorf(`expected "${1}.12345 Test", got "%s"`, newStr) } newStr = r.ReplaceAllStringFunc(testStr, func(s string) string { return strings.Replace(s, ".", ",", -1) }) if newStr != "123,54321 Test" { t.Errorf(`expected "123,54321 Test", got "%s"`, newStr) } } func TestSplit(t *testing.T) { r := pcre.MustCompile("a*") defer r.Close() split := r.Split("abaabaccadaaae", 5) expected := [5]string{"", "b", "b", "c", "cadaaae"} if *(*[5]string)(split) != expected { t.Errorf("expected %v, got %v", expected, split) } split = r.Split("", 0) if split != nil { t.Errorf("expected nil, got %v", split) } split = r.Split("", 5) if split[0] != "" { t.Errorf(`expected []string{""}, got %v`, split) } } func TestFind(t *testing.T) { r := pcre.MustCompile(`(\d+)`) defer r.Close() testStr := "3 times 4 is 12" matches := r.FindAllString(testStr, -1) if len(matches) != 3 { t.Errorf("expected length 3, got %d", len(matches)) } matches = r.FindAllString(testStr, 2) if len(matches) != 2 { t.Errorf("expected length 2, got %d", len(matches)) } if matches[0] != "3" || matches[1] != "4" { t.Errorf("expected [3 4], got %v", matches) } index := r.FindStringIndex(testStr) if index[0] != 0 || index[1] != 1 { t.Errorf("expected [0 1], got %v", index) } submatch := r.FindStringSubmatch(testStr) if submatch[0] != "3" { t.Errorf("expected 3, got %s", submatch[0]) } index = r.FindStringSubmatchIndex(testStr) if *(*[4]int)(index) != [4]int{0, 1, 0, 1} { t.Errorf("expected [0 1 0 1], got %v", index) } submatches := r.FindAllStringSubmatchIndex(testStr, 2) if len(submatches) != 2 { t.Errorf("expected length 2, got %d", len(submatches)) } expected := [2][4]int{{0, 1, 0, 1}, {8, 9, 8, 9}} if *(*[4]int)(submatches[0]) != expected[0] { t.Errorf("expected %v, got %v", expected[0], submatches[0]) } if *(*[4]int)(submatches[1]) != expected[1] { t.Errorf("expected %v, got %v", expected[0], submatches[0]) } } func TestSubexpIndex(t *testing.T) { r := pcre.MustCompile(`(?P\d)`) defer r.Close() index := r.SubexpIndex("number") if index != 1 { t.Errorf("expected 1, got %d", index) } index = r.SubexpIndex("string") if index != -1 { t.Errorf("expected -1, got %d", index) } num := r.NumSubexp() if num != 1 { t.Errorf("expected 1, got %d", num) } } func TestConcurrency(t *testing.T) { r := pcre.MustCompile(`\d*`) defer r.Close() wg := &sync.WaitGroup{} wg.Add(1) go func() { defer wg.Done() found := r.FindString("Test string 12345") if found != "12345" { t.Errorf("expected 12345, got %s", found) } }() wg.Add(1) go func() { defer wg.Done() matched := r.MatchString("Hello") if matched { t.Errorf("Expected Hello not to match") } }() wg.Add(1) go func() { defer wg.Done() matched := r.MatchString("54321") if !matched { t.Errorf("expected 54321 to match") } }() wg.Wait() } func TestString(t *testing.T) { const expr = `()` r := pcre.MustCompile(expr) defer r.Close() if r.String() != expr { t.Errorf("expected %s, got %s", expr, r.String()) } }