@zzzen/pyright-internal 1.2.0-dev.20230305 → 1.2.0-dev.20230319

package/dist/parser/tokenizer.js

@@ -633,7 +633,7 @@ class Tokenizer {
                 if (!isNaN(intValue)) {
                     const bigIntValue = BigInt(simpleIntText);
                     if (!isFinite(intValue) ||
-                        intValue < Number.MAX_SAFE_INTEGER ||
+                        intValue < Number.MIN_SAFE_INTEGER ||
                         intValue > Number.MAX_SAFE_INTEGER) {
                         intValue = bigIntValue;
                     }

package/dist/server.d.ts

@@ -6,11 +6,12 @@ import { FileSystem } from './common/fileSystem';
 import { FullAccessHost } from './common/fullAccessHost';
 import { Host } from './common/host';
 import { ProgressReporter } from './common/progressReporter';
-import { LanguageServerBase, ServerSettings, WorkspaceServiceInstance } from './languageServerBase';
+import { LanguageServerBase, ServerSettings } from './languageServerBase';
+import { Workspace } from './workspaceFactory';
 export declare class PyrightServer extends LanguageServerBase {
     private _controller;
     constructor(connection: Connection);
-    getSettings(workspace: WorkspaceServiceInstance): Promise<ServerSettings>;
+    getSettings(workspace: Workspace): Promise<ServerSettings>;
     createBackgroundAnalysis(serviceId: string): BackgroundAnalysisBase | undefined;
     protected createHost(): FullAccessHost;
     protected createImportResolver(fs: FileSystem, options: ConfigOptions, host: Host): ImportResolver;

package/dist/server.js

@@ -22,7 +22,6 @@ const pathUtils_1 = require("./common/pathUtils");
 const realFileSystem_1 = require("./common/realFileSystem");
 const languageServerBase_1 = require("./languageServerBase");
 const codeActionProvider_1 = require("./languageService/codeActionProvider");
-const workspaceMap_1 = require("./workspaceMap");
 const maxAnalysisTimeInForeground = { openFilesTimeInMs: 50, noOpenFilesTimeInMs: 200 };
 class PyrightServer extends languageServerBase_1.LanguageServerBase {
     constructor(connection) {
@@ -33,14 +32,12 @@ class PyrightServer extends languageServerBase_1.LanguageServerBase {
         // be __dirname.
         const rootDirectory = global.__rootDirectory || __dirname;
         const console = new console_1.ConsoleWithLogLevel(connection.console);
-        const workspaceMap = new workspaceMap_1.WorkspaceMap();
         const fileWatcherProvider = new realFileSystem_1.WorkspaceFileWatcherProvider();
         const fileSystem = (0, realFileSystem_1.createFromRealFileSystem)(console, fileWatcherProvider);
         super({
             productName: 'Pyright',
             rootDirectory,
             version,
-            workspaceMap,
             fileSystem,
             fileWatcherHandler: fileWatcherProvider,
             cancellationProvider: new fileBasedCancellationUtils_1.FileBasedCancellationProvider('bg'),
@@ -108,7 +105,7 @@ class PyrightServer extends languageServerBase_1.LanguageServerBase {
                 if (pythonAnalysisSection.useLibraryCodeForTypes !== undefined) {
                     serverSettings.useLibraryCodeForTypes = !!pythonAnalysisSection.useLibraryCodeForTypes;
                 }
-                serverSettings.logLevel = this.convertLogLevel(pythonAnalysisSection.logLevel);
+                serverSettings.logLevel = (0, console_1.convertLogLevel)(pythonAnalysisSection.logLevel);
                 serverSettings.autoSearchPaths = !!pythonAnalysisSection.autoSearchPaths;
                 const extraPaths = pythonAnalysisSection.extraPaths;
                 if (extraPaths && Array.isArray(extraPaths) && extraPaths.length > 0) {

package/dist/server.js.map

@@ -1 +1 @@
-{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,iEAS+B;AAG/B,8DAA2D;AAC3D,gEAA4D;AAC5D,6DAA0D;AAE1D,oEAAiE;AACjE,kEAAuE;AACvE,0DAA6E;AAC7E,8CAAiE;AACjE,wCAAsD;AACtD,sDAA2D;AAC3D,oFAAoF;AAEpF,4DAAyD;AAEzD,kDAAkD;AAElD,4DAAiG;AACjG,6DAAoG;AACpG,6EAA0E;AAC1E,iDAA8C;AAE9C,MAAM,2BAA2B,GAAG,EAAE,iBAAiB,EAAE,EAAE,EAAE,mBAAmB,EAAE,GAAG,EAAE,CAAC;AAExF,MAAa,aAAc,SAAQ,uCAAkB;IAGjD,YAAY,UAAsB;QAC9B,8DAA8D;QAC9D,MAAM,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAEzD,0EAA0E;QAC1E,6EAA6E;QAC7E,gBAAgB;QAChB,MAAM,aAAa,GAAI,MAAc,CAAC,eAAe,IAAI,SAAS,CAAC;QAEnE,MAAM,OAAO,GAAG,IAAI,6BAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC5D,MAAM,YAAY,GAAG,IAAI,2BAAY,EAAE,CAAC;QACxC,MAAM,mBAAmB,GAAG,IAAI,6CAA4B,EAAE,CAAC;QAC/D,MAAM,UAAU,GAAG,IAAA,yCAAwB,EAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QAE1E,KAAK,CACD;YACI,WAAW,EAAE,SAAS;YACtB,aAAa;YACb,OAAO;YACP,YAAY;YACZ,UAAU;YACV,kBAAkB,EAAE,mBAAmB;YACvC,oBAAoB,EAAE,IAAI,0DAA6B,CAAC,IAAI,CAAC;YAC7D,2BAA2B;YAC3B,oBAAoB,EAAE,CAAC,sCAAc,CAAC,QAAQ,EAAE,sCAAc,CAAC,qBAAqB,CAAC;SACxF,EACD,UAAU,EACV,OAAO,CACV,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,IAAI,qCAAiB,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,SAAmC;QACjD,MAAM,cAAc,GAAmB;YACnC,qBAAqB,EAAE,IAAI;YAC3B,sBAAsB,EAAE,IAAI;YAC5B,qBAAqB,EAAE,IAAI;YAC3B,aAAa,EAAE,IAAI;YACnB,sBAAsB,EAAE,KAAK;YAC7B,uBAAuB,EAAE,KAAK;YAC9B,sBAAsB,EAAE,KAAK;YAC7B,gBAAgB,EAAE,OAAO;YACzB,2BAA2B,EAAE,EAAE;YAC/B,QAAQ,EAAE,kBAAQ,CAAC,IAAI;YACvB,qBAAqB,EAAE,IAAI;YAC3B,wBAAwB,EAAE,oCAAoB,CAAC,SAAS;SAC3D,CAAC;QAEF,IAAI;YACA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC3E,IAAI,aAAa,EAAE;gBACf,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC;gBAC5C,IAAI,UAAU,IAAI,IAAA,eAAQ,EAAC,UAAU,CAAC,IAAI,CAAC,IAAA,gCAAc,EAAC,UAAU,CAAC,EAAE;oBACnE,cAAc,CAAC,UAAU,GAAG,IAAA,wBAAY,EACpC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,CACtD,CAAC;iBACL;gBAED,MAAM,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC;gBAExC,IAAI,QAAQ,IAAI,IAAA,eAAQ,EAAC,QAAQ,CAAC,EAAE;oBAChC,cAAc,CAAC,QAAQ,GAAG,IAAA,wBAAY,EAClC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACpD,CAAC;iBACL;aACJ;YAED,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;YAC5F,IAAI,qBAAqB,EAAE;gBACvB,MAAM,aAAa,GAAG,qBAAqB,CAAC,aAAa,CAAC;gBAC1D,IAAI,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;oBAC3E,MAAM,YAAY,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;oBACtC,IAAI,YAAY,IAAI,IAAA,eAAQ,EAAC,YAAY,CAAC,EAAE;wBACxC,cAAc,CAAC,YAAY,GAAG,IAAA,wBAAY,EACtC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,YAAY,CAAC,CACxD,CAAC;qBACL;iBACJ;gBAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC;gBAChD,IAAI,QAAQ,IAAI,IAAA,eAAQ,EAAC,QAAQ,CAAC,EAAE;oBAChC,cAAc,CAAC,QAAQ,GAAG,IAAA,wBAAY,EAClC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACpD,CAAC;iBACL;gBAED,MAAM,2BAA2B,GAAG,qBAAqB,CAAC,2BAA2B,CAAC;gBACtF,IAAI,2BAA2B,EAAE;oBAC7B,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,2BAA2B,CAAC,EAAE;wBACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;wBAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAe,CAAC,CAAC;wBAC5D,IAAI,QAAQ,IAAI,QAAQ,EAAE;4BACtB,cAAc,CAAC,2BAA4B,CAAC,QAAQ,CAAC,GAAG,QAAS,CAAC;yBACrE;qBACJ;iBACJ;gBAED,IAAI,qBAAqB,CAAC,cAAc,KAAK,SAAS,EAAE;oBACpD,cAAc,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;iBAC7F;qBAAM,IAAI,qBAAqB,CAAC,aAAa,KAAK,SAAS,EAAE;oBAC1D,cAAc,CAAC,aAAa,GAAG,CAAC,CAAC,qBAAqB,CAAC,aAAa,CAAC;iBACxE;gBAED,IAAI,qBAAqB,CAAC,sBAAsB,KAAK,SAAS,EAAE;oBAC5D,cAAc,CAAC,sBAAsB,GAAG,CAAC,CAAC,qBAAqB,CAAC,sBAAsB,CAAC;iBAC1F;gBAED,cAAc,CAAC,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;gBAC/E,cAAc,CAAC,eAAe,GAAG,CAAC,CAAC,qBAAqB,CAAC,eAAe,CAAC;gBAEzE,MAAM,UAAU,GAAG,qBAAqB,CAAC,UAAU,CAAC;gBACpD,IAAI,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;oBAClE,cAAc,CAAC,UAAU,GAAG,UAAU;yBACjC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAA,eAAQ,EAAC,CAAC,CAAC,CAAC;yBAC/B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,wBAAY,EAAC,SAAS,CAAC,QAAQ,EAAE,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;iBACjG;gBAED,IAAI,qBAAqB,CAAC,gBAAgB,KAAK,SAAS,EAAE;oBACtD,cAAc,CAAC,gBAAgB,GAAG,qBAAqB,CAAC,gBAAgB,CAAC;iBAC5E;gBAED,IAAI,qBAAqB,CAAC,qBAAqB,KAAK,SAAS,EAAE;oBAC3D,cAAc,CAAC,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC;iBACtF;gBAED,IACI,cAAc,CAAC,QAAQ,KAAK,kBAAQ,CAAC,GAAG;oBACxC,qBAAqB,CAAC,qBAAqB,KAAK,SAAS,EAC3D;oBACE,cAAc,CAAC,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC;iBACtF;gBAED,IAAI,qBAAqB,CAAC,2BAA2B,KAAK,SAAS,EAAE;oBACjE,cAAc,CAAC,2BAA2B,GAAG,qBAAqB,CAAC,2BAA2B,CAAC;iBAClG;aACJ;iBAAM;gBACH,cAAc,CAAC,eAAe,GAAG,IAAI,CAAC;aACzC;YAED,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YAC7E,IAAI,cAAc,EAAE;gBAChB,IAAI,cAAc,CAAC,aAAa,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,aAAa,GAAG,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC;iBACjE;gBAED,IAAI,cAAc,CAAC,sBAAsB,KAAK,SAAS,EAAE;oBACrD,cAAc,CAAC,sBAAsB,GAAG,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC;iBACnF;gBAED,cAAc,CAAC,uBAAuB,GAAG,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC;gBAClF,cAAc,CAAC,sBAAsB,GAAG,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC;gBAEhF,MAAM,gBAAgB,GAAG,cAAc,CAAC,gBAAgB,CAAC;gBACzD,IAAI,gBAAgB,IAAI,IAAA,eAAQ,EAAC,gBAAgB,CAAC,EAAE;oBAChD,cAAc,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;iBACtD;aACJ;SACJ;QAAC,OAAO,KAAK,EAAE;YACZ,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,2BAA2B,KAAK,EAAE,CAAC,CAAC;SAC1D;QACD,OAAO,cAAc,CAAC;IAC1B,CAAC;IAED,wBAAwB,CAAC,SAAiB;QACtC,IAAI,IAAA,kBAAW,GAAE,IAAI,CAAC,IAAA,6CAAyB,GAAE,EAAE;YAC/C,uEAAuE;YACvE,+CAA+C;YAC/C,OAAO,SAAS,CAAC;SACpB;QAED,OAAO,IAAI,uCAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC;IAEkB,UAAU;QACzB,OAAO,IAAI,+BAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC;IAEkB,oBAAoB,CAAC,EAAc,EAAE,OAAsB,EAAE,IAAU;QACtF,MAAM,cAAc,GAAG,IAAI,+BAAc,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;QAE7D,qEAAqE;QACrE,wCAAwC;QACxC,cAAc,CAAC,eAAe,EAAE,CAAC;QAEjC,OAAO,cAAc,CAAC;IAC1B,CAAC;IAES,cAAc,CAAC,MAA4B,EAAE,KAAwB;QAC3E,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;IAES,oBAAoB,CAAC,OAAe;QAC1C,OAAO,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC;IAES,KAAK,CAAC,iBAAiB,CAC7B,MAAwB,EACxB,KAAwB;QAExB,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAEjC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAChF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC3D,OAAO,uCAAkB,CAAC,yBAAyB,CAC/C,SAAS,EACT,QAAQ,EACR,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,OAAO,CAAC,IAAI,EACnB,KAAK,CACR,CAAC;IACN,CAAC;IAES,sBAAsB;QAC5B,2EAA2E;QAC3E,kDAAkD;QAElD,IAAI,gBAAqE,CAAC;QAC1E,OAAO;YACH,SAAS,EAAE,CAAC,IAAqB,EAAE,EAAE,CAAC,IAAI;YAC1C,KAAK,EAAE,GAAG,EAAE;gBACR,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,EAAE;oBACzC,gBAAgB,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAC;oBACpE,gBAAgB;yBACX,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACf,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACvB,CAAC,CAAC;yBACD,YAAY,EAAE,CAAC;iBACvB;qBAAM;oBACH,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC;iBAC9D;YACL,CAAC;YACD,MAAM,EAAE,CAAC,OAAe,EAAE,EAAE;gBACxB,IAAI,gBAAgB,EAAE;oBAClB,gBAAgB;yBACX,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACf,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBAC7B,CAAC,CAAC;yBACD,YAAY,EAAE,CAAC;iBACvB;qBAAM;oBACH,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC;iBACxE;YACL,CAAC;YACD,GAAG,EAAE,GAAG,EAAE;gBACN,IAAI,gBAAgB,EAAE;oBAClB,gBAAgB;yBACX,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACf,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACpB,CAAC,CAAC;yBACD,YAAY,EAAE,CAAC;oBACpB,gBAAgB,GAAG,SAAS,CAAC;iBAChC;qBAAM;oBACH,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;iBAC5D;YACL,CAAC;SACJ,CAAC;IACN,CAAC;CACJ;AAxQD,sCAwQC"}
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,iEAS+B;AAG/B,8DAA2D;AAC3D,gEAA4D;AAC5D,6DAA0D;AAE1D,oEAAiE;AACjE,kEAAuE;AACvE,0DAA6E;AAC7E,8CAAkF;AAClF,wCAAsD;AACtD,sDAA2D;AAC3D,oFAAoF;AAEpF,4DAAyD;AAEzD,kDAAkD;AAElD,4DAAiG;AACjG,6DAA0E;AAC1E,6EAA0E;AAG1E,MAAM,2BAA2B,GAAG,EAAE,iBAAiB,EAAE,EAAE,EAAE,mBAAmB,EAAE,GAAG,EAAE,CAAC;AAExF,MAAa,aAAc,SAAQ,uCAAkB;IAGjD,YAAY,UAAsB;QAC9B,8DAA8D;QAC9D,MAAM,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAEzD,0EAA0E;QAC1E,6EAA6E;QAC7E,gBAAgB;QAChB,MAAM,aAAa,GAAI,MAAc,CAAC,eAAe,IAAI,SAAS,CAAC;QAEnE,MAAM,OAAO,GAAG,IAAI,6BAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC5D,MAAM,mBAAmB,GAAG,IAAI,6CAA4B,EAAE,CAAC;QAC/D,MAAM,UAAU,GAAG,IAAA,yCAAwB,EAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QAE1E,KAAK,CACD;YACI,WAAW,EAAE,SAAS;YACtB,aAAa;YACb,OAAO;YACP,UAAU;YACV,kBAAkB,EAAE,mBAAmB;YACvC,oBAAoB,EAAE,IAAI,0DAA6B,CAAC,IAAI,CAAC;YAC7D,2BAA2B;YAC3B,oBAAoB,EAAE,CAAC,sCAAc,CAAC,QAAQ,EAAE,sCAAc,CAAC,qBAAqB,CAAC;SACxF,EACD,UAAU,EACV,OAAO,CACV,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,IAAI,qCAAiB,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,SAAoB;QAClC,MAAM,cAAc,GAAmB;YACnC,qBAAqB,EAAE,IAAI;YAC3B,sBAAsB,EAAE,IAAI;YAC5B,qBAAqB,EAAE,IAAI;YAC3B,aAAa,EAAE,IAAI;YACnB,sBAAsB,EAAE,KAAK;YAC7B,uBAAuB,EAAE,KAAK;YAC9B,sBAAsB,EAAE,KAAK;YAC7B,gBAAgB,EAAE,OAAO;YACzB,2BAA2B,EAAE,EAAE;YAC/B,QAAQ,EAAE,kBAAQ,CAAC,IAAI;YACvB,qBAAqB,EAAE,IAAI;YAC3B,wBAAwB,EAAE,oCAAoB,CAAC,SAAS;SAC3D,CAAC;QAEF,IAAI;YACA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC3E,IAAI,aAAa,EAAE;gBACf,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC;gBAC5C,IAAI,UAAU,IAAI,IAAA,eAAQ,EAAC,UAAU,CAAC,IAAI,CAAC,IAAA,gCAAc,EAAC,UAAU,CAAC,EAAE;oBACnE,cAAc,CAAC,UAAU,GAAG,IAAA,wBAAY,EACpC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,CACtD,CAAC;iBACL;gBAED,MAAM,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC;gBAExC,IAAI,QAAQ,IAAI,IAAA,eAAQ,EAAC,QAAQ,CAAC,EAAE;oBAChC,cAAc,CAAC,QAAQ,GAAG,IAAA,wBAAY,EAClC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACpD,CAAC;iBACL;aACJ;YAED,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;YAC5F,IAAI,qBAAqB,EAAE;gBACvB,MAAM,aAAa,GAAG,qBAAqB,CAAC,aAAa,CAAC;gBAC1D,IAAI,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;oBAC3E,MAAM,YAAY,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;oBACtC,IAAI,YAAY,IAAI,IAAA,eAAQ,EAAC,YAAY,CAAC,EAAE;wBACxC,cAAc,CAAC,YAAY,GAAG,IAAA,wBAAY,EACtC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,YAAY,CAAC,CACxD,CAAC;qBACL;iBACJ;gBAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC;gBAChD,IAAI,QAAQ,IAAI,IAAA,eAAQ,EAAC,QAAQ,CAAC,EAAE;oBAChC,cAAc,CAAC,QAAQ,GAAG,IAAA,wBAAY,EAClC,SAAS,CAAC,QAAQ,EAClB,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACpD,CAAC;iBACL;gBAED,MAAM,2BAA2B,GAAG,qBAAqB,CAAC,2BAA2B,CAAC;gBACtF,IAAI,2BAA2B,EAAE;oBAC7B,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,2BAA2B,CAAC,EAAE;wBACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;wBAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAe,CAAC,CAAC;wBAC5D,IAAI,QAAQ,IAAI,QAAQ,EAAE;4BACtB,cAAc,CAAC,2BAA4B,CAAC,QAAQ,CAAC,GAAG,QAAS,CAAC;yBACrE;qBACJ;iBACJ;gBAED,IAAI,qBAAqB,CAAC,cAAc,KAAK,SAAS,EAAE;oBACpD,cAAc,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;iBAC7F;qBAAM,IAAI,qBAAqB,CAAC,aAAa,KAAK,SAAS,EAAE;oBAC1D,cAAc,CAAC,aAAa,GAAG,CAAC,CAAC,qBAAqB,CAAC,aAAa,CAAC;iBACxE;gBAED,IAAI,qBAAqB,CAAC,sBAAsB,KAAK,SAAS,EAAE;oBAC5D,cAAc,CAAC,sBAAsB,GAAG,CAAC,CAAC,qBAAqB,CAAC,sBAAsB,CAAC;iBAC1F;gBAED,cAAc,CAAC,QAAQ,GAAG,IAAA,yBAAe,EAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;gBAC1E,cAAc,CAAC,eAAe,GAAG,CAAC,CAAC,qBAAqB,CAAC,eAAe,CAAC;gBAEzE,MAAM,UAAU,GAAG,qBAAqB,CAAC,UAAU,CAAC;gBACpD,IAAI,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;oBAClE,cAAc,CAAC,UAAU,GAAG,UAAU;yBACjC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAA,eAAQ,EAAC,CAAC,CAAC,CAAC;yBAC/B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,wBAAY,EAAC,SAAS,CAAC,QAAQ,EAAE,IAAA,iCAAmB,EAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;iBACjG;gBAED,IAAI,qBAAqB,CAAC,gBAAgB,KAAK,SAAS,EAAE;oBACtD,cAAc,CAAC,gBAAgB,GAAG,qBAAqB,CAAC,gBAAgB,CAAC;iBAC5E;gBAED,IAAI,qBAAqB,CAAC,qBAAqB,KAAK,SAAS,EAAE;oBAC3D,cAAc,CAAC,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC;iBACtF;gBAED,IACI,cAAc,CAAC,QAAQ,KAAK,kBAAQ,CAAC,GAAG;oBACxC,qBAAqB,CAAC,qBAAqB,KAAK,SAAS,EAC3D;oBACE,cAAc,CAAC,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC;iBACtF;gBAED,IAAI,qBAAqB,CAAC,2BAA2B,KAAK,SAAS,EAAE;oBACjE,cAAc,CAAC,2BAA2B,GAAG,qBAAqB,CAAC,2BAA2B,CAAC;iBAClG;aACJ;iBAAM;gBACH,cAAc,CAAC,eAAe,GAAG,IAAI,CAAC;aACzC;YAED,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YAC7E,IAAI,cAAc,EAAE;gBAChB,IAAI,cAAc,CAAC,aAAa,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,aAAa,GAAG,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC;iBACjE;gBAED,IAAI,cAAc,CAAC,sBAAsB,KAAK,SAAS,EAAE;oBACrD,cAAc,CAAC,sBAAsB,GAAG,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC;iBACnF;gBAED,cAAc,CAAC,uBAAuB,GAAG,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC;gBAClF,cAAc,CAAC,sBAAsB,GAAG,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC;gBAEhF,MAAM,gBAAgB,GAAG,cAAc,CAAC,gBAAgB,CAAC;gBACzD,IAAI,gBAAgB,IAAI,IAAA,eAAQ,EAAC,gBAAgB,CAAC,EAAE;oBAChD,cAAc,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;iBACtD;aACJ;SACJ;QAAC,OAAO,KAAK,EAAE;YACZ,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,2BAA2B,KAAK,EAAE,CAAC,CAAC;SAC1D;QACD,OAAO,cAAc,CAAC;IAC1B,CAAC;IAED,wBAAwB,CAAC,SAAiB;QACtC,IAAI,IAAA,kBAAW,GAAE,IAAI,CAAC,IAAA,6CAAyB,GAAE,EAAE;YAC/C,uEAAuE;YACvE,+CAA+C;YAC/C,OAAO,SAAS,CAAC;SACpB;QAED,OAAO,IAAI,uCAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC;IAEkB,UAAU;QACzB,OAAO,IAAI,+BAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC;IAEkB,oBAAoB,CAAC,EAAc,EAAE,OAAsB,EAAE,IAAU;QACtF,MAAM,cAAc,GAAG,IAAI,+BAAc,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;QAE7D,qEAAqE;QACrE,wCAAwC;QACxC,cAAc,CAAC,eAAe,EAAE,CAAC;QAEjC,OAAO,cAAc,CAAC;IAC1B,CAAC;IAES,cAAc,CAAC,MAA4B,EAAE,KAAwB;QAC3E,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;IAES,oBAAoB,CAAC,OAAe;QAC1C,OAAO,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC;IAES,KAAK,CAAC,iBAAiB,CAC7B,MAAwB,EACxB,KAAwB;QAExB,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAEjC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAChF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC3D,OAAO,uCAAkB,CAAC,yBAAyB,CAC/C,SAAS,EACT,QAAQ,EACR,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,OAAO,CAAC,IAAI,EACnB,KAAK,CACR,CAAC;IACN,CAAC;IAES,sBAAsB;QAC5B,2EAA2E;QAC3E,kDAAkD;QAElD,IAAI,gBAAqE,CAAC;QAC1E,OAAO;YACH,SAAS,EAAE,CAAC,IAAqB,EAAE,EAAE,CAAC,IAAI;YAC1C,KAAK,EAAE,GAAG,EAAE;gBACR,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,EAAE;oBACzC,gBAAgB,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAC;oBACpE,gBAAgB;yBACX,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACf,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACvB,CAAC,CAAC;yBACD,YAAY,EAAE,CAAC;iBACvB;qBAAM;oBACH,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC;iBAC9D;YACL,CAAC;YACD,MAAM,EAAE,CAAC,OAAe,EAAE,EAAE;gBACxB,IAAI,gBAAgB,EAAE;oBAClB,gBAAgB;yBACX,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACf,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBAC7B,CAAC,CAAC;yBACD,YAAY,EAAE,CAAC;iBACvB;qBAAM;oBACH,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC;iBACxE;YACL,CAAC;YACD,GAAG,EAAE,GAAG,EAAE;gBACN,IAAI,gBAAgB,EAAE;oBAClB,gBAAgB;yBACX,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACf,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACpB,CAAC,CAAC;yBACD,YAAY,EAAE,CAAC;oBACpB,gBAAgB,GAAG,SAAS,CAAC;iBAChC;qBAAM;oBACH,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;iBAC5D;YACL,CAAC;SACJ,CAAC;IACN,CAAC;CACJ;AAtQD,sCAsQC"}

package/dist/tests/cacheManager.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tests/cacheManager.test.js

@@ -0,0 +1,69 @@
+"use strict";
+/*
+ * cacheManager.test.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Unit tests for cache manager
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert_1 = __importDefault(require("assert"));
+const cacheManager_1 = require("../analyzer/cacheManager");
+test('basic', () => {
+    const manager = new cacheManager_1.CacheManager();
+    const mock = new MockCacheOwner(10);
+    manager.registerCacheOwner(mock);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 10);
+    manager.unregisterCacheOwner(mock);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 0);
+});
+test('nested stopTracking', () => {
+    const manager = new cacheManager_1.CacheManager();
+    const mock = new MockCacheOwner(10);
+    manager.registerCacheOwner(mock);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 10);
+    const handle1 = manager.pauseTracking();
+    assert_1.default.strictEqual(manager.getCacheUsage(), -1);
+    // nested
+    const handle2 = manager.pauseTracking();
+    assert_1.default.strictEqual(manager.getCacheUsage(), -1);
+    handle2.dispose();
+    assert_1.default.strictEqual(manager.getCacheUsage(), -1);
+    handle1.dispose();
+    assert_1.default.strictEqual(manager.getCacheUsage(), 10);
+    manager.unregisterCacheOwner(mock);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 0);
+});
+test('multiple owners', () => {
+    const manager = new cacheManager_1.CacheManager();
+    const mock1 = new MockCacheOwner(10);
+    const mock2 = new MockCacheOwner(20);
+    manager.registerCacheOwner(mock1);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 10);
+    manager.registerCacheOwner(mock2);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 30);
+    const handle = manager.pauseTracking();
+    assert_1.default.strictEqual(manager.getCacheUsage(), -1);
+    manager.unregisterCacheOwner(mock1);
+    assert_1.default.strictEqual(manager.getCacheUsage(), -1);
+    handle.dispose();
+    assert_1.default.strictEqual(manager.getCacheUsage(), 20);
+    manager.unregisterCacheOwner(mock2);
+    assert_1.default.strictEqual(manager.getCacheUsage(), 0);
+});
+class MockCacheOwner {
+    constructor(_used) {
+        this._used = _used;
+        // empty
+    }
+    getCacheUsage() {
+        return this._used;
+    }
+    emptyCache() {
+        this._used = 0;
+    }
+}
+//# sourceMappingURL=cacheManager.test.js.map

package/dist/tests/cacheManager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cacheManager.test.js","sourceRoot":"","sources":["../../../src/tests/cacheManager.test.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;AAEH,oDAA4B;AAE5B,2DAAoE;AAEpE,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;IACf,MAAM,OAAO,GAAG,IAAI,2BAAY,EAAE,CAAC;IACnC,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;IAEpC,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;IACjC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;IACnC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IAC7B,MAAM,OAAO,GAAG,IAAI,2BAAY,EAAE,CAAC;IACnC,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;IAEpC,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;IACjC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,MAAM,OAAO,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IACxC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhD,SAAS;IACT,MAAM,OAAO,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IACxC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhD,OAAO,CAAC,OAAO,EAAE,CAAC;IAClB,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhD,OAAO,CAAC,OAAO,EAAE,CAAC;IAClB,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;IACnC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE;IACzB,MAAM,OAAO,GAAG,IAAI,2BAAY,EAAE,CAAC;IACnC,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;IAErC,OAAO,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAClC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,OAAO,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAClC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IACvC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhD,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACpC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhD,MAAM,CAAC,OAAO,EAAE,CAAC;IACjB,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACpC,gBAAM,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,MAAM,cAAc;IAChB,YAAoB,KAAa;QAAb,UAAK,GAAL,KAAK,CAAQ;QAC7B,QAAQ;IACZ,CAAC;IAED,aAAa;QACT,OAAO,IAAI,CAAC,KAAK,CAAC;IACtB,CAAC;IAED,UAAU;QACN,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;IACnB,CAAC;CACJ"}

package/dist/tests/checker.test.js

@@ -387,10 +387,10 @@ test('RegionComments1', () => {
 test('Deprecated2', () => {
     const configOptions = new configOptions_1.ConfigOptions('.');
     const analysisResults1 = TestUtils.typeAnalyzeSampleFiles(['deprecated2.py'], configOptions);
-    TestUtils.validateResults(analysisResults1, 0, 0, 0, undefined, undefined, 5);
+    TestUtils.validateResults(analysisResults1, 0, 0, 0, undefined, undefined, 6);
     configOptions.diagnosticRuleSet.reportDeprecated = 'error';
     const analysisResults2 = TestUtils.typeAnalyzeSampleFiles(['deprecated2.py'], configOptions);
-    TestUtils.validateResults(analysisResults2, 5);
+    TestUtils.validateResults(analysisResults2, 6);
 });
 test('Deprecated3', () => {
     const configOptions = new configOptions_1.ConfigOptions('.');

package/dist/tests/completions.test.js

@@ -4,7 +4,12 @@
  *
  * completions tests.
  */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+const assert_1 = __importDefault(require("assert"));
+const vscode_languageserver_1 = require("vscode-languageserver");
 const vscode_languageserver_types_1 = require("vscode-languageserver-types");
 const testState_1 = require("./harness/fourslash/testState");
 test('completion import statement tooltip', async () => {
@@ -328,4 +333,444 @@ test('completion from import statement tooltip - implicit module', async () => {
         },
     });
 });
+test('include literals in expression completion', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import TypedDict
+//// 
+//// class TestType(TypedDict):
+////     A: str
+////     B: int
+//// 
+//// var: TestType = {}
+//// 
+//// var[[|A/*marker*/|]]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: "'A'",
+                    textEdit: { range: state.getPositionRange('marker'), newText: "'A'" },
+                },
+            ],
+        },
+    });
+});
+test('include literals in set key', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import TypedDict
+//// 
+//// class TestType(TypedDict):
+////     A: str
+////     B: int
+//// 
+//// var: TestType = { [|A/*marker*/|] }
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: "'A'",
+                    textEdit: { range: state.getPositionRange('marker'), newText: "'A'" },
+                },
+            ],
+        },
+    });
+});
+test('include literals in dict key', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import TypedDict
+//// 
+//// class TestType(TypedDict):
+////     A: str
+////     B: int
+//// 
+//// var: TestType = { [|A/*marker*/|] : "hello" }
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"A"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"A"' },
+                },
+            ],
+        },
+    });
+});
+test('literals support for binary operators - equals', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     if c == [|"/*marker*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"EUR"' },
+                },
+            ],
+        },
+    });
+});
+test('literals support for binary operators - not equals', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     if c != [|"/*marker*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"EUR"' },
+                },
+            ],
+        },
+    });
+});
+test('literals support for binary operators without string node', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     if c != [|/*marker*/|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                },
+            ],
+        },
+    });
+});
+test('literals support for binary operators with prior word', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     if c != [|US/*marker*/|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                },
+            ],
+        },
+    });
+});
+test('literals support for binary operators - assignment expression', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     if c := [|"/*marker*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"EUR"' },
+                },
+            ],
+        },
+    });
+});
+test('literals support for call', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency) -> Currency:
+////     return c
+////
+//// if foo([|"/*marker1*/"|]) == [|"/*marker2*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker1 = state.getMarkerByName('marker1');
+    state.openFile(marker1.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker1: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker1'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker1'), newText: '"EUR"' },
+                },
+            ],
+        },
+        marker2: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker2'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker2'), newText: '"EUR"' },
+                },
+            ],
+        },
+    });
+});
+test('list with literal types', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// a: list[Currency] = [[|"/*marker*/"|]]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"EUR"' },
+                },
+            ],
+        },
+    });
+});
+test('literals support for match - error case', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     match c:
+////         case [|/*marker*/|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                },
+            ],
+        },
+    });
+});
+test('literals support for match - simple case', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     match c:
+////         case [|"/*marker*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"USD"' },
+                },
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"EUR"',
+                    textEdit: { range: state.getPositionRange('marker'), newText: '"EUR"' },
+                },
+            ],
+        },
+    });
+});
+test('literals support for match - simple case without string', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     match c:
+////         case [|US/*marker*/|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    state.openFile(marker.fileName);
+    await state.verifyCompletion('included', vscode_languageserver_types_1.MarkupKind.Markdown, {
+        marker: {
+            completions: [
+                {
+                    kind: vscode_languageserver_types_1.CompletionItemKind.Constant,
+                    label: '"USD"',
+                },
+            ],
+        },
+    });
+});
+test('completion quote trigger', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["USD", "EUR"]
+//// 
+//// def foo(c: Currency):
+////     match c:
+////         case [|"/*marker*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    const filePath = marker.fileName;
+    const position = state.convertOffsetToPosition(filePath, marker.position);
+    const options = {
+        format: 'markdown',
+        snippet: false,
+        lazyEdit: false,
+        autoImport: false,
+        extraCommitChars: false,
+        importFormat: "absolute" /* Absolute */,
+        includeUserSymbolsInAutoImport: false,
+        triggerCharacter: '"',
+    };
+    const result = await state.workspace.service.getCompletionsForPosition(filePath, position, state.workspace.rootPath, options, undefined, vscode_languageserver_1.CancellationToken.None);
+    (0, assert_1.default)(result);
+    const item = result.completionList.items.find((a) => a.label === '"USD"');
+    (0, assert_1.default)(item);
+});
+test('completion quote trigger - middle', async () => {
+    const code = `
+// @filename: test.py
+//// from typing import Literal
+//// 
+//// Currency = Literal["Quote'Middle"]
+//// 
+//// def foo(c: Currency):
+////     match c:
+////         case [|"Quote'/*marker*/"|]
+    `;
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    const filePath = marker.fileName;
+    const position = state.convertOffsetToPosition(filePath, marker.position);
+    const options = {
+        format: 'markdown',
+        snippet: false,
+        lazyEdit: false,
+        autoImport: false,
+        extraCommitChars: false,
+        importFormat: "absolute" /* Absolute */,
+        includeUserSymbolsInAutoImport: false,
+        triggerCharacter: "'",
+    };
+    const result = await state.workspace.service.getCompletionsForPosition(filePath, position, state.workspace.rootPath, options, undefined, vscode_languageserver_1.CancellationToken.None);
+    assert_1.default.strictEqual(result === null || result === void 0 ? void 0 : result.completionList.items.length, 0);
+});
 //# sourceMappingURL=completions.test.js.map