Test that we preserve line endings when formatting CRLF documents

ccordoba12 · ccordoba12 · commit 0c82bf8a2053 · 2022-03-14T13:01:44.000-05:00
diff --git a/tests/fixtures/formatted-crlf.py b/tests/fixtures/formatted-crlf.py
@@ -0,0 +1,4 @@
+if True:
+    print("foo")
+
+print("bar")  # noqa
diff --git a/tests/fixtures/unformatted-crlf.py b/tests/fixtures/unformatted-crlf.py
@@ -0,0 +1,4 @@
+if True:
+    print("foo")
+
+print("bar")  # noqa
diff --git a/tests/test_plugin.py b/tests/test_plugin.py
@@ -39,6 +39,15 @@ def unformatted_pyi_document(workspace):
     return Document(uri, workspace)
 
 
+@pytest.fixture
+def unformatted_crlf_document(workspace):
+    path = fixtures_dir / "unformatted-crlf.py"
+    uri = f"file:/{path}"
+    with open(path, "r", newline="") as f:
+        source = f.read()
+    return Document(uri, workspace, source=source)
+
+
 @pytest.fixture
 def formatted_document(workspace):
     path = fixtures_dir / "formatted.txt"
@@ -53,6 +62,15 @@ def formatted_pyi_document(workspace):
     return Document(uri, workspace)
 
 
+@pytest.fixture
+def formatted_crlf_document(workspace):
+    path = fixtures_dir / "formatted-crlf.py"
+    uri = f"file:/{path}"
+    with open(path, "r", newline="") as f:
+        source = f.read()
+    return Document(uri, workspace, source=source)
+
+
 @pytest.fixture
 def invalid_document(workspace):
     path = fixtures_dir / "invalid.txt"
@@ -225,3 +243,17 @@ def test_entry_point():
 
     module = entry_point.load()
     assert isinstance(module, types.ModuleType)
+
+
+def test_pylsp_format_crlf_document(unformatted_crlf_document, formatted_crlf_document):
+    result = pylsp_format_document(unformatted_crlf_document)
+
+    assert result == [
+        {
+            "range": {
+                "start": {"line": 0, "character": 0},
+                "end": {"line": 4, "character": 0},
+            },
+            "newText": formatted_crlf_document.source,
+        }
+    ]
