From fd22bccbe8de5e7ef7010d29c3722cbd066dabb9 Mon Sep 17 00:00:00 2001
From: Devon Stewart <blast@hardchee.se>
Date: Fri, 15 Mar 2024 19:29:18 +0000
Subject: [PATCH] Adding formatting LSP tests

---
 tests/pyright_extended_tests/__init__.py | 52 ++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/tests/pyright_extended_tests/__init__.py b/tests/pyright_extended_tests/__init__.py
index 7f046ddb1..4dd7ff9a1 100644
--- a/tests/pyright_extended_tests/__init__.py
+++ b/tests/pyright_extended_tests/__init__.py
@@ -8,6 +8,8 @@
 from lsprotocol.types import (
     CompletionParams,
     CompletionList,
+    DocumentFormattingParams,
+    FormattingOptions,
     InitializeParams,
     Position,
     TextDocumentIdentifier,
@@ -127,3 +129,53 @@ async def test_linter(client: LanguageClient):
     assert (
         diagnostics[1].code == "ruff[F401]"
     ), f"Expected second warning to be unused import: {fileuri}"
+
+
+@pytest.mark.asyncio
+async def test_formatting(client: LanguageClient):
+    fileuri = f"file://{testroot}/cases/3.py"
+    contents = dedent(
+        """\
+    import os
+    x     =      "this is\\na test"
+    print(os.getenv(  "PATH",   x))
+    if True:
+                  pass
+    """
+    )
+    expected = dedent(
+        """\
+    import os
+
+    x = "this is\\na test"
+    print(os.getenv("PATH", x))
+    if True:
+        pass
+    """
+    )
+    print(repr(contents))
+    result = client.text_document_did_open(
+        DidOpenTextDocumentParams(
+            text_document=TextDocumentItem(
+                uri=fileuri,
+                language_id="python",
+                text=contents,
+                version=0,
+            )
+        )
+    )
+
+    result = await client.text_document_formatting_async(
+        params=DocumentFormattingParams(
+            text_document=TextDocumentIdentifier(uri=fileuri),
+            options=FormattingOptions(
+                tab_size=4,
+                insert_spaces=True,
+                trim_trailing_whitespace=True,
+            )
+        )
+    )
+
+    assert isinstance(result, list), result
+    assert len(result) > 0
+    assert result[0].new_text == expected