Add a test

Author: krassowski

Diff for: test/plugins/test_hover.py

@@ -10,7 +10,7 @@
 DOC_URI = uris.from_fs_path(__file__)
 DOC = """
 
-def main():
+def main(a: float, b: float):
     \"\"\"hello world\"\"\"
     pass
 """
@@ -79,13 +79,36 @@ def test_hover(workspace) -> None:
 
     doc = Document(DOC_URI, workspace, DOC)
 
-    contents = {"kind": "markdown", "value": "```python\nmain()\n```\n\n\nhello world"}
+    contents = {
+        "kind": "markdown",
+        "value": "```python\nmain(a: float, b: float)\n```\n\n\nhello world",
+    }
 
     assert {"contents": contents} == pylsp_hover(doc._config, doc, hov_position)
 
     assert {"contents": ""} == pylsp_hover(doc._config, doc, no_hov_position)
 
 
+def test_hover_custom_signature(workspace) -> None:
+    # Over 'main' in def main():
+    hov_position = {"line": 2, "character": 6}
+
+    doc = Document(DOC_URI, workspace, DOC)
+
+    contents = {
+        "kind": "markdown",
+        "value": "```python\nmain(\n    a: float,\n    b: float\n)\n```\n\n\nhello world",
+    }
+
+    def signatures_to_markdown(signatures: list):
+        # dummy implementation for tests
+        return "```python\nmain(\n    a: float,\n    b: float\n)\n```\n"
+
+    assert {"contents": contents} == pylsp_hover(
+        doc._config, doc, hov_position, signatures_to_markdown=signatures_to_markdown
+    )
+
+
 def test_document_path_hover(workspace_other_root_path, tmpdir) -> None:
     # Create a dummy module out of the workspace's root_path and try to get
     # a definition on it in another file placed next to it.