Add multiple tests and fix them

diff --git a/extractor.go b/extractor.go
index ad83d6eacf648a8ef98136f9534a92f97bf04e44..678f009e50a5b7c92375f0924744625036efac20 100644 (file)
--- a/extractor.go
+++ b/extractor.go
@@ -9,7 +9,7 @@ type MatchOptions struct {
 }
 
 func pathExtractor(input string) [][][]byte {
-       surroundRegex := "[^]\\^\\+()!#[ \\t:'\"\\\\=]*"
+       surroundRegex := "[@~a-zA-Z/.]*"
        r := regexp.MustCompile("(" + surroundRegex + "[\\./]" + surroundRegex + ")")
        temp := [][][]byte{}
        temp = r.FindAllSubmatch([]byte(input), -1)

diff --git a/pe_test.go b/pe_test.go
index 719be92364388d95299b419e370c626b57fdd7c7..a9c15589ce9d89c5ae70ca007233509b274ea377 100644 (file)
--- a/pe_test.go
+++ b/pe_test.go
@@ -13,6 +13,26 @@ func TestEverything(t *testing.T) {
                t.Errorf("Matches sentence", output)
        }
 
+       output = GetAllMatches("'/usr/bin", MatchOptions{})
+       if output[0] != "/usr/bin" {
+               t.Errorf("Doesn't match statement correctly", output)
+       }
+
+       output = GetAllMatches("\"/usr/bin", MatchOptions{})
+       if output[0] != "/usr/bin" {
+               t.Errorf("Doesn't match statement correctly", output)
+       }
+
+       output = GetAllMatches("`/usr/bin", MatchOptions{})
+       if output[0] != "/usr/bin" {
+               t.Errorf("Doesn't match statement correctly", output)
+       }
+
+       output = GetAllMatches("€/usr/bin", MatchOptions{})
+       if output[0] != "/usr/bin" {
+               t.Errorf("Doesn't match statement correctly", output)
+       }
+
        output = GetAllMatches("prefix=/usr/bin", MatchOptions{})
        if len(output) != 1 {
                t.Errorf("Should match =/usr/bin", output)