Merge pull request #4 from pkasprzyk/windows-compat

compatibility with Windows newline format

Author: HaaLeo

pylintfileheader/file_header_checker.py

@@ -2,7 +2,6 @@
 #  Copyright (c) Leo Hanisch. All rights reserved.
 #  Licensed under the MIT License. See LICENSE.txt in the project root for license information.
 # ---------------------------------------------------------------------------------------------
-
 import re
 import sys
 
@@ -65,8 +64,8 @@ def __init__(self, linter=None):
     def open(self):
         self.header = self.config.file_header
         if not self.header and self.config.file_header_path:
-            with open(self.config.file_header_path, 'r') as header_file:
-                self.header = header_file.read()
+            with open(self.config.file_header_path, 'rb') as header_file:
+                self.header = header_file.read().decode('utf-8')
 
         if self.header:
             if sys.version_info[0] < 3:

pylintfileheadertest/file_header_checker_test.py

@@ -16,13 +16,14 @@
 
 class TestFileHeaderChecker(pylint.testutils.CheckerTestCase):
     CHECKER_CLASS = FileHeaderChecker
-    CONFIG = {'file_header': '# Valid\n# Header'}
+    EXPECTED_HEADER = '# Valid\n# Header'
+    CONFIG = {'file_header': EXPECTED_HEADER}
 
     def test_valid_header_no_message_added(self):
         """Test whether no message is added, when the file header is valid."""
 
         node_mock = MagicMock()
-        node_mock.stream.return_value.__enter__.return_value.read.return_value.decode.return_value = '# Valid\n# Header'
+        node_mock.stream.return_value.__enter__.return_value.read.return_value.decode.return_value = self.EXPECTED_HEADER
         with self.assertNoMessages():
             self.checker.process_module(node_mock)
 
@@ -34,7 +35,7 @@ def test_invalid_header_message_added(self):
         with self.assertAddsMessages(pylint.testutils.Message(
                 msg_id='invalid-file-header',
                 line=1,
-                args='# Valid\n# Header')):
+                args=self.EXPECTED_HEADER)):
             self.checker.process_module(node_mock)
 
     def test_valid_header_not_at_top_message_added(self):
@@ -45,7 +46,7 @@ def test_valid_header_not_at_top_message_added(self):
         with self.assertAddsMessages(pylint.testutils.Message(
                 msg_id='invalid-file-header',
                 line=1,
-                args='# Valid\n# Header')):
+                args=self.EXPECTED_HEADER)):
             self.checker.process_module(node_mock)
 
     def test_ignore_empty_files(self):
@@ -65,7 +66,7 @@ def test_do_not_ignore_empty_files(self):
         with self.assertAddsMessages(pylint.testutils.Message(
                 msg_id='invalid-file-header',
                 line=1,
-                args='# Valid\n# Header')):
+                args=self.EXPECTED_HEADER)):
             self.checker.process_module(node_mock)
 
 
@@ -87,6 +88,12 @@ class TestFileHeaderCheckerPathMain(TestFileHeaderChecker):
     CONFIG = {'file_header_path': 'pylintfileheadertest/header.txt'}
 
 
+class TestFileHeaderCheckerPathWindows(TestFileHeaderChecker):
+    CHECKER_CLASS = FileHeaderChecker
+    EXPECTED_HEADER = '# Valid\r\n# Header'
+    CONFIG = {'file_header_path': 'pylintfileheadertest/windows_header.txt'}
+
+
 class TestFileHeaderCheckerPathExtra:
     CHECKER_CLASS = FileHeaderChecker
 

pylintfileheadertest/windows_header.txt

@@ -0,0 +1,2 @@
+# Valid
+# Header