Fix type hints not being provided for async functions.

Author: njqdev

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Change Log
 
+## [1.1.1] - 2020-04-22
+
+* Fixed type hints not being provided for async functions.
+
 ## [1.1.0] - 2020-04-19
 
 ### Improvements:

package.json

@@ -1,7 +1,7 @@
 {
 	"name": "vscode-python-typehint",
     "displayName": "Python Type Hint",
-    "version": "1.1.0",
+    "version": "1.1.1",
     "publisher": "njqdev",
     "description": "Type hint completion for Python.",
     "icon": "images/icon.png",
@@ -46,7 +46,7 @@
                 "workspace.searchEnabled": {
                     "type": "boolean",
                     "default": true,
-                    "description": "Search other files in the workspace when estimating types for a parameter. Disabling this will increase type hinting speed."
+                    "description": "If enabled, other files in the workspace are searched when estimating types for a parameter. Disabling this will increase performance."
                   },
               "workspace.searchLimit": {
                 "type": "number",

src/completionProvider.ts

@@ -150,14 +150,14 @@ export class ParamHintCompletionProvider extends CompletionProvider implements C
     private shouldProvideItems(precedingText: string, activePos: Position, doc: TextDocument): boolean {
 
         if (activePos.character > 0 && !/#/.test(precedingText)) {
-            let provide = /^[ \t]*def /.test(precedingText);
+            let provide = /^[ \t]*(def |async *def )/.test(precedingText);
 
             if (!provide) {
                 const nLinesToCheck = activePos.line > 4 ? 4 : activePos.line;
                 const previousLines = doc.getText(
                     new Range(doc.lineAt(activePos.line - nLinesToCheck).range.start, activePos)
                 );
-                provide = new RegExp(`^[ \t]*def(?![\\s\\S]+(\\):|-> *${simpleIdentifier}:))`, "m").test(previousLines);
+                provide = new RegExp(`^[ \t]*(async *)?def(?![\\s\\S]+(\\):|-> *${simpleIdentifier}:))`, "m").test(previousLines);
             }
             return provide;
         }

test/providers/completionProvider.test.ts

@@ -51,6 +51,20 @@ suite('ParamHintCompletionProvider', () => {
         let actual = await providerResult(provider, data);
         assert.notEqual(actual, null);
     });
+
+    test("provides items for async function", async () => {
+        let data = "async def test(test:";
+        let pos = new vsc.Position(0, data.length);
+        let expected = null;
+        let actual = await provideCompletionItems(provider, data, pos);
+        assert.notEqual(actual, null, messageFor({ data, expected }, actual));
+
+        let line2 = "        test:";
+        data = "async def test(\n" + line2;
+        pos = new vsc.Position(1, line2.length);
+        actual = await provideCompletionItems(provider, data, pos);
+        assert.notEqual(actual, null, messageFor({ data, expected }, actual));
+    });
 
     test("does not provide items unless a function def is detected", async () => {
         let text = " :";
@@ -116,7 +130,7 @@ async function providerResult(
     functionText: string,
     trailingText?: string
 ): Promise<vsc.CompletionList | null> {
-    let content = `def func(${functionText}`;
+    let content = `    def func(${functionText}`;
     const lines: string[] = content.split("\n");
     const lastLineIdx = lines.length - 1;
     const lastPos = new vsc.Position(lastLineIdx, lines[lastLineIdx].length);