completed code_completion test

Author: ch0mler

test/extension.test.ts

@@ -8,7 +8,7 @@ Please refer to their documentation on https://mochajs.org/ for help.
 import * as path from "path";
 import * as vscode from "vscode";
 import * as yara from "../yara/src/extension";
-import {YaraCompletionItemProvider} from "../yara/src/modules";
+import {YaraCompletionItemProvider} from "../yara/src/completionProvider";
 
 let workspace = path.join(__dirname, "..", "..", "test/rules/");
 
@@ -146,14 +146,20 @@ suite("YARA: Provider", function () {
     });
 
     test("code completion", function (done) {
-        const filepath: string = path.join(workspace, "peek_rules.yara");
+        const filepath: string = path.join(workspace, "code_completion.yara");
         vscode.workspace.openTextDocument(filepath).then(function (doc) {
             const ccProvider: YaraCompletionItemProvider = new YaraCompletionItemProvider();
-            // ModuleCompletionExample: Line 53, Col 12
-            let pos: vscode.Position = new vscode.Position(52, 12);
+            // "cuckoo.": Line 8, Col 12
+            let pos: vscode.Position = new vscode.Position(7, 12);
             let tokenSource: vscode.CancellationTokenSource = new vscode.CancellationTokenSource();
-            let items = ccProvider.provideCompletionItems(doc, pos, tokenSource.token, undefined);
-            console.log(items);
+            ccProvider.provideCompletionItems(doc, pos, tokenSource.token, undefined).then(function (items) {
+                if (items[0].label == "network" || items[0].kind == vscode.CompletionItemKind.Class &&
+                    items[1].label == "registry" || items[1].kind == vscode.CompletionItemKind.Class &&
+                    items[2].label == "filesystem" || items[2].kind == vscode.CompletionItemKind.Class &&
+                    items[3].label == "sync" || items[3].kind == vscode.CompletionItemKind.Class) {
+                        done();
+                    }
+            });
         });
     });
 

test/rules/code_completion.yara

@@ -0,0 +1,9 @@
+rule ModuleCompletionExample
+{
+    meta:
+        description = "Module Completion Example"
+        author = "Test"
+        reference = "https://infosec-intern.github.io"
+    condition:
+        cuckoo.
+}

test/rules/peek_rules.yara

@@ -42,13 +42,3 @@ rule RuleReferenceExample
     condition:
         SyntaxExample and $hex_string
 }
-
-rule ModuleCompletionExample
-{
-    meta:
-        description = "Module Completion Example"
-        author = "Test"
-        reference = "https://infosec-intern.github.io"
-    condition:
-        cuckoo.
-}

yara/src/completionProvider.ts

@@ -45,7 +45,7 @@ function parseSchema(symbols: Array<string>, schema: Object, depth: number): nul
 export class YaraCompletionItemProvider implements vscode.CompletionItemProvider {
     public provideCompletionItems(doc: vscode.TextDocument, pos: vscode.Position, token: vscode.CancellationToken, context: vscode.CompletionContext): vscode.ProviderResult<vscode.CompletionItem[] | vscode.CompletionList> {
         return new Promise((resolve, reject) => {
-            if (context != undefined && context.triggerCharacter == ".") {
+            if (context == undefined || context.triggerCharacter == ".") {
                 let items: vscode.CompletionItem[] = Array<vscode.CompletionItem>();
                 let fields: any = modules.get(doc, pos);
                 if (fields != null) {