yapf_format: fix config lookup (python-lsp#473)

perrinjerome · gatesn · commit db16c239dad0 · 2019-01-19T13:05:15.000Z
diff --git a/pyls/plugins/yapf_format.py b/pyls/plugins/yapf_format.py
@@ -37,7 +37,7 @@ def _format(document, lines=None):
         lines=lines,
         filename=document.filename,
         style_config=file_resources.GetDefaultStyleForDir(
-            os.path.dirname(document.filename)
+            os.path.dirname(document.path)
         )
     )
 
diff --git a/test/plugins/test_yapf_format.py b/test/plugins/test_yapf_format.py
@@ -45,3 +45,14 @@ def test_range_format():
 def test_no_change():
     doc = Document(DOC_URI, GOOD_DOC)
     assert not pyls_format_document(doc)
+
+
+def test_config_file(tmpdir):
+    # a config file in the same directory as the source file will be used
+    conf = tmpdir.join('.style.yapf')
+    conf.write('[style]\ncolumn_limit = 14')
+    src = tmpdir.join('test.py')
+    doc = Document(uris.from_fs_path(src.strpath), DOC)
+
+    # A was split on multiple lines because of column_limit from config file
+    assert pyls_format_document(doc)[0]['newText'] == "A = [\n    'h', 'w',\n    'a'\n]\n\nB = ['h', 'w']\n"

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def _format(document, lines=None):`
`37`	`37`	`lines=lines,`
`38`	`38`	`filename=document.filename,`
`39`	`39`	`style_config=file_resources.GetDefaultStyleForDir(`
`40`		`- os.path.dirname(document.filename)`
	`40`	`+ os.path.dirname(document.path)`
`41`	`41`	`)`
`42`	`42`	`)`
`43`	`43`