@zzzen/pyright-internal 1.2.0-dev.20260920 → 1.2.0-dev.20260927

package/dist/tests/protocolRequirementRecursive.test.js

@@ -0,0 +1,118 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = __importStar(require("assert"));
+const constraintTracker_1 = require("../analyzer/constraintTracker");
+const protocols_1 = require("../analyzer/protocols");
+const types_1 = require("../analyzer/types");
+const diagnostic_1 = require("../common/diagnostic");
+const testState_1 = require("./harness/fourslash/testState");
+describe.each([
+    { element: 'RecursiveLeaf', accepted: true },
+    { element: 'RecursiveMixed', accepted: true },
+    { element: 'RecursiveSequence', accepted: true },
+    { element: 'MutualLeft', accepted: true },
+    { element: 'RecursiveGeneric[Good]', accepted: true },
+    { element: 'RecursiveGeneric[Bad]', accepted: false },
+    { element: 'BadRecursiveSequence', accepted: false },
+])('ProtocolRequirementRecursive $element', ({ element, accepted }) => {
+    test.each(accepted ? ['shortcut', 'cold', 'diagnostic'] : ['cold', 'diagnostic'])('%s', (mode) => {
+        const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: test.py
+//// from collections.abc import Iterator
+//// from typing import Protocol, TypeAlias, TypeVar
+//// T_co = TypeVar("T_co", covariant=True)
+//// class SupportsArray(Protocol[T_co]):
+////     def __array__(self) -> T_co: ...
+//// class NestedSequence(Protocol[T_co]):
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> T_co | "NestedSequence[T_co]": ...
+////     def __iter__(self, /) -> Iterator[T_co | "NestedSequence[T_co]"]: ...
+//// class Good:
+////     def __array__(self) -> int: ...
+//// class Bad:
+////     def __array__(self) -> str: ...
+//// class BadRecursiveSequence:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> Bad | "BadRecursiveSequence": ...
+////     def __iter__(self, /) -> Iterator[Bad | "BadRecursiveSequence"]: ...
+//// RecursiveLeaf: TypeAlias = Good | list["RecursiveLeaf"]
+//// RecursiveMixed: TypeAlias = Good | list["RecursiveMixed"] | tuple["RecursiveMixed", ...]
+//// class RecursiveSequence:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> Good | "RecursiveSequence": ...
+////     def __iter__(self, /) -> Iterator[Good | "RecursiveSequence"]: ...
+//// class MutualLeft:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> Good | "MutualRight": ...
+////     def __iter__(self, /) -> Iterator[Good | "MutualRight"]: ...
+//// class MutualRight:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> Good | MutualLeft: ...
+////     def __iter__(self, /) -> Iterator[Good | MutualLeft]: ...
+//// type RecursiveGeneric[Value] = Value | list[RecursiveGeneric[Value]]
+//// def check(source: list[${element}], destination: NestedSequence[SupportsArray[int]]):
+////     /*source*/source
+////     /*destination*/destination
+            `).state;
+        const evaluator = state.program.evaluator;
+        const typeAt = (marker) => {
+            const node = (0, testState_1.getNodeAtMarker)(state, marker);
+            assert.strictEqual(node.nodeType, 38 /* ParseNodeType.Name */);
+            return evaluator.getTypeOfExpression(node).type;
+        };
+        const source = typeAt('source');
+        const destination = typeAt('destination');
+        assert.ok((0, types_1.isClassInstance)(source) && (0, types_1.isClassInstance)(destination));
+        const target = types_1.ClassType.cloneAsInstantiable(destination);
+        const constraints = new constraintTracker_1.ConstraintTracker();
+        constraints.getMainConstraintSet().addScopeId('caller');
+        const sourceBefore = evaluator.printType(source);
+        for (let repetition = 0; repetition < 2; repetition++) {
+            if (mode === 'shortcut') {
+                assert.strictEqual((0, protocols_1.tryFastRejectSequenceProtocol)(evaluator, target, source, constraints, 0 /* AssignTypeFlags.Default */, 0), undefined);
+            }
+            else {
+                assert.strictEqual((0, protocols_1.assignClassToProtocol)(evaluator, target, source, mode === 'diagnostic' ? new diagnostic_1.DiagnosticAddendum() : undefined, constraints, 0 /* AssignTypeFlags.Default */, 0), accepted);
+            }
+            assert.strictEqual(evaluator.printType(source), sourceBefore);
+            assert.deepStrictEqual(constraints.getConstraintSets().map((set) => ({
+                scopes: [...set.getScopeIds()],
+                bounds: set.getTypeVars(),
+            })), [{ scopes: ['caller'], bounds: [] }]);
+        }
+    });
+});
+//# sourceMappingURL=protocolRequirementRecursive.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementRecursive.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementRecursive.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AAEjC,qEAAkE;AAClE,qDAA6F;AAE7F,6CAA+D;AAC/D,qDAA0D;AAE1D,6DAAsF;AAEtF,QAAQ,CAAC,IAAI,CAAC;IACV,EAAE,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE;IAC5C,EAAE,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,IAAI,EAAE;IAC7C,EAAE,OAAO,EAAE,mBAAmB,EAAE,QAAQ,EAAE,IAAI,EAAE;IAChD,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE;IACzC,EAAE,OAAO,EAAE,wBAAwB,EAAE,QAAQ,EAAE,IAAI,EAAE;IACrD,EAAE,OAAO,EAAE,uBAAuB,EAAE,QAAQ,EAAE,KAAK,EAAE;IACrD,EAAE,OAAO,EAAE,sBAAsB,EAAE,QAAQ,EAAE,KAAK,EAAE;CACvD,CAAC,CAAC,uCAAuC,EAAE,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE;IAClE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;QAC7F,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;8BAkCb,OAAO;;;aAGxB,CAAC,CAAC,KAAK,CAAC;QACb,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAU,CAAC;QAC3C,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,EAAE;YAC9B,MAAM,IAAI,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,8BAAqB,CAAC;YACtD,OAAO,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QACpD,CAAC,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChC,MAAM,WAAW,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;QAC1C,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,MAAM,CAAC,IAAI,IAAA,uBAAe,EAAC,WAAW,CAAC,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,iBAAS,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;QAC5C,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACjD,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC;YACpD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACtB,MAAM,CAAC,WAAW,CACd,IAAA,yCAA6B,EAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,mCAA2B,CAAC,CAAC,EACjG,SAAS,CACZ,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,MAAM,CAAC,WAAW,CACd,IAAA,iCAAqB,EACjB,SAAS,EACT,MAAM,EACN,MAAM,EACN,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,+BAAkB,EAAE,CAAC,CAAC,CAAC,SAAS,EAC5D,WAAW,mCAEX,CAAC,CACJ,EACD,QAAQ,CACX,CAAC;YACN,CAAC;YACD,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;YAC9D,MAAM,CAAC,eAAe,CAClB,WAAW,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC1C,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;gBAC9B,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE;aAC5B,CAAC,CAAC,EACH,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACvC,CAAC;QACN,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

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

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

package/dist/tests/protocolRequirementSpeculative.test.js

@@ -0,0 +1,200 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = __importStar(require("assert"));
+const constraintTracker_1 = require("../analyzer/constraintTracker");
+const protocols_1 = require("../analyzer/protocols");
+const types_1 = require("../analyzer/types");
+const diagnostic_1 = require("../common/diagnostic");
+const testState_1 = require("./harness/fourslash/testState");
+describe.each(['Plain', 'Callable[[], int]', 'int'])('ProtocolRequirementSpeculative %s', (element) => {
+    test.each([0 /* AssignTypeFlags.Default */, 256 /* AssignTypeFlags.RetainLiteralsForTypeVar */].flatMap((flags) => [false, true].flatMap((multipleSets) => [false, true].flatMap((positiveFirst) => [false, true].map((populated) => ({ flags, multipleSets, positiveFirst, populated }))))))('flags=$flags multipleSets=$multipleSets positiveFirst=$positiveFirst populated=$populated', ({ flags, multipleSets, positiveFirst, populated }) => {
+        const run = (speculative) => {
+            const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: test.py
+//// from collections.abc import Callable, Iterator
+//// from typing import Literal, Protocol, TypeVar
+//// T_co = TypeVar("T_co", covariant=True)
+//// class Leaf(Protocol):
+////     def serialize(self) -> int: ...
+//// class Plain: pass
+//// class Good:
+////     def serialize(self) -> int: ...
+//// class NestedSequence(Protocol[T_co]):
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> T_co | "NestedSequence[T_co]": ...
+////     def __iter__(self, /) -> Iterator[T_co | "NestedSequence[T_co]"]: ...
+//// def check[T](variable: T, literal: Literal[7], other: Literal[8], integer: int, source: list[${element}], destination: NestedSequence[${element === 'int' ? 'str' : 'Leaf'}], valid: list[${element === 'int' ? 'str' : 'Good'}]):
+////     /*source*/source
+////     /*destination*/destination
+////     /*valid*/valid
+////     /*variable*/variable
+////     /*literal*/literal
+////     /*other*/other
+////     /*integer*/integer
+            `).state;
+            const evaluator = state.program.evaluator;
+            const typeAt = (marker) => {
+                const node = (0, testState_1.getNodeAtMarker)(state, marker);
+                assert.strictEqual(node.nodeType, 38 /* ParseNodeType.Name */);
+                const type = evaluator.getTypeOfExpression(node).type;
+                assert.ok((0, types_1.isClassInstance)(type));
+                return type;
+            };
+            const source = typeAt('source');
+            const target = types_1.ClassType.cloneAsInstantiable(typeAt('destination'));
+            const valid = typeAt('valid');
+            const accepted = types_1.ClassType.specialize(target, [source.priv.typeArgs[0]]);
+            const constraints = new constraintTracker_1.ConstraintTracker();
+            const variableNode = (0, testState_1.getNodeAtMarker)(state, 'variable');
+            assert.strictEqual(variableNode.nodeType, 38 /* ParseNodeType.Name */);
+            const variable = evaluator.getTypeOfExpression(variableNode).type;
+            assert.ok((0, types_1.isTypeVar)(variable) && variable.priv.freeTypeVar);
+            const freeVariable = variable.priv.freeTypeVar;
+            if (populated) {
+                constraints.setBounds(freeVariable, typeAt('literal'), typeAt('integer'), false);
+            }
+            constraints.getMainConstraintSet().addScopeId('caller');
+            if (multipleSets) {
+                const other = constraints.getMainConstraintSet().clone();
+                other.addScopeId('alternative');
+                if (populated) {
+                    other.setBounds(freeVariable, typeAt('other'), typeAt('integer'), true);
+                }
+                constraints.addConstraintSets([constraints.getMainConstraintSet(), other]);
+            }
+            const observe = () => ({
+                source: evaluator.printType(source),
+                sets: constraints.getConstraintSets().map((set) => ({
+                    scopes: [...set.getScopeIds()],
+                    bounds: set.getTypeVars().map((entry) => ({
+                        variable: evaluator.printType(entry.typeVar),
+                        lower: entry.lowerBound && evaluator.printType(entry.lowerBound),
+                        upper: entry.upperBound && evaluator.printType(entry.upperBound),
+                        retain: entry.retainLiterals,
+                    })),
+                })),
+            });
+            const before = observe();
+            const assign = (destination, input = source, details = new diagnostic_1.DiagnosticAddendum()) => (0, protocols_1.assignClassToProtocol)(evaluator, destination, input, details, constraints, flags, 0);
+            if (positiveFirst) {
+                assert.strictEqual(assign(accepted), true);
+                assert.deepStrictEqual(observe(), before);
+            }
+            const observations = [];
+            const reject = () => {
+                assert.strictEqual(evaluator.isSpeculativeModeInUse(undefined), speculative);
+                assert.strictEqual(assign(target), false);
+                assert.deepStrictEqual(observe(), before);
+                observations.push(observe());
+            };
+            if (speculative) {
+                evaluator.useSpeculativeMode((0, testState_1.getNodeAtMarker)(state, 'source'), () => {
+                    reject();
+                    evaluator.useSpeculativeMode((0, testState_1.getNodeAtMarker)(state, 'destination'), reject);
+                    reject();
+                });
+            }
+            else {
+                reject();
+                reject();
+                reject();
+            }
+            assert.strictEqual(evaluator.isSpeculativeModeInUse(undefined), false);
+            const details = new diagnostic_1.DiagnosticAddendum();
+            assert.strictEqual(assign(target, source, details), false);
+            assert.ok(details.getString().length > 0);
+            assert.strictEqual(assign(accepted), true);
+            assert.strictEqual(assign(target, valid), true);
+            assert.deepStrictEqual(observe(), before);
+            observations.push(observe());
+            return { observations, diagnostic: details.getString() };
+        };
+        assert.deepStrictEqual(run(true), run(false));
+    });
+});
+test.each([false, true])('ProtocolRequirementExceptionUnwind speculative=%s', (speculative) => {
+    const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: test.py
+//// from collections.abc import Iterator
+//// from typing import Protocol, TypeVar
+//// T_co = TypeVar("T_co", covariant=True)
+//// class NestedSequence(Protocol[T_co]):
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> T_co | "NestedSequence[T_co]": ...
+////     def __iter__(self, /) -> Iterator[T_co | "NestedSequence[T_co]"]: ...
+//// def check(source: list[int], destination: NestedSequence[str]):
+////     /*source*/source
+////     /*destination*/destination
+    `).state;
+    const evaluator = state.program.evaluator;
+    const typeAt = (marker) => {
+        const node = (0, testState_1.getNodeAtMarker)(state, marker);
+        assert.strictEqual(node.nodeType, 38 /* ParseNodeType.Name */);
+        const type = evaluator.getTypeOfExpression(node).type;
+        assert.ok((0, types_1.isClassInstance)(type));
+        return type;
+    };
+    const source = typeAt('source');
+    const target = types_1.ClassType.cloneAsInstantiable(typeAt('destination'));
+    evaluator.inferVarianceForClass(target);
+    const constraints = new constraintTracker_1.ConstraintTracker();
+    constraints.getMainConstraintSet().addScopeId('caller');
+    const assign = (destination = target) => (0, protocols_1.assignClassToProtocol)(evaluator, destination, source, undefined, constraints, 0 /* AssignTypeFlags.Default */, 0);
+    const failure = new Error('Injected protocol member lookup failure');
+    const spy = jest.spyOn(evaluator, 'getDeclaredTypeOfSymbol').mockImplementationOnce(() => {
+        throw failure;
+    });
+    try {
+        assert.throws(() => {
+            if (speculative) {
+                evaluator.useSpeculativeMode((0, testState_1.getNodeAtMarker)(state, 'source'), () => evaluator.useSpeculativeMode((0, testState_1.getNodeAtMarker)(state, 'destination'), () => assign()));
+            }
+            else {
+                assign();
+            }
+        }, (error) => error === failure);
+        assert.strictEqual(spy.mock.calls.length, 1);
+    }
+    finally {
+        spy.mockRestore();
+    }
+    assert.strictEqual(evaluator.isSpeculativeModeInUse(undefined), false);
+    assert.strictEqual(assign(), false);
+    assert.strictEqual(assign(types_1.ClassType.specialize(target, [source.priv.typeArgs[0]])), true);
+    assert.strictEqual(assign(), false);
+    assert.deepStrictEqual(constraints.getConstraintSets().map((set) => ({ scopes: [...set.getScopeIds()], bounds: set.getTypeVars() })), [{ scopes: ['caller'], bounds: [] }]);
+});
+//# sourceMappingURL=protocolRequirementSpeculative.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementSpeculative.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementSpeculative.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AAEjC,qEAAkE;AAClE,qDAA8D;AAE9D,6CAA0E;AAC1E,qDAA0D;AAE1D,6DAAsF;AAEtF,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,mBAAmB,EAAE,KAAK,CAAC,CAAC,CAAC,mCAAmC,EAAE,CAAC,OAAO,EAAE,EAAE;IAClG,IAAI,CAAC,IAAI,CACL,qFAAmE,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAClF,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CACnC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,EAAE,CACpC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,CAAC,CACxF,CACJ,CACJ,CACJ,CACG,2FAA2F,EAC3F,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,EAAE;QAClD,MAAM,GAAG,GAAG,CAAC,WAAoB,EAAE,EAAE;YACjC,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;;;;;;;;;oGAciD,OAAO,kCACvF,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAChC,kBAAkB,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;;;;;;;;aAQvD,CAAC,CAAC,KAAK,CAAC;YACL,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAU,CAAC;YAC3C,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,EAAE;gBAC9B,MAAM,IAAI,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC5C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,8BAAqB,CAAC;gBACtD,MAAM,IAAI,GAAG,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;gBACtD,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,IAAI,CAAC,CAAC,CAAC;gBACjC,OAAO,IAAI,CAAC;YAChB,CAAC,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,MAAM,GAAG,iBAAS,CAAC,mBAAmB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,QAAQ,GAAG,iBAAS,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1E,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACxD,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,QAAQ,8BAAqB,CAAC;YAC9D,MAAM,QAAQ,GAAG,SAAS,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC;YAClE,MAAM,CAAC,EAAE,CAAC,IAAA,iBAAS,EAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5D,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC;YAC/C,IAAI,SAAS,EAAE,CAAC;gBACZ,WAAW,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;YACrF,CAAC;YACD,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACxD,IAAI,YAAY,EAAE,CAAC;gBACf,MAAM,KAAK,GAAG,WAAW,CAAC,oBAAoB,EAAE,CAAC,KAAK,EAAE,CAAC;gBACzD,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;gBAChC,IAAI,SAAS,EAAE,CAAC;oBACZ,KAAK,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;gBAC5E,CAAC;gBACD,WAAW,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,oBAAoB,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;YAC/E,CAAC;YACD,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,CAAC;gBACnB,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC;gBACnC,IAAI,EAAE,WAAW,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBAChD,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;oBAC9B,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACtC,QAAQ,EAAE,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC;wBAC5C,KAAK,EAAE,KAAK,CAAC,UAAU,IAAI,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC;wBAChE,KAAK,EAAE,KAAK,CAAC,UAAU,IAAI,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC;wBAChE,MAAM,EAAE,KAAK,CAAC,cAAc;qBAC/B,CAAC,CAAC;iBACN,CAAC,CAAC;aACN,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,OAAO,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,CAAC,WAAsB,EAAE,KAAK,GAAG,MAAM,EAAE,OAAO,GAAG,IAAI,+BAAkB,EAAE,EAAE,EAAE,CAC1F,IAAA,iCAAqB,EAAC,SAAS,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YACzF,IAAI,aAAa,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;gBAC3C,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,MAAM,CAAC,CAAC;YAC9C,CAAC;YACD,MAAM,YAAY,GAAiC,EAAE,CAAC;YACtD,MAAM,MAAM,GAAG,GAAG,EAAE;gBAChB,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,sBAAsB,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,CAAC;gBAC7E,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;gBAC1C,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,MAAM,CAAC,CAAC;gBAC1C,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YACjC,CAAC,CAAC;YACF,IAAI,WAAW,EAAE,CAAC;gBACd,SAAS,CAAC,kBAAkB,CAAC,IAAA,2BAAe,EAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,GAAG,EAAE;oBAChE,MAAM,EAAE,CAAC;oBACT,SAAS,CAAC,kBAAkB,CAAC,IAAA,2BAAe,EAAC,KAAK,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,CAAC;oBAC5E,MAAM,EAAE,CAAC;gBACb,CAAC,CAAC,CAAC;YACP,CAAC;iBAAM,CAAC;gBACJ,MAAM,EAAE,CAAC;gBACT,MAAM,EAAE,CAAC;gBACT,MAAM,EAAE,CAAC;YACb,CAAC;YACD,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,sBAAsB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,IAAI,+BAAkB,EAAE,CAAC;YACzC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;YAC3D,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;YAChD,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,MAAM,CAAC,CAAC;YAC1C,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7B,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;QAC7D,CAAC,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,CAAC,CACJ,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,mDAAmD,EAAE,CAAC,WAAW,EAAE,EAAE;IAC1F,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;;;;;;;KAYlC,CAAC,CAAC,KAAK,CAAC;IACT,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAU,CAAC;IAC3C,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,EAAE;QAC9B,MAAM,IAAI,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,8BAAqB,CAAC;QACtD,MAAM,IAAI,GAAG,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QACtD,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,IAAI,CAAC,CAAC,CAAC;QACjC,OAAO,IAAI,CAAC;IAChB,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChC,MAAM,MAAM,GAAG,iBAAS,CAAC,mBAAmB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;IACpE,SAAS,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;IACxC,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;IAC5C,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,CAAC,WAAW,GAAG,MAAM,EAAE,EAAE,CACpC,IAAA,iCAAqB,EAAC,SAAS,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,mCAA2B,CAAC,CAAC,CAAC;IAC9G,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IACrE,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC,sBAAsB,CAAC,GAAG,EAAE;QACrF,MAAM,OAAO,CAAC;IAClB,CAAC,CAAC,CAAC;IACH,IAAI,CAAC;QACD,MAAM,CAAC,MAAM,CACT,GAAG,EAAE;YACD,IAAI,WAAW,EAAE,CAAC;gBACd,SAAS,CAAC,kBAAkB,CAAC,IAAA,2BAAe,EAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,GAAG,EAAE,CAChE,SAAS,CAAC,kBAAkB,CAAC,IAAA,2BAAe,EAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CACtF,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,MAAM,EAAE,CAAC;YACb,CAAC;QACL,CAAC,EACD,CAAC,KAAc,EAAE,EAAE,CAAC,KAAK,KAAK,OAAO,CACxC,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACjD,CAAC;YAAS,CAAC;QACP,GAAG,CAAC,WAAW,EAAE,CAAC;IACtB,CAAC;IACD,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,sBAAsB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;IACvE,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,KAAK,CAAC,CAAC;IACpC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,iBAAS,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAC3F,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,KAAK,CAAC,CAAC;IACpC,MAAM,CAAC,eAAe,CAClB,WAAW,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,EAC7G,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACvC,CAAC;AACN,CAAC,CAAC,CAAC"}

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

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

package/dist/tests/protocolRequirementState.test.js

@@ -0,0 +1,161 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = __importStar(require("assert"));
+const constraintTracker_1 = require("../analyzer/constraintTracker");
+const protocols_1 = require("../analyzer/protocols");
+const types_1 = require("../analyzer/types");
+const typeUtils_1 = require("../analyzer/typeUtils");
+const diagnostic_1 = require("../common/diagnostic");
+const testState_1 = require("./harness/fourslash/testState");
+const cases = ['Plain', 'Plain | Callable[[], int]'].flatMap((element) => ['serialize', 'read'].flatMap((member) => [false, true].flatMap((concrete) => ['empty', 'literal', 'integer', 'none'].map((seed) => ({ element, member, concrete, seed })))));
+test.each(cases)('ProtocolRequirementState $element $member concrete=$concrete seed=$seed', ({ element, member, concrete, seed }) => {
+    const run = (diagnostics) => {
+        const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: test.py
+//// from collections.abc import Callable, Iterator
+//// from typing import Literal, Protocol, TypeVar
+//// T_co = TypeVar("T_co", covariant=True)
+//// class Leaf(Protocol[T_co]):
+////     def ${member}(self) -> T_co: ...
+//// class NestedSequence(Protocol[T_co]):
+////     def __len__(self, /) -> int: ...
+${member === 'read' ? '////     def __iter__(self, /) -> Iterator[T_co | "NestedSequence[T_co]"]: ...' : ''}
+////     def __getitem__(self, index: int, /) -> T_co | "NestedSequence[T_co]": ...
+${member === 'serialize' ? '////     def __iter__(self, /) -> Iterator[T_co | "NestedSequence[T_co]"]: ...' : ''}
+//// class Plain: pass
+//// def check[T](source: list[${element}], rejected: NestedSequence[Leaf[${concrete ? 'int' : 'T'}]], accepted: NestedSequence[${element}], variable: T, literal: Literal[7], other: Literal[8], integer: int, text: str, container: list[T]):
+////     /*source*/source
+////     /*rejected*/rejected
+////     /*accepted*/accepted
+////     /*variable*/variable
+////     /*literal*/literal
+////     /*other*/other
+////     /*integer*/integer
+////     /*text*/text
+////     /*container*/container
+            `).state;
+        const evaluator = state.program.evaluator;
+        const typeAt = (marker) => {
+            const node = (0, testState_1.getNodeAtMarker)(state, marker);
+            assert.strictEqual(node.nodeType, 38 /* ParseNodeType.Name */);
+            return evaluator.getTypeOfExpression(node).type;
+        };
+        const source = typeAt('source');
+        const rejected = typeAt('rejected');
+        const accepted = typeAt('accepted');
+        const variable = typeAt('variable');
+        assert.ok((0, types_1.isClassInstance)(source) && (0, types_1.isClassInstance)(rejected) && (0, types_1.isClassInstance)(accepted));
+        assert.ok((0, types_1.isTypeVar)(variable));
+        const freeVariable = variable.priv.freeTypeVar;
+        assert.ok(freeVariable?.priv.scopeId);
+        const scopes = [freeVariable.priv.scopeId];
+        const target = (0, typeUtils_1.makeTypeVarsFree)(types_1.ClassType.cloneAsInstantiable(rejected), scopes);
+        const container = (0, typeUtils_1.makeTypeVarsFree)(typeAt('container'), scopes);
+        const initial = new constraintTracker_1.ConstraintTracker();
+        if (seed !== 'empty') {
+            initial.setBounds(freeVariable, typeAt('literal'), seed === 'none' ? undefined : typeAt(seed), false);
+        }
+        initial.getMainConstraintSet().addScopeId('caller');
+        const observe = (tracker) => ({
+            solved: evaluator.printType(evaluator.solveAndApplyConstraints(freeVariable, tracker)),
+            nested: evaluator.printType(evaluator.solveAndApplyConstraints(container, tracker)),
+            score: tracker.getScore(),
+            sets: tracker.getConstraintSets().map((set) => ({
+                scopes: [...set.getScopeIds()],
+                bounds: set.getTypeVars().map((entry) => ({
+                    variable: evaluator.printType(entry.typeVar),
+                    lower: entry.lowerBound && evaluator.printType(entry.lowerBound),
+                    upper: entry.upperBound && evaluator.printType(entry.upperBound),
+                    retain: entry.retainLiterals,
+                })),
+            })),
+        });
+        const before = observe(initial);
+        const sourceBefore = evaluator.printType(source);
+        const eligible = concrete || seed === 'empty';
+        assert.strictEqual((0, protocols_1.tryFastRejectSequenceProtocol)(evaluator, target, source, initial, 0 /* AssignTypeFlags.Default */, 0), eligible ? '__getitem__' : undefined);
+        assert.deepStrictEqual(observe(initial), before);
+        const results = [];
+        for (const matchFlags of [0 /* AssignTypeFlags.Default */, 256 /* AssignTypeFlags.RetainLiteralsForTypeVar */]) {
+            const constraints = initial.clone();
+            for (let repetition = 0; repetition < 2; repetition++) {
+                assert.strictEqual((0, protocols_1.assignClassToProtocol)(evaluator, target, source, diagnostics ? new diagnostic_1.DiagnosticAddendum() : undefined, constraints, matchFlags, 0), false);
+                if (eligible) {
+                    assert.deepStrictEqual(observe(constraints), before);
+                }
+                else {
+                    assert.strictEqual(observe(constraints).solved, seed === 'literal' ? 'Literal[7]' : 'int');
+                }
+                results.push(observe(constraints));
+            }
+            for (const flags of [
+                0 /* AssignTypeFlags.Default */,
+                1 /* AssignTypeFlags.Invariant */,
+                2 /* AssignTypeFlags.Contravariant */,
+            ]) {
+                for (const transfer of ['clone', 'signature', 'copy']) {
+                    const parentBefore = observe(constraints);
+                    const continued = transfer === 'signature' ? constraints.cloneWithSignature('caller') : constraints.clone();
+                    if (transfer === 'copy') {
+                        continued.copyFromClone(constraints);
+                    }
+                    for (const marker of ['literal', 'other', 'integer', 'text']) {
+                        results.push({
+                            assigned: evaluator.assignType(freeVariable, typeAt(marker), undefined, continued, flags),
+                            observation: observe(continued),
+                            parent: observe(constraints),
+                        });
+                    }
+                    assert.deepStrictEqual(observe(initial), before);
+                    if (transfer === 'signature') {
+                        assert.strictEqual(continued.getMainConstraintSet(), constraints.getMainConstraintSet());
+                        assert.deepStrictEqual(observe(constraints), observe(continued));
+                    }
+                    else {
+                        assert.notStrictEqual(continued.getMainConstraintSet(), constraints.getMainConstraintSet());
+                        assert.deepStrictEqual(observe(constraints), parentBefore);
+                    }
+                }
+            }
+            assert.strictEqual((0, protocols_1.assignClassToProtocol)(evaluator, types_1.ClassType.cloneAsInstantiable(accepted), source, undefined, constraints, 0 /* AssignTypeFlags.Default */, 0), true);
+            results.push(observe(constraints));
+            assert.strictEqual(evaluator.printType(source), sourceBefore);
+        }
+        return results;
+    };
+    assert.deepStrictEqual(run(false), run(true));
+});
+//# sourceMappingURL=protocolRequirementState.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementState.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementState.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AAEjC,qEAAkE;AAClE,qDAA6F;AAE7F,6CAA0E;AAC1E,qDAAyD;AACzD,qDAA0D;AAE1D,6DAAsF;AAEtF,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,2BAA2B,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CACrE,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CACrC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAC/B,CAAC,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/F,CACJ,CACJ,CAAC;AAEF,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CACZ,yEAAyE,EACzE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;IACpC,MAAM,GAAG,GAAG,CAAC,WAAoB,EAAE,EAAE;QACjC,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;eAMhC,MAAM;;;EAGnB,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,gFAAgF,CAAC,CAAC,CAAC,EAAE;;EAEzG,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,gFAAgF,CAAC,CAAC,CAAC,EAAE;;iCAE/E,OAAO,oCACxB,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GACvB,gCAAgC,OAAO;;;;;;;;;;aAUtC,CAAC,CAAC,KAAK,CAAC;QACT,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAU,CAAC;QAC3C,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,EAAE;YAC9B,MAAM,IAAI,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,8BAAqB,CAAC;YACtD,OAAO,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QACpD,CAAC,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACpC,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,MAAM,CAAC,IAAI,IAAA,uBAAe,EAAC,QAAQ,CAAC,IAAI,IAAA,uBAAe,EAAC,QAAQ,CAAC,CAAC,CAAC;QAC7F,MAAM,CAAC,EAAE,CAAC,IAAA,iBAAS,EAAC,QAAQ,CAAC,CAAC,CAAC;QAC/B,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC;QAC/C,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAA,4BAAgB,EAAC,iBAAS,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC;QACjF,MAAM,SAAS,GAAG,IAAA,4BAAgB,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,IAAI,qCAAiB,EAAE,CAAC;QACxC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACnB,OAAO,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC;QAC1G,CAAC;QACD,OAAO,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,CAAC,OAA0B,EAAE,EAAE,CAAC,CAAC;YAC7C,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,wBAAwB,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACtF,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,wBAAwB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACnF,KAAK,EAAE,OAAO,CAAC,QAAQ,EAAE;YACzB,IAAI,EAAE,OAAO,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC5C,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;gBAC9B,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBACtC,QAAQ,EAAE,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC;oBAC5C,KAAK,EAAE,KAAK,CAAC,UAAU,IAAI,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC;oBAChE,KAAK,EAAE,KAAK,CAAC,UAAU,IAAI,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC;oBAChE,MAAM,EAAE,KAAK,CAAC,cAAc;iBAC/B,CAAC,CAAC;aACN,CAAC,CAAC;SACN,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,QAAQ,IAAI,IAAI,KAAK,OAAO,CAAC;QAC9C,MAAM,CAAC,WAAW,CACd,IAAA,yCAA6B,EAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,mCAA2B,CAAC,CAAC,EAC7F,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CACvC,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,EAAE,CAAC;QACnB,KAAK,MAAM,UAAU,IAAI,qFAAmE,EAAE,CAAC;YAC3F,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;YACpC,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC;gBACpD,MAAM,CAAC,WAAW,CACd,IAAA,iCAAqB,EACjB,SAAS,EACT,MAAM,EACN,MAAM,EACN,WAAW,CAAC,CAAC,CAAC,IAAI,+BAAkB,EAAE,CAAC,CAAC,CAAC,SAAS,EAClD,WAAW,EACX,UAAU,EACV,CAAC,CACJ,EACD,KAAK,CACR,CAAC;gBACF,IAAI,QAAQ,EAAE,CAAC;oBACX,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,CAAC;gBACzD,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,MAAM,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBAC/F,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;YACvC,CAAC;YACD,KAAK,MAAM,KAAK,IAAI;;;;aAInB,EAAE,CAAC;gBACA,KAAK,MAAM,QAAQ,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE,CAAC;oBACpD,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;oBAC1C,MAAM,SAAS,GACX,QAAQ,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;oBAC9F,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;wBACtB,SAAS,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;oBACzC,CAAC;oBACD,KAAK,MAAM,MAAM,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;wBAC3D,OAAO,CAAC,IAAI,CAAC;4BACT,QAAQ,EAAE,SAAS,CAAC,UAAU,CAC1B,YAAY,EACZ,MAAM,CAAC,MAAM,CAAC,EACd,SAAS,EACT,SAAS,EACT,KAAK,CACR;4BACD,WAAW,EAAE,OAAO,CAAC,SAAS,CAAC;4BAC/B,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC;yBAC/B,CAAC,CAAC;oBACP,CAAC;oBACD,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;oBACjD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;wBAC3B,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,oBAAoB,EAAE,EAAE,WAAW,CAAC,oBAAoB,EAAE,CAAC,CAAC;wBACzF,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;oBACrE,CAAC;yBAAM,CAAC;wBACJ,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,oBAAoB,EAAE,EAAE,WAAW,CAAC,oBAAoB,EAAE,CAAC,CAAC;wBAC5F,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,YAAY,CAAC,CAAC;oBAC/D,CAAC;gBACL,CAAC;YACL,CAAC;YACD,MAAM,CAAC,WAAW,CACd,IAAA,iCAAqB,EACjB,SAAS,EACT,iBAAS,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EACvC,MAAM,EACN,SAAS,EACT,WAAW,mCAEX,CAAC,CACJ,EACD,IAAI,CACP,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,OAAO,CAAC;IACnB,CAAC,CAAC;IACF,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC,CACJ,CAAC"}

package/dist/tests/testState.test.js

@@ -47,6 +47,7 @@ const assert_1 = __importDefault(require("assert"));
 const pathUtils_1 = require("../common/pathUtils");
 const stringUtils_1 = require("../common/stringUtils");
 const uri_1 = require("../common/uri/uri");
+const callHierarchyProvider_1 = require("../languageService/callHierarchyProvider");
 const runner_1 = require("./harness/fourslash/runner");
 const testState_1 = require("./harness/fourslash/testState");
 const factory = __importStar(require("./harness/vfs/factory"));
@@ -451,6 +452,69 @@ helper.getMarkerByName("position");
     `;
     (0, runner_1.runFourSlashTestContent)((0, pathUtils_1.normalizeSlashes)('/'), 'unused.py', code);
 });
+describe('Incoming call hierarchy assertions', () => {
+    function createTestState() {
+        const code = `
+// @filename: declare.py
+//// def /*target*/target():
+////     pass
+
+// @filename: first.py
+//// from declare import target
+////
+//// [|/*firstRange*/def [|/*firstSelection*/first|]():
+////     target()|]
+
+// @filename: second.py
+//// from declare import target
+////
+////
+//// [|/*secondRange*/def [|/*secondSelection*/second|]():
+////     target()|]
+        `;
+        const { state } = (0, testState_1.parseAndGetTestState)(code);
+        const items = ['first', 'second'].map((name) => ({
+            filePath: state.getMappedFilePath(`${name}.py`),
+            name,
+            range: state.getPositionRange(`${name}Range`),
+            selectionRange: state.getPositionRange(`${name}Selection`),
+        }));
+        return { state, items };
+    }
+    test.each([0, 1])('accepts an omitted selectionRange for caller %i', (index) => {
+        const { state, items } = createTestState();
+        const expectations = items.map((item, i) => i === index ? { filePath: item.filePath, name: item.name, range: item.range } : item);
+        state.verifyShowCallHierarchyGetIncomingCalls({ target: { items: expectations } });
+    });
+    test.each(['filePath', 'name', 'range', 'selectionRange'])('rejects %s swapped between callers', (field) => {
+        const { state, items } = createTestState();
+        const expectations = [
+            { ...items[0], [field]: items[1][field] },
+            { ...items[1], [field]: items[0][field] },
+        ];
+        assert_1.default.throws(() => state.verifyShowCallHierarchyGetIncomingCalls({ target: { items: expectations } }), assert_1.default.AssertionError);
+    });
+    test('matches each caller exactly once', () => {
+        const { state, items } = createTestState();
+        state.verifyShowCallHierarchyGetIncomingCalls({ target: { items: [...items].reverse() } });
+        assert_1.default.throws(() => state.verifyShowCallHierarchyGetIncomingCalls({ target: { items: [items[0], items[0]] } }), assert_1.default.AssertionError);
+        assert_1.default.throws(() => state.verifyShowCallHierarchyGetIncomingCalls({ target: { items: [items[0]] } }), assert_1.default.AssertionError);
+        const getIncomingCalls = callHierarchyProvider_1.CallHierarchyProvider.prototype.getIncomingCalls;
+        const spy = jest
+            .spyOn(callHierarchyProvider_1.CallHierarchyProvider.prototype, 'getIncomingCalls')
+            .mockImplementation(function () {
+            const calls = getIncomingCalls.call(this);
+            (0, assert_1.default)(calls && calls.length === 2);
+            return [calls[0], calls[0]];
+        });
+        try {
+            assert_1.default.throws(() => state.verifyShowCallHierarchyGetIncomingCalls({ target: { items } }), assert_1.default.AssertionError);
+        }
+        finally {
+            spy.mockRestore();
+        }
+    });
+});
 test('VerifyDiagnosticsTest1', () => {
     const code = `
 /// <reference path="typings/fourslash.d.ts" />