Add unit test for mixed_line_ending --fix={cr,crlf}

tests/mixed_line_ending_test.py

@@ -48,3 +48,53 @@ def test_detect_mixed_line_ending(input_s, expected_retval, output, tmpdir):
 
     assert ret == expected_retval
     assert path.read() == output
+
+
+# Input, expected return value, expected output
+TESTS_FIX_FORCE_LF = (
+    # only 'LF'
+    (b'foo\nbar\nbaz\n', 1, b'foo\nbar\nbaz\n'),
+    # only 'CRLF'
+    (b'foo\r\nbar\r\nbaz\r\n', 1, b'foo\nbar\nbaz\n'),
+    # mixed with majority of 'LF'
+    (b'foo\r\nbar\nbaz\n', 1, b'foo\nbar\nbaz\n'),
+    # mixed with majority of 'CRLF'
+    (b'foo\r\nbar\nbaz\r\n', 1, b'foo\nbar\nbaz\n'),
+)
+
+
+@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'),
+                         TESTS_FIX_FORCE_LF)
+def test_mixed_line_ending_fix_force_lf(input_s, expected_retval, output,
+                                        tmpdir):
+    path = tmpdir.join('file.txt')
+    path.write(input_s)
+    ret = mixed_line_ending(('--fix=lf', '-vv', path.strpath))
+
+    assert ret == expected_retval
+    assert path.read() == output
+
+
+# Input, expected return value, expected output
+TESTS_FIX_FORCE_CRLF = (
+    # only 'LF'
+    (b'foo\nbar\nbaz\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
+    # only 'CRLF'
+    (b'foo\r\nbar\r\nbaz\r\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
+    # mixed with majority of 'LF'
+    (b'foo\r\nbar\nbaz\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
+    # mixed with majority of 'CRLF'
+    (b'foo\r\nbar\nbaz\r\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
+)
+
+
+@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'),
+                         TESTS_FIX_FORCE_CRLF)
+def test_mixed_line_ending_fix_force_crlf(input_s, expected_retval, output,
+                                          tmpdir):
+    path = tmpdir.join('file.txt')
+    path.write(input_s)
+    ret = mixed_line_ending(('--fix=crlf', '-vv', path.strpath))
+
+    assert ret == expected_retval
+    assert path.read() == output