Escape filenames which looks like regexp

codingjoe · Oct 24, 2019 · c670676 · c670676
1 parent b63fb2c
commit c670676
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/relint.py b/relint.py
@@ -188,7 +188,7 @@ def main():
     paths = {
         path
         for file in args.files
-        for path in glob.iglob(file, recursive=True)
+        for path in glob.iglob(glob.escape(file), recursive=True)
     }
 
     tests = list(load_config(args.config))

diff --git a/test_relint.py b/test_relint.py
@@ -8,8 +8,9 @@
 
 
 class TestMain:
-    def test_main_execution(self, mocker):
-        mocker.patch.object(sys, 'argv', ['relint.py', 'test_relint.py'])
+    @pytest.mark.parametrize('filename', ['test_relint.py', '[a-b].py', '[b-a].py'])
+    def test_main_execution(self, mocker, filename):
+        mocker.patch.object(sys, 'argv', ['relint.py', filename])
 
         with pytest.raises(SystemExit) as exc_info:
             main()