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

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementBoundary.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementBoundary.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AACjC,2BAAkC;AAClC,+BAA+B;AAE/B,qEAAkE;AAClE,qDAA8D;AAE9D,6CAAuF;AACvF,qDAA0D;AAE1D,6DAAsF;AAEtF,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,4BAA4B,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAC1G,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CACjC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAClC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/E,CACJ,CACJ,CAAC;AAEF,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CACZ,oGAAoG,EACpG,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;IAC9C,IAAI,SAAS,GAAG,IAAA,iBAAY,EAAC,IAAA,cAAO,EAAC,SAAS,EAAE,0CAA0C,CAAC,EAAE,MAAM,CAAC,CAAC;IACrG,IAAI,UAAU,EAAE,CAAC;QACb,MAAM,SAAS,GAAG,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,cAAc,CAAC;QACrF,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,SAAS,GAAG,CAAC,CAAC,CAAC;QACrD,SAAS,GAAG,SAAS,CAAC,OAAO,CACzB,SAAS,SAAS,GAAG,EACrB,SAAS,SAAS,wCAAwC,CAC7D,CAAC;IACN,CAAC;IACD,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;EAEzC,SAAS;SACN,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,IAAI,EAAE,CAAC;SAC7B,IAAI,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uBA6BQ,UAAU;;;;;SAKxB,CAAC,CAAC,KAAK,CAAC;IACT,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAU,CAAC;IAC3C,MAAM,UAAU,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IACpD,MAAM,eAAe,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IAC9D,MAAM,aAAa,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,QAAQ,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAChD,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,QAAQ,8BAAqB,CAAC;IAC5D,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,QAAQ,8BAAqB,CAAC;IACjE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,QAAQ,8BAAqB,CAAC;IAC/D,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,8BAAqB,CAAC;IAC1D,MAAM,MAAM,GAAG,SAAS,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC;IAC9D,MAAM,WAAW,GAAG,SAAS,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC;IACxE,MAAM,SAAS,GAAG,SAAS,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC;IACpE,MAAM,IAAI,GAAG,SAAS,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC;IAC1D,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,MAAM,CAAC,CAAC,CAAC;IACnC,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,WAAW,CAAC,CAAC,CAAC;IACxC,MAAM,CAAC,EAAE,CAAC,IAAA,8BAAsB,EAAC,SAAS,CAAC,CAAC,CAAC;IAC7C,MAAM,UAAU,GAAG,iBAAS,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAC7D,MAAM,cAAc,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACvD,MAAM,MAAM,GAAG,iBAAS,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAC1D,MAAM,MAAM,GAAG,CAAC,OAAuC,EAAE,EAAE,CACvD,IAAA,iCAAqB,EACjB,SAAS,EACT,MAAM,EACN,UAAU,EACV,OAAO,EACP,IAAI,qCAAiB,EAAE,mCAEvB,CAAC,CACJ,CAAC;IACN,IAAI,IAAI,EAAE,CAAC;QACP,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,+BAAkB,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,+BAAkB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,CAAC;IAC3F,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,SAAS,CAAC,EAAE,UAAU,CAAC,CAAC;IACtE,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,cAAc,CAAC,CAAC;AACxE,CAAC,CACJ,CAAC"}

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

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

package/dist/tests/protocolRequirementComplexity.test.js

@@ -0,0 +1,159 @@
+"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 testState_1 = require("./harness/fourslash/testState");
+const TestUtils = __importStar(require("./testUtils"));
+test.each([2, 4, 8, 16])('ProtocolRequirementComplexity overloads=%s', (overloadCount) => {
+    const declarations = (name, count) => Array.from({ length: count }, (_, index) => `//// @overload\n//// def ${name}[T](value: T, tag: Literal[${index}]) -> T: ...`).join('\n') + `\n//// def ${name}(value, tag): return value`;
+    const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: test.py
+//// from collections.abc import Iterator
+//// from typing import Literal, Protocol, TypeVar, overload
+//// T_co = TypeVar("T_co", covariant=True)
+//// class Leaf(Protocol[T_co]):
+////     def serialize(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]"]: ...
+${declarations('first', overloadCount)}
+${declarations('second', overloadCount + 1)}
+//// def check(source: list[object], destination: NestedSequence[Leaf[int]]):
+////     /*first*/first
+////     /*second*/second
+////     /*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');
+    const first = typeAt('first');
+    const second = typeAt('second');
+    assert.ok((0, types_1.isClassInstance)(source) && (0, types_1.isClassInstance)(destination));
+    assert.ok((0, types_1.isFunctionOrOverloaded)(first) && (0, types_1.isFunctionOrOverloaded)(second));
+    const sourceType = types_1.ClassType.specialize(source, [(0, types_1.combineTypes)([first, second])]);
+    const originalType = evaluator.printType(sourceType);
+    assert.ok(originalType.includes('Overload['));
+    const constraints = new constraintTracker_1.ConstraintTracker();
+    constraints.getMainConstraintSet().addScopeId('caller');
+    const originalClone = constraintTracker_1.ConstraintSet.prototype.clone;
+    let clones = 0;
+    const spy = jest.spyOn(constraintTracker_1.ConstraintSet.prototype, 'clone').mockImplementation(function () {
+        clones++;
+        assert.ok(clones <= 64, `overloads=${overloadCount}: exceeded 64 constraint-set clones`);
+        return originalClone.call(this);
+    });
+    try {
+        assert.strictEqual((0, protocols_1.assignClassToProtocol)(evaluator, types_1.ClassType.cloneAsInstantiable(destination), sourceType, undefined, constraints, 0 /* AssignTypeFlags.Default */, 0), false);
+        assert.ok(clones > 0, 'The work counter must observe the assignment');
+        assert.deepStrictEqual(constraints
+            .getConstraintSets()
+            .map((set) => ({ scopes: [...set.getScopeIds()], bounds: set.getTypeVars() })), [{ scopes: ['caller'], bounds: [] }]);
+        assert.strictEqual(evaluator.printType(sourceType), originalType);
+    }
+    finally {
+        spy.mockRestore();
+    }
+});
+test('ProtocolRequirementComplexityPreservesOverloadSelection', () => {
+    const results = TestUtils.typeAnalyzeSampleFiles(['protocolRequirementOverloads.py']);
+    TestUtils.validateResults(results, 0);
+});
+describe.each([2, 4, 6])('ProtocolRequirementComplexity depth=%s', (depth) => {
+    test.each([2, 3])('members=%s', (memberCount) => {
+        for (const compatible of [true, false]) {
+            const declarations = (prefix, protocol) => Array.from({ length: depth }, (_, level) => {
+                const next = `${prefix}${(level + 1) % depth}`;
+                const members = Array.from({ length: memberCount }, (_, member) => `////     def member${member}(self) -> "${next}": ...`).join('\n');
+                const leaf = protocol || compatible ? 'int' : 'str';
+                return (`//// class ${prefix}${level}${protocol ? '(Protocol)' : ''}:\n${members}` +
+                    (level === depth - 1 ? `\n////     def value(self) -> ${leaf}: ...` : ''));
+            }).join('\n');
+            const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: test.py
+//// from typing import Protocol
+${declarations('Target', true)}
+${declarations('Source', false)}
+//// def check(source: Source0, destination: Target0):
+////     /*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 sourceBefore = evaluator.printType(source);
+            const constraints = new constraintTracker_1.ConstraintTracker();
+            constraints.getMainConstraintSet().addScopeId('caller');
+            const originalLookup = evaluator.getDeclaredTypeOfSymbol;
+            const budget = 32 * depth * memberCount;
+            let lookups = 0;
+            const spy = jest.spyOn(evaluator, 'getDeclaredTypeOfSymbol').mockImplementation((...args) => {
+                lookups++;
+                assert.ok(lookups <= budget, `depth=${depth}, members=${memberCount}, compatible=${compatible}: exceeded ${budget} member lookups`);
+                return originalLookup(...args);
+            });
+            try {
+                for (let repetition = 0; repetition < 2; repetition++) {
+                    assert.strictEqual((0, protocols_1.assignClassToProtocol)(evaluator, types_1.ClassType.cloneAsInstantiable(destination), source, undefined, constraints, 0 /* AssignTypeFlags.Default */, 0), compatible);
+                    assert.strictEqual(evaluator.printType(source), sourceBefore);
+                    assert.deepStrictEqual(constraints.getConstraintSets().map((set) => ({
+                        scopes: [...set.getScopeIds()],
+                        bounds: set.getTypeVars(),
+                    })), [{ scopes: ['caller'], bounds: [] }]);
+                }
+                assert.ok(lookups >= depth * memberCount, 'The work counter must observe the recursive walk');
+            }
+            finally {
+                spy.mockRestore();
+            }
+        }
+    });
+});
+//# sourceMappingURL=protocolRequirementComplexity.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementComplexity.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementComplexity.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AAEjC,qEAAiF;AACjF,qDAA8D;AAE9D,6CAAqG;AAErG,6DAAsF;AACtF,uDAAyC;AAEzC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,4CAA4C,EAAE,CAAC,aAAa,EAAE,EAAE;IACrF,MAAM,YAAY,GAAG,CAAC,IAAY,EAAE,KAAa,EAAE,EAAE,CACjD,KAAK,CAAC,IAAI,CACN,EAAE,MAAM,EAAE,KAAK,EAAE,EACjB,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,4BAA4B,IAAI,8BAA8B,KAAK,cAAc,CAClG,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,cAAc,IAAI,4BAA4B,CAAC;IAClE,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;;;;;;EAWrC,YAAY,CAAC,OAAO,EAAE,aAAa,CAAC;EACpC,YAAY,CAAC,QAAQ,EAAE,aAAa,GAAG,CAAC,CAAC;;;;;;KAMtC,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,OAAO,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;IACpD,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChC,MAAM,WAAW,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;IAC1C,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChC,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,MAAM,CAAC,IAAI,IAAA,uBAAe,EAAC,WAAW,CAAC,CAAC,CAAC;IACnE,MAAM,CAAC,EAAE,CAAC,IAAA,8BAAsB,EAAC,KAAK,CAAC,IAAI,IAAA,8BAAsB,EAAC,MAAM,CAAC,CAAC,CAAC;IAC3E,MAAM,UAAU,GAAG,iBAAS,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,IAAA,oBAAY,EAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACjF,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;IAC9C,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;IAC5C,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACxD,MAAM,aAAa,GAAG,iCAAa,CAAC,SAAS,CAAC,KAAK,CAAC;IACpD,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,iCAAa,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC;QACxE,MAAM,EAAE,CAAC;QACT,MAAM,CAAC,EAAE,CAAC,MAAM,IAAI,EAAE,EAAE,aAAa,aAAa,qCAAqC,CAAC,CAAC;QACzF,OAAO,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IACH,IAAI,CAAC;QACD,MAAM,CAAC,WAAW,CACd,IAAA,iCAAqB,EACjB,SAAS,EACT,iBAAS,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAC1C,UAAU,EACV,SAAS,EACT,WAAW,mCAEX,CAAC,CACJ,EACD,KAAK,CACR,CAAC;QACF,MAAM,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,8CAA8C,CAAC,CAAC;QACtE,MAAM,CAAC,eAAe,CAClB,WAAW;aACN,iBAAiB,EAAE;aACnB,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,EAClF,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACvC,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,YAAY,CAAC,CAAC;IACtE,CAAC;YAAS,CAAC;QACP,GAAG,CAAC,WAAW,EAAE,CAAC;IACtB,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yDAAyD,EAAE,GAAG,EAAE;IACjE,MAAM,OAAO,GAAG,SAAS,CAAC,sBAAsB,CAAC,CAAC,iCAAiC,CAAC,CAAC,CAAC;IACtF,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,wCAAwC,EAAE,CAAC,KAAK,EAAE,EAAE;IACzE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,WAAW,EAAE,EAAE;QAC5C,KAAK,MAAM,UAAU,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;YACrC,MAAM,YAAY,GAAG,CAAC,MAAc,EAAE,QAAiB,EAAE,EAAE,CACvD,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE;gBACvC,MAAM,IAAI,GAAG,GAAG,MAAM,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC;gBAC/C,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CACtB,EAAE,MAAM,EAAE,WAAW,EAAE,EACvB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,sBAAsB,MAAM,cAAc,IAAI,QAAQ,CACxE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACb,MAAM,IAAI,GAAG,QAAQ,IAAI,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;gBACpD,OAAO,CACH,cAAc,MAAM,GAAG,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,MAAM,OAAO,EAAE;oBAC1E,CAAC,KAAK,KAAK,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,iCAAiC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAC5E,CAAC;YACN,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClB,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;EAG7C,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC;EAC5B,YAAY,CAAC,QAAQ,EAAE,KAAK,CAAC;;;;aAIlB,CAAC,CAAC,KAAK,CAAC;YACT,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,OAAO,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;YACpD,CAAC,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,WAAW,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;YAC1C,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,MAAM,CAAC,IAAI,IAAA,uBAAe,EAAC,WAAW,CAAC,CAAC,CAAC;YACnE,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACjD,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;YAC5C,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACxD,MAAM,cAAc,GAAG,SAAS,CAAC,uBAAuB,CAAC;YACzD,MAAM,MAAM,GAAG,EAAE,GAAG,KAAK,GAAG,WAAW,CAAC;YACxC,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAI,EAAE,EAAE;gBACxF,OAAO,EAAE,CAAC;gBACV,MAAM,CAAC,EAAE,CACL,OAAO,IAAI,MAAM,EACjB,SAAS,KAAK,aAAa,WAAW,gBAAgB,UAAU,cAAc,MAAM,iBAAiB,CACxG,CAAC;gBACF,OAAO,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;YACnC,CAAC,CAAC,CAAC;YACH,IAAI,CAAC;gBACD,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC;oBACpD,MAAM,CAAC,WAAW,CACd,IAAA,iCAAqB,EACjB,SAAS,EACT,iBAAS,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAC1C,MAAM,EACN,SAAS,EACT,WAAW,mCAEX,CAAC,CACJ,EACD,UAAU,CACb,CAAC;oBACF,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;oBAC9D,MAAM,CAAC,eAAe,CAClB,WAAW,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1C,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;wBAC9B,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE;qBAC5B,CAAC,CAAC,EACH,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACvC,CAAC;gBACN,CAAC;gBACD,MAAM,CAAC,EAAE,CAAC,OAAO,IAAI,KAAK,GAAG,WAAW,EAAE,kDAAkD,CAAC,CAAC;YAClG,CAAC;oBAAS,CAAC;gBACP,GAAG,CAAC,WAAW,EAAE,CAAC;YACtB,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

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

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

package/dist/tests/protocolRequirementCycles.test.js

@@ -0,0 +1,160 @@
+"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(['RecursiveList', 'SelfOnly', 'MutualLeft'])('ProtocolRequirementCycles %s', (element) => {
+    test.each(['shortcut', 'cold', 'diagnostic'])('%s', (mode) => {
+        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 Leaf(Protocol[T_co]):
+////     def serialize(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]"]: ...
+//// type RecursiveList = list[RecursiveList]
+//// class SelfOnly:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> "SelfOnly": ...
+////     def __iter__(self, /) -> Iterator["SelfOnly"]: ...
+//// class MutualLeft:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> "MutualRight": ...
+////     def __iter__(self, /) -> Iterator["MutualRight"]: ...
+//// class MutualRight:
+////     def __len__(self, /) -> int: ...
+////     def __getitem__(self, index: int, /) -> MutualLeft: ...
+////     def __iter__(self, /) -> Iterator[MutualLeft]: ...
+//// def check(source: list[${element}], destination: NestedSequence[Leaf[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), true);
+            }
+            assert.strictEqual(evaluator.printType(source), sourceBefore);
+            assert.deepStrictEqual(constraints.getConstraintSets().map((set) => ({
+                scopes: [...set.getScopeIds()],
+                bounds: set.getTypeVars(),
+            })), [{ scopes: ['caller'], bounds: [] }]);
+        }
+    });
+});
+test.each([false, true])('ProtocolRequirementFailedRecursiveAncestor speculative=%s', (speculative) => {
+    const run = (warm) => {
+        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]"]: ...
+//// type RecursiveList = list[RecursiveList | float]
+//// def check(outer: list[RecursiveList], inner: RecursiveList, bad: NestedSequence[int], good: NestedSequence[float]):
+////     /*outer*/outer
+////     /*inner*/inner
+////     /*bad*/bad
+////     /*good*/good
+        `).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 outer = typeAt('outer');
+        const inner = typeAt('inner');
+        const bad = types_1.ClassType.cloneAsInstantiable(typeAt('bad'));
+        const good = types_1.ClassType.cloneAsInstantiable(typeAt('good'));
+        const constraints = new constraintTracker_1.ConstraintTracker();
+        constraints.getMainConstraintSet().addScopeId('caller');
+        const before = [evaluator.printType(outer), evaluator.printType(inner)];
+        const assign = (destination, source, diagnostic) => (0, protocols_1.assignClassToProtocol)(evaluator, destination, source, diagnostic, constraints, 0 /* AssignTypeFlags.Default */, 0);
+        if (warm) {
+            const reject = () => assert.strictEqual(assign(bad, outer), false);
+            if (speculative) {
+                evaluator.useSpeculativeMode((0, testState_1.getNodeAtMarker)(state, 'outer'), reject);
+            }
+            else {
+                reject();
+            }
+        }
+        assert.strictEqual(evaluator.isSpeculativeModeInUse(undefined), false);
+        for (let repetition = 0; repetition < 2; repetition++) {
+            assert.strictEqual(assign(bad, inner), false);
+            assert.strictEqual(assign(good, inner), true);
+            assert.strictEqual(assign(good, outer), true);
+        }
+        const diagnostic = new diagnostic_1.DiagnosticAddendum();
+        assert.strictEqual(assign(bad, inner, diagnostic), false);
+        assert.ok(diagnostic.getString().length > 0);
+        assert.deepStrictEqual([evaluator.printType(outer), evaluator.printType(inner)], before);
+        assert.deepStrictEqual(constraints
+            .getConstraintSets()
+            .map((set) => ({ scopes: [...set.getScopeIds()], bounds: set.getTypeVars() })), [{ scopes: ['caller'], bounds: [] }]);
+        return diagnostic.getString();
+    };
+    assert.strictEqual(run(true), run(false));
+});
+//# sourceMappingURL=protocolRequirementCycles.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementCycles.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementCycles.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AAEjC,qEAAkE;AAClE,qDAA6F;AAE7F,6CAA+D;AAC/D,qDAA0D;AAE1D,6DAAsF;AAEtF,QAAQ,CAAC,IAAI,CAAC,CAAC,eAAe,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC,8BAA8B,EAAE,CAAC,OAAO,EAAE,EAAE;IACnG,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;QACzD,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;;;;;;;;;;;;;;;;;;;8BAwBb,OAAO;;;SAG5B,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,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,IAAI,CACP,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;AAEH,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,2DAA2D,EAAE,CAAC,WAAW,EAAE,EAAE;IAClG,MAAM,GAAG,GAAG,CAAC,IAAa,EAAE,EAAE;QAC1B,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;;;;;;;;;;;;;;SAelC,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,MAAM,IAAI,GAAG,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;YACtD,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,IAAI,CAAC,CAAC,CAAC;YACjC,OAAO,IAAI,CAAC;QAChB,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,GAAG,GAAG,iBAAS,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACzD,MAAM,IAAI,GAAG,iBAAS,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3D,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;QAC5C,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,CAAC,WAAsB,EAAE,MAAiB,EAAE,UAA+B,EAAW,EAAE,CACnG,IAAA,iCAAqB,EAAC,SAAS,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,mCAA2B,CAAC,CAAC,CAAC;QAC/G,IAAI,IAAI,EAAE,CAAC;YACP,MAAM,MAAM,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;YACnE,IAAI,WAAW,EAAE,CAAC;gBACd,SAAS,CAAC,kBAAkB,CAAC,IAAA,2BAAe,EAAC,KAAK,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;YAC1E,CAAC;iBAAM,CAAC;gBACJ,MAAM,EAAE,CAAC;YACb,CAAC;QACL,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,sBAAsB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;QACvE,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC;YACpD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;YAC9C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,+BAAkB,EAAE,CAAC;QAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;QAC1D,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,eAAe,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QACzF,MAAM,CAAC,eAAe,CAClB,WAAW;aACN,iBAAiB,EAAE;aACnB,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,EAClF,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACvC,CAAC;QACF,OAAO,UAAU,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC,CAAC;IACF,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;AAC9C,CAAC,CAAC,CAAC"}

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

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

package/dist/tests/protocolRequirementGeneral.test.js

@@ -0,0 +1,120 @@
+"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 fs_1 = require("fs");
+const path_1 = require("path");
+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 = ['serialize', 'read', '__array__'].flatMap((member) => ['missing', 'leaf', 'sequence'].flatMap((capability) => ['shortcut', 'cold', 'diagnostic'].map((mode) => ({ member, capability, mode }))));
+cases.push(...['shortcut', 'cold', 'diagnostic'].map((mode) => ({ member: '__call__', capability: 'callback', mode })));
+test.each(cases)('ProtocolRequirementGeneral $member $capability $mode', ({ member, capability, mode }) => {
+    let typesStub = (0, fs_1.readFileSync)((0, path_1.resolve)(__dirname, '../../typeshed-fallback/stdlib/types.pyi'), 'utf8');
+    assert.ok(typesStub.includes('class FunctionType:'));
+    if (capability === 'leaf') {
+        typesStub = typesStub.replace('class FunctionType:', `class FunctionType:\n    def ${member}(self) -> int: ...`);
+    }
+    else if (capability === 'sequence') {
+        typesStub = typesStub.replace('class FunctionType:', 'class FunctionType:\n    def __len__(self) -> int: ...\n' +
+            '    def __getitem__(self, index: int, /) -> Any: ...\n' +
+            '    def __iter__(self) -> Iterator[Any]: ...');
+    }
+    const state = (0, testState_1.parseAndGetTestState)(`
+// @filename: /typeshed-fallback/stdlib/types.pyi
+${typesStub
+        .split('\n')
+        .map((line) => `//// ${line}`)
+        .join('\n')}
+// @filename: test.py
+//// from collections.abc import Iterator
+//// from typing import 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: ...
+////     def __getitem__(self, index: int, /) -> T_co | "NestedSequence[T_co]": ...
+////     def __iter__(self, /) -> Iterator[T_co | "NestedSequence[T_co]"]: ...
+//// def operation() -> int:
+////     return 1
+//// def check[T](variable: T, destination: NestedSequence[Leaf[T]]):
+////     /*operation*/operation
+////     /*source*/[operation]
+////     /*destination*/destination
+////     /*variable*/variable
+    `).state;
+    const evaluator = state.program.evaluator;
+    const sourceNode = (0, testState_1.getNodeAtMarker)(state, 'source');
+    const destinationNode = (0, testState_1.getNodeAtMarker)(state, 'destination');
+    const operationNode = (0, testState_1.getNodeAtMarker)(state, 'operation');
+    const variableNode = (0, testState_1.getNodeAtMarker)(state, 'variable');
+    assert.strictEqual(sourceNode.nodeType, 34 /* ParseNodeType.List */);
+    assert.strictEqual(destinationNode.nodeType, 38 /* ParseNodeType.Name */);
+    assert.strictEqual(operationNode.nodeType, 38 /* ParseNodeType.Name */);
+    assert.strictEqual(variableNode.nodeType, 38 /* ParseNodeType.Name */);
+    const source = evaluator.getTypeOfExpression(sourceNode).type;
+    const destination = evaluator.getTypeOfExpression(destinationNode).type;
+    const operation = evaluator.getTypeOfExpression(operationNode).type;
+    const variable = evaluator.getTypeOfExpression(variableNode).type;
+    assert.ok((0, types_1.isClassInstance)(source));
+    assert.ok((0, types_1.isClassInstance)(destination));
+    assert.ok((0, types_1.isFunction)(operation));
+    assert.ok((0, types_1.isTypeVar)(variable));
+    const freeVariable = variable.priv.freeTypeVar;
+    assert.ok(freeVariable?.priv.scopeId);
+    const target = (0, typeUtils_1.makeTypeVarsFree)(types_1.ClassType.cloneAsInstantiable(destination), [freeVariable.priv.scopeId]);
+    const sourceType = types_1.ClassType.specialize(source, [operation]);
+    const constraints = new constraintTracker_1.ConstraintTracker();
+    constraints.getMainConstraintSet().addScopeId('caller');
+    if (mode === 'shortcut') {
+        assert.strictEqual((0, protocols_1.tryFastRejectSequenceProtocol)(evaluator, target, sourceType, constraints, 0 /* AssignTypeFlags.Default */, 0), capability === 'missing' ? '__getitem__' : undefined);
+    }
+    else {
+        assert.strictEqual((0, protocols_1.assignClassToProtocol)(evaluator, target, sourceType, mode === 'diagnostic' ? new diagnostic_1.DiagnosticAddendum() : undefined, constraints, 0 /* AssignTypeFlags.Default */, 0), capability !== 'missing');
+        if (capability === 'leaf' || capability === 'callback') {
+            assert.strictEqual(evaluator.printType(evaluator.solveAndApplyConstraints(freeVariable, constraints)), 'int');
+        }
+    }
+    if (mode === 'shortcut' || capability === 'missing') {
+        expect(constraints
+            .getConstraintSets()
+            .map((set) => ({ scopes: [...set.getScopeIds()], bindings: set.getTypeVars() }))).toEqual([{ scopes: ['caller'], bindings: [] }]);
+    }
+});
+//# sourceMappingURL=protocolRequirementGeneral.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"protocolRequirementGeneral.test.js","sourceRoot":"","sources":["../../../../../src/tests/protocolRequirementGeneral.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AACjC,2BAAkC;AAClC,+BAA+B;AAE/B,qEAAkE;AAClE,qDAA6F;AAE7F,6CAAsF;AACtF,qDAAyD;AACzD,qDAA0D;AAE1D,6DAAsF;AAEtF,MAAM,KAAK,GAAG,CAAC,WAAW,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAChE,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CACnD,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CACnF,CACJ,CAAC;AAEF,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AAExH,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,sDAAsD,EAAE,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE;IACtG,IAAI,SAAS,GAAG,IAAA,iBAAY,EAAC,IAAA,cAAO,EAAC,SAAS,EAAE,0CAA0C,CAAC,EAAE,MAAM,CAAC,CAAC;IACrG,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,CAAC;IACrD,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;QACxB,SAAS,GAAG,SAAS,CAAC,OAAO,CACzB,qBAAqB,EACrB,gCAAgC,MAAM,oBAAoB,CAC7D,CAAC;IACN,CAAC;SAAM,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;QACnC,SAAS,GAAG,SAAS,CAAC,OAAO,CACzB,qBAAqB,EACrB,0DAA0D;YACtD,wDAAwD;YACxD,8CAA8C,CACrD,CAAC;IACN,CAAC;IACD,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC;;EAErC,SAAS;SACN,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,IAAI,EAAE,CAAC;SAC7B,IAAI,CAAC,IAAI,CAAC;;;;;;eAMA,MAAM;;;;;;;;;;;;KAYhB,CAAC,CAAC,KAAK,CAAC;IACT,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAU,CAAC;IAC3C,MAAM,UAAU,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IACpD,MAAM,eAAe,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IAC9D,MAAM,aAAa,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,YAAY,GAAG,IAAA,2BAAe,EAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACxD,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,QAAQ,8BAAqB,CAAC;IAC5D,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,QAAQ,8BAAqB,CAAC;IACjE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,QAAQ,8BAAqB,CAAC;IAC/D,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,QAAQ,8BAAqB,CAAC;IAC9D,MAAM,MAAM,GAAG,SAAS,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC;IAC9D,MAAM,WAAW,GAAG,SAAS,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC;IACxE,MAAM,SAAS,GAAG,SAAS,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC;IACpE,MAAM,QAAQ,GAAG,SAAS,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC;IAClE,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,MAAM,CAAC,CAAC,CAAC;IACnC,MAAM,CAAC,EAAE,CAAC,IAAA,uBAAe,EAAC,WAAW,CAAC,CAAC,CAAC;IACxC,MAAM,CAAC,EAAE,CAAC,IAAA,kBAAU,EAAC,SAAS,CAAC,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,CAAC,IAAA,iBAAS,EAAC,QAAQ,CAAC,CAAC,CAAC;IAC/B,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC;IAC/C,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,MAAM,GAAG,IAAA,4BAAgB,EAAC,iBAAS,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IACzG,MAAM,UAAU,GAAG,iBAAS,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAC7D,MAAM,WAAW,GAAG,IAAI,qCAAiB,EAAE,CAAC;IAC5C,WAAW,CAAC,oBAAoB,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACxD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;QACtB,MAAM,CAAC,WAAW,CACd,IAAA,yCAA6B,EAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,mCAA2B,CAAC,CAAC,EACrG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CACvD,CAAC;IACN,CAAC;SAAM,CAAC;QACJ,MAAM,CAAC,WAAW,CACd,IAAA,iCAAqB,EACjB,SAAS,EACT,MAAM,EACN,UAAU,EACV,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,+BAAkB,EAAE,CAAC,CAAC,CAAC,SAAS,EAC5D,WAAW,mCAEX,CAAC,CACJ,EACD,UAAU,KAAK,SAAS,CAC3B,CAAC;QACF,IAAI,UAAU,KAAK,MAAM,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;YACrD,MAAM,CAAC,WAAW,CACd,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,wBAAwB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,EAClF,KAAK,CACR,CAAC;QACN,CAAC;IACL,CAAC;IACD,IAAI,IAAI,KAAK,UAAU,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAClD,MAAM,CACF,WAAW;aACN,iBAAiB,EAAE;aACnB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CACvF,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IACtD,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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