Escape filenames which looks like regexp

Author: JanBednarik

Diff for: 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 for: setup.cfg

@@ -29,6 +29,7 @@ setup_requires =
 tests_require =
     pytest
     pytest-cov
+    pytest-mock
 py_modules = relint
 
 [options.entry_points]

Diff for: test_relint.py

@@ -1,3 +1,5 @@
+import difflib
+import io
 import sys
 
 import pytest
@@ -7,18 +9,32 @@
 
 
 class TestMain:
-    def test_main_execution(self):
-        sys.argv.extend(['relint.py', 'test_relint.py'])
+    @pytest.mark.parametrize('filename', ['test_relint.py', '[a-b].py', '[b-a].py'])
+    def test_main_execution(self, filename, mocker):
+        mocker.patch.object(sys, 'argv', ['relint.py', filename])
+
         with pytest.raises(SystemExit) as exc_info:
             main()
 
         assert exc_info.value.code == 0
 
-    def test_main_execution_with_diff(self, capsys):
-        sys.argv.extend(['relint.py', 'test_relint.py', '--diff'])
-        sys.stdin = open('test.diff')
-        with pytest.raises(SystemExit) as exc_info:
-            main()
+    def test_main_execution_with_diff(self, capsys, mocker, tmpdir):
+        old_content = ["# dummy test file\n", "pass"]
+        new_content = ["# dummy test file\n", "# TODO do something\n", "pass"]
+        tmpdir.join('dummy.py').write(''.join(new_content))
+
+        diff = io.StringIO()
+        diff.write("diff --git a/dummy.py b/dummy.py\n")
+        diff.writelines(difflib.unified_diff(old_content, new_content))
+        diff.seek(0)
+
+        mocker.patch.object(sys, 'argv', ['relint.py', 'dummy.py', '--diff'])
+        mocker.patch.object(sys, 'stdin', diff)
+
+        tmpdir.join('.relint.yml').write(open('.relint.yml').read())
+        with tmpdir.as_cwd():
+            with pytest.raises(SystemExit) as exc_info:
+                main()
 
         expected_message = 'Hint: Get it done right away!'