tests for pep484 and pep526

Author: DonJayamanne

src/test/autocomplete/pep484.test.ts

@@ -0,0 +1,67 @@
+
+// Note: This example test is leveraging the Mocha test framework.
+// Please refer to their documentation on https://mochajs.org/ for help.
+
+
+// Place this right on top
+import { initialize, closeActiveWindows, setPythonExecutable } from '../initialize';
+// The module 'assert' provides assertion methods from node
+import * as assert from 'assert';
+// You can import and use all API from the 'vscode' module
+// as well as import your extension to test it
+import * as vscode from 'vscode';
+import * as path from 'path';
+import * as settings from '../../client/common/configSettings';
+import { execPythonFile } from '../../client/common/utils';
+import { createDeferred } from '../../client/common/helpers';
+
+let pythonSettings = settings.PythonSettings.getInstance();
+let disposable: vscode.Disposable;
+
+let autoCompPath = path.join(__dirname, '..', '..', '..', 'src', 'test', 'pythonFiles', 'autocomp');
+const filePep484 = path.join(autoCompPath, 'pep484.py');
+
+suite('Autocomplete PEP 484', () => {
+    const isPython3Deferred = createDeferred<boolean>();
+    const isPython3 = isPython3Deferred.promise;
+    suiteSetup(async () => {
+        disposable = setPythonExecutable(pythonSettings);
+        await initialize();
+        let version = await execPythonFile(pythonSettings.pythonPath, ['--version'], __dirname, true);
+        isPython3Deferred.resolve(version.indexOf('3.') >= 0);
+    });
+    suiteTeardown(done => {
+        disposable.dispose();
+        closeActiveWindows().then(() => done(), () => done());
+    });
+    teardown(done => {
+        closeActiveWindows().then(() => done(), () => done());
+    });
+
+    test('argument', async () => {
+        if (!await isPython3) {
+            return;
+        }
+        let textDocument = await vscode.workspace.openTextDocument(filePep484);
+        await vscode.window.showTextDocument(textDocument);
+        assert(vscode.window.activeTextEditor, 'No active editor');
+        const position = new vscode.Position(2, 27);
+        let list = await vscode.commands.executeCommand<vscode.CompletionList>('vscode.executeCompletionItemProvider', textDocument.uri, position);
+        assert.notEqual(list.items.filter(item => item.label === 'capitalize').length, 0, 'capitalize not found');
+        assert.notEqual(list.items.filter(item => item.label === 'upper').length, 0, 'upper not found');
+        assert.notEqual(list.items.filter(item => item.label === 'lower').length, 0, 'lower not found');
+    });
+
+    test('return value', async () => {
+        if (!await isPython3) {
+            return;
+        }
+        let textDocument = await vscode.workspace.openTextDocument(filePep484);
+        await vscode.window.showTextDocument(textDocument);
+        assert(vscode.window.activeTextEditor, 'No active editor');
+        const position = new vscode.Position(8, 6);
+        let list = await vscode.commands.executeCommand<vscode.CompletionList>('vscode.executeCompletionItemProvider', textDocument.uri, position);
+        assert.notEqual(list.items.filter(item => item.label === 'bit_length').length, 0, 'bit_length not found');
+        assert.notEqual(list.items.filter(item => item.label === 'from_bytes').length, 0, 'from_bytes not found');
+    });
+});

src/test/autocomplete/pep526.test.ts

@@ -0,0 +1,54 @@
+
+// Note: This example test is leveraging the Mocha test framework.
+// Please refer to their documentation on https://mochajs.org/ for help.
+
+
+// Place this right on top
+import { initialize, closeActiveWindows, setPythonExecutable } from '../initialize';
+// The module 'assert' provides assertion methods from node
+import * as assert from 'assert';
+// You can import and use all API from the 'vscode' module
+// as well as import your extension to test it
+import * as vscode from 'vscode';
+import * as path from 'path';
+import * as settings from '../../client/common/configSettings';
+import { execPythonFile } from '../../client/common/utils';
+import { createDeferred } from '../../client/common/helpers';
+
+let pythonSettings = settings.PythonSettings.getInstance();
+let disposable: vscode.Disposable;
+
+let autoCompPath = path.join(__dirname, '..', '..', '..', 'src', 'test', 'pythonFiles', 'autocomp');
+const filePep526 = path.join(autoCompPath, 'pep526.py');
+
+suite('Autocomplete PEP 526', () => {
+    const isPython3Deferred = createDeferred<boolean>();
+    const isPython3 = isPython3Deferred.promise;
+    suiteSetup(async () => {
+        disposable = setPythonExecutable(pythonSettings);
+        await initialize();
+        let version = await execPythonFile(pythonSettings.pythonPath, ['--version'], __dirname, true);
+        isPython3Deferred.resolve(version.indexOf('3.') >= 0);
+    });
+    suiteTeardown(done => {
+        disposable.dispose();
+        closeActiveWindows().then(() => done(), () => done());
+    });
+    teardown(done => {
+        closeActiveWindows().then(() => done(), () => done());
+    });
+
+    test('variable', async () => {
+        if (!await isPython3) {
+            return;
+        }
+        let textDocument = await vscode.workspace.openTextDocument(filePep526);
+        await vscode.window.showTextDocument(textDocument);
+        assert(vscode.window.activeTextEditor, 'No active editor');
+        const position = new vscode.Position(3, 8);
+        let list = await vscode.commands.executeCommand<vscode.CompletionList>('vscode.executeCompletionItemProvider', textDocument.uri, position);
+        assert.notEqual(list.items.filter(item => item.label === 'capitalize').length, 0, 'capitalize not found');
+        assert.notEqual(list.items.filter(item => item.label === 'upper').length, 0, 'upper not found');
+        assert.notEqual(list.items.filter(item => item.label === 'lower').length, 0, 'lower not found');
+    });
+});