@zzzen/pyright-internal 1.2.0-dev.20260823 → 1.2.0-dev.20260906

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (149) hide show
  1. package/dist/analyzer/analyzerNodeInfo.d.ts +4 -0
  2. package/dist/analyzer/analyzerNodeInfo.js +29 -8
  3. package/dist/analyzer/analyzerNodeInfo.js.map +1 -1
  4. package/dist/analyzer/backgroundAnalysisProgram.d.ts +2 -1
  5. package/dist/analyzer/backgroundAnalysisProgram.js +4 -8
  6. package/dist/analyzer/backgroundAnalysisProgram.js.map +1 -1
  7. package/dist/analyzer/checker.js +31 -25
  8. package/dist/analyzer/checker.js.map +1 -1
  9. package/dist/analyzer/constructorTransform.js +8 -0
  10. package/dist/analyzer/constructorTransform.js.map +1 -1
  11. package/dist/analyzer/dataClasses.js +9 -3
  12. package/dist/analyzer/dataClasses.js.map +1 -1
  13. package/dist/analyzer/parseTreeUtils.js +12 -13
  14. package/dist/analyzer/parseTreeUtils.js.map +1 -1
  15. package/dist/analyzer/program.js +4 -1
  16. package/dist/analyzer/program.js.map +1 -1
  17. package/dist/analyzer/sentinel.d.ts +2 -2
  18. package/dist/analyzer/sentinel.js +3 -3
  19. package/dist/analyzer/sentinel.js.map +1 -1
  20. package/dist/analyzer/service.d.ts +8 -11
  21. package/dist/analyzer/service.js +27 -96
  22. package/dist/analyzer/service.js.map +1 -1
  23. package/dist/analyzer/sourceFile.js +1 -0
  24. package/dist/analyzer/sourceFile.js.map +1 -1
  25. package/dist/analyzer/sourceFileInfo.d.ts +1 -0
  26. package/dist/analyzer/sourceFileInfo.js +3 -0
  27. package/dist/analyzer/sourceFileInfo.js.map +1 -1
  28. package/dist/analyzer/typeEvaluator.js +93 -32
  29. package/dist/analyzer/typeEvaluator.js.map +1 -1
  30. package/dist/analyzer/typeGuards.js +133 -10
  31. package/dist/analyzer/typeGuards.js.map +1 -1
  32. package/dist/analyzer/typeStubGeneration.d.ts +64 -0
  33. package/dist/analyzer/typeStubGeneration.js +226 -0
  34. package/dist/analyzer/typeStubGeneration.js.map +1 -0
  35. package/dist/analyzer/typeStubMessages.d.ts +4 -0
  36. package/dist/analyzer/typeStubMessages.js +25 -0
  37. package/dist/analyzer/typeStubMessages.js.map +1 -0
  38. package/dist/analyzer/typeStubOutput.d.ts +9 -0
  39. package/dist/analyzer/typeStubOutput.js +52 -0
  40. package/dist/analyzer/typeStubOutput.js.map +1 -0
  41. package/dist/analyzer/typeStubRenderer.d.ts +5 -0
  42. package/dist/analyzer/{typeStubWriter.js → typeStubRenderer.js} +11 -93
  43. package/dist/analyzer/typeStubRenderer.js.map +1 -0
  44. package/dist/analyzer/typeUtils.d.ts +1 -0
  45. package/dist/analyzer/typeUtils.js +112 -0
  46. package/dist/analyzer/typeUtils.js.map +1 -1
  47. package/dist/analyzer/types.d.ts +2 -0
  48. package/dist/analyzer/types.js +84 -58
  49. package/dist/analyzer/types.js.map +1 -1
  50. package/dist/backgroundAnalysisBase.d.ts +5 -4
  51. package/dist/backgroundAnalysisBase.js +11 -19
  52. package/dist/backgroundAnalysisBase.js.map +1 -1
  53. package/dist/commands/createTypeStub.d.ts +2 -6
  54. package/dist/commands/createTypeStub.js +14 -27
  55. package/dist/commands/createTypeStub.js.map +1 -1
  56. package/dist/common/commandLineOptions.d.ts +0 -1
  57. package/dist/common/commandLineOptions.js.map +1 -1
  58. package/dist/common/extensibility.d.ts +1 -0
  59. package/dist/common/extensibility.js.map +1 -1
  60. package/dist/languageServerBase.d.ts +1 -1
  61. package/dist/languageServerBase.js +2 -2
  62. package/dist/languageServerBase.js.map +1 -1
  63. package/dist/languageService/analyzerServiceExecutor.d.ts +1 -3
  64. package/dist/languageService/analyzerServiceExecutor.js +8 -28
  65. package/dist/languageService/analyzerServiceExecutor.js.map +1 -1
  66. package/dist/languageService/fileWatcherDynamicFeature.d.ts +6 -4
  67. package/dist/languageService/fileWatcherDynamicFeature.js +19 -15
  68. package/dist/languageService/fileWatcherDynamicFeature.js.map +1 -1
  69. package/dist/languageService/hoverProvider.d.ts +1 -1
  70. package/dist/languageService/hoverProvider.js +13 -8
  71. package/dist/languageService/hoverProvider.js.map +1 -1
  72. package/dist/languageService/tooltipUtils.d.ts +2 -1
  73. package/dist/languageService/tooltipUtils.js +15 -17
  74. package/dist/languageService/tooltipUtils.js.map +1 -1
  75. package/dist/localization/localize.d.ts +1 -0
  76. package/dist/localization/localize.js +1 -0
  77. package/dist/localization/localize.js.map +1 -1
  78. package/dist/localization/package.nls.cs.json +4 -0
  79. package/dist/localization/package.nls.de.json +4 -0
  80. package/dist/localization/package.nls.en-us.json +1 -0
  81. package/dist/localization/package.nls.es.json +4 -0
  82. package/dist/localization/package.nls.fr.json +4 -0
  83. package/dist/localization/package.nls.it.json +4 -0
  84. package/dist/localization/package.nls.ja.json +4 -0
  85. package/dist/localization/package.nls.ko.json +4 -0
  86. package/dist/localization/package.nls.pl.json +4 -0
  87. package/dist/localization/package.nls.pt-br.json +4 -0
  88. package/dist/localization/package.nls.qps-ploc.json +4 -0
  89. package/dist/localization/package.nls.ru.json +4 -0
  90. package/dist/localization/package.nls.tr.json +4 -0
  91. package/dist/localization/package.nls.zh-cn.json +4 -0
  92. package/dist/localization/package.nls.zh-tw.json +4 -0
  93. package/dist/parser/parser.js +7 -1
  94. package/dist/parser/parser.js.map +1 -1
  95. package/dist/parser/tokenizer.js +3 -5
  96. package/dist/parser/tokenizer.js.map +1 -1
  97. package/dist/pyright.js +30 -17
  98. package/dist/pyright.js.map +1 -1
  99. package/dist/tests/analyzerNodeInfo.test.js +32 -0
  100. package/dist/tests/analyzerNodeInfo.test.js.map +1 -1
  101. package/dist/tests/config.test.js +0 -1
  102. package/dist/tests/config.test.js.map +1 -1
  103. package/dist/tests/fourslash/hover.builtinDocstrings.fourslash.js +5 -5
  104. package/dist/tests/fourslash/hover.docFromSrc.stubs-package.fourslash.js +0 -1
  105. package/dist/tests/fourslash/hover.docFromSrc.stubs-package.fourslash.js.map +1 -1
  106. package/dist/tests/fourslash/hover.fourslash.js +1 -1
  107. package/dist/tests/fourslash/hover.import.django.view.fourslash.js +1 -1
  108. package/dist/tests/fourslash/hover.plainText.fourslash.js +1 -1
  109. package/dist/tests/hoverProvider.test.js +2 -2
  110. package/dist/tests/parseTreeUtils.test.js +41 -0
  111. package/dist/tests/parseTreeUtils.test.js.map +1 -1
  112. package/dist/tests/pyrightCli.test.d.ts +1 -0
  113. package/dist/tests/pyrightCli.test.js +205 -0
  114. package/dist/tests/pyrightCli.test.js.map +1 -0
  115. package/dist/tests/service.test.js +152 -0
  116. package/dist/tests/service.test.js.map +1 -1
  117. package/dist/tests/typeEquality.test.d.ts +1 -0
  118. package/dist/tests/typeEquality.test.js +95 -0
  119. package/dist/tests/typeEquality.test.js.map +1 -0
  120. package/dist/tests/typeEvaluator1.test.js +11 -0
  121. package/dist/tests/typeEvaluator1.test.js.map +1 -1
  122. package/dist/tests/typeEvaluator2.test.js +4 -0
  123. package/dist/tests/typeEvaluator2.test.js.map +1 -1
  124. package/dist/tests/typeEvaluator3.test.js +8 -0
  125. package/dist/tests/typeEvaluator3.test.js.map +1 -1
  126. package/dist/tests/typeEvaluator4.test.js +4 -0
  127. package/dist/tests/typeEvaluator4.test.js.map +1 -1
  128. package/dist/tests/typeEvaluator5.test.js +1 -1
  129. package/dist/tests/typeEvaluator6.test.js +5 -1
  130. package/dist/tests/typeEvaluator6.test.js.map +1 -1
  131. package/dist/tests/typeEvaluator8.test.js +4 -0
  132. package/dist/tests/typeEvaluator8.test.js.map +1 -1
  133. package/dist/tests/typeServer/stubGenerator.test.d.ts +1 -0
  134. package/dist/tests/typeServer/stubGenerator.test.js +73 -0
  135. package/dist/tests/typeServer/stubGenerator.test.js.map +1 -0
  136. package/dist/tests/typeStubGeneration.test.d.ts +1 -0
  137. package/dist/tests/typeStubGeneration.test.js +397 -0
  138. package/dist/tests/typeStubGeneration.test.js.map +1 -0
  139. package/dist/tests/typeStubOutput.test.d.ts +1 -0
  140. package/dist/tests/typeStubOutput.test.js +136 -0
  141. package/dist/tests/typeStubOutput.test.js.map +1 -0
  142. package/dist/tests/typeUtils.test.d.ts +1 -0
  143. package/dist/tests/typeUtils.test.js +133 -0
  144. package/dist/tests/typeUtils.test.js.map +1 -0
  145. package/dist/typeServer/stubGenerator.js +32 -7
  146. package/dist/typeServer/stubGenerator.js.map +1 -1
  147. package/package.json +1 -1
  148. package/dist/analyzer/typeStubWriter.d.ts +0 -17
  149. package/dist/analyzer/typeStubWriter.js.map +0 -1
@@ -0,0 +1,397 @@
1
+ "use strict";
2
+ /*
3
+ * typeStubGeneration.test.ts
4
+ * Copyright (c) Microsoft Corporation.
5
+ * Licensed under the MIT license.
6
+ *
7
+ * Tests for declarative type stub generation.
8
+ */
9
+ var __importDefault = (this && this.__importDefault) || function (mod) {
10
+ return (mod && mod.__esModule) ? mod : { "default": mod };
11
+ };
12
+ Object.defineProperty(exports, "__esModule", { value: true });
13
+ const assert_1 = __importDefault(require("assert"));
14
+ const vscode_languageserver_1 = require("vscode-languageserver");
15
+ const typeStubGeneration_1 = require("../analyzer/typeStubGeneration");
16
+ const cancellationUtils_1 = require("../common/cancellationUtils");
17
+ const uri_1 = require("../common/uri/uri");
18
+ const testState_1 = require("./harness/fourslash/testState");
19
+ test('type stub generation returns files without writing to the file system', async () => {
20
+ const code = `
21
+ // @filename: sample.py
22
+ //// def answer():
23
+ //// return 42
24
+ `;
25
+ const state = (0, testState_1.parseAndGetTestState)(code, '/projectRoot').state;
26
+ const sourceUri = uri_1.Uri.file('/projectRoot/sample.py', state.serviceProvider);
27
+ const outputPath = uri_1.Uri.file('/projectRoot/typings/sample', state.serviceProvider);
28
+ const stubUri = uri_1.Uri.file('/projectRoot/typings/sample/__init__.pyi', state.serviceProvider);
29
+ const result = await state.workspace.service.backgroundAnalysisProgram.generateTypeStubFiles({
30
+ source: {
31
+ targetImportPath: sourceUri,
32
+ targetIsSingleFile: true,
33
+ outputPath,
34
+ },
35
+ additionalFiles: [],
36
+ }, vscode_languageserver_1.CancellationToken.None);
37
+ assert_1.default.strictEqual(state.workspace.service.fs.existsSync(stubUri), false);
38
+ assert_1.default.strictEqual(result.files.length, 1);
39
+ assert_1.default.strictEqual(result.files[0].uri.toString(), stubUri.toString());
40
+ assert_1.default.strictEqual(result.files[0].kind, 'stub');
41
+ assert_1.default.match(result.files[0].contents, /def answer\(\): # -> Literal\[42\]:/);
42
+ });
43
+ test.each([
44
+ {
45
+ title: 'single-file module',
46
+ fileName: 'sample.py',
47
+ expectedTargetSuffix: 'site-packages',
48
+ targetIsSingleFile: true,
49
+ },
50
+ {
51
+ title: 'package',
52
+ fileName: 'sample/__init__.py',
53
+ expectedTargetSuffix: 'site-packages/sample',
54
+ targetIsSingleFile: false,
55
+ },
56
+ ])('target resolution finds a $title without service-specific state', ({ fileName, expectedTargetSuffix, targetIsSingleFile }) => {
57
+ const state = (0, testState_1.parseAndGetTestState)(`
58
+ // @filename: ${fileName}
59
+ // @library: true
60
+ //// def answer(): return 42
61
+ `, '/projectRoot').state;
62
+ const service = state.workspace.service;
63
+ const target = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'sample');
64
+ assert_1.default.ok(target.targetImportPath.pathEndsWith(expectedTargetSuffix));
65
+ assert_1.default.strictEqual(target.targetIsSingleFile, targetIsSingleFile);
66
+ assert_1.default.strictEqual(target.sourceFileUris.length, 1);
67
+ assert_1.default.ok(target.sourceFileUris[0].pathEndsWith(fileName));
68
+ assert_1.default.ok(target.outputPath.pathEndsWith('typings/sample'));
69
+ });
70
+ test('target resolution honors the effective custom stub path', () => {
71
+ const state = (0, testState_1.parseAndGetTestState)(`
72
+ // @filename: pyrightconfig.json
73
+ //// { "stubPath": "custom-stubs" }
74
+ // @filename: sample.py
75
+ // @library: true
76
+ //// def answer(): return 42
77
+ `, '/projectRoot').state;
78
+ const service = state.workspace.service;
79
+ const target = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'sample');
80
+ const effectiveStubPath = service.getConfigOptions().stubPath;
81
+ assert_1.default.strictEqual(target.stubPath.toString(), effectiveStubPath.toString());
82
+ assert_1.default.strictEqual(target.outputPath.toString(), effectiveStubPath.resolvePaths('sample').toString());
83
+ assert_1.default.notStrictEqual(target.stubPath.toString(), service.getConfigOptions().projectRoot.resolvePaths('typings').toString());
84
+ });
85
+ test('target resolution uses the source file execution environment', () => {
86
+ const state = (0, testState_1.parseAndGetTestState)(`
87
+ // @filename: pyrightconfig.json
88
+ //// {
89
+ //// "executionEnvironments": [
90
+ //// { "root": "nested", "extraPaths": ["nested-extra"] },
91
+ //// { "root": ".", "extraPaths": ["root-extra"] }
92
+ //// ]
93
+ //// }
94
+ // @filename: nested/main.py
95
+ //// import sample
96
+ // @filename: nested-extra/sample/__init__.py
97
+ //// def nested_answer(): return 1
98
+ // @filename: root-extra/sample/__init__.py
99
+ //// def root_answer(): return 2
100
+ `).state;
101
+ const service = state.workspace.service;
102
+ const sourceFileUri = uri_1.Uri.file('/nested/main.py', state.serviceProvider);
103
+ const defaultTarget = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'sample');
104
+ const sourceTarget = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'sample', sourceFileUri);
105
+ assert_1.default.strictEqual(defaultTarget.targetImportPath.toString(), 'file:///root-extra/sample');
106
+ assert_1.default.strictEqual(sourceTarget.targetImportPath.toString(), 'file:///nested-extra/sample');
107
+ assert_1.default.deepStrictEqual(sourceTarget.sourceFileUris.map((uri) => uri.toString()), ['file:///nested-extra/sample/__init__.py']);
108
+ });
109
+ test('target resolution reconstructs a dotted full target from the root package', () => {
110
+ const state = (0, testState_1.parseAndGetTestState)(`
111
+ // @filename: sample/__init__.py
112
+ // @library: true
113
+ //// # package marker
114
+ // @filename: sample/sub/__init__.py
115
+ // @library: true
116
+ //// # package marker
117
+ // @filename: sample/sub/core.py
118
+ // @library: true
119
+ //// def answer(): return 42
120
+ `, '/projectRoot').state;
121
+ const service = state.workspace.service;
122
+ const target = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'sample.sub.core');
123
+ assert_1.default.strictEqual(target.targetIsSingleFile, false);
124
+ assert_1.default.ok(target.targetImportPath.pathEndsWith('site-packages/sample'));
125
+ assert_1.default.strictEqual(target.sourceFileUris.length, 1);
126
+ assert_1.default.ok(target.sourceFileUris[0].pathEndsWith('sample/sub/core.py'));
127
+ assert_1.default.ok(target.outputPath.pathEndsWith('typings/sample'));
128
+ assert_1.default.strictEqual(target.targetImportPath.getRelativePath(target.sourceFileUris[0]), './sub/core.py');
129
+ });
130
+ test('target resolution starts at the first concrete package below namespace ancestors', () => {
131
+ const state = (0, testState_1.parseAndGetTestState)(`
132
+ // @filename: testLib/aa/bb/__init__.py
133
+ // @library: true
134
+ //// # package marker
135
+ // @filename: testLib/aa/bb/cc.py
136
+ // @library: true
137
+ //// value = 1
138
+ // @filename: testLib/aa/sibling.py
139
+ // @library: true
140
+ //// sibling_value = 2
141
+ // @filename: testLib/root_sibling.py
142
+ // @library: true
143
+ //// root_sibling_value = 3
144
+ `, '/projectRoot').state;
145
+ const service = state.workspace.service;
146
+ const target = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'testLib.aa.bb');
147
+ const sourceFileUris = (0, typeStubGeneration_1.collectTypeStubSourceFileUris)(service.fs, target, vscode_languageserver_1.CancellationToken.None);
148
+ assert_1.default.ok(target.targetImportPath.pathEndsWith('site-packages/testLib/aa/bb'));
149
+ assert_1.default.ok(target.outputPath.pathEndsWith('typings/testLib/aa/bb'));
150
+ assert_1.default.strictEqual(target.targetIsSingleFile, false);
151
+ assert_1.default.deepStrictEqual(sourceFileUris.map((uri) => target.targetImportPath.getRelativePath(uri)), ['./__init__.py', './cc.py']);
152
+ });
153
+ test('full generation of a concrete module below namespace ancestors excludes sibling modules', async () => {
154
+ const state = (0, testState_1.parseAndGetTestState)(`
155
+ // @filename: nspkg/sub/core.py
156
+ // @library: true
157
+ //// def answer(): return 42
158
+ // @filename: nspkg/sub/sibling.py
159
+ // @library: true
160
+ //// sibling_value = 1
161
+ // @filename: unrelated/__init__.py
162
+ // @library: true
163
+ //// unrelated_value = 2
164
+ `, '/projectRoot').state;
165
+ const service = state.workspace.service;
166
+ const result = await (0, typeStubGeneration_1.runTypeStubGeneration)(() => service.clone('namespace module stub generation test', 'namespace module stub generation test'), { kind: 'full', importName: 'nspkg.sub.core' }, (_generationService, generated) => generated, vscode_languageserver_1.CancellationToken.None);
167
+ assert_1.default.deepStrictEqual(result.files.map((file) => file.uri.toString()), ['file:///typings/nspkg/sub/core.pyi']);
168
+ });
169
+ test('full generation from a dotted import tracks every module under the root package', async () => {
170
+ const state = (0, testState_1.parseAndGetTestState)(`
171
+ // @filename: sample/__init__.py
172
+ // @library: true
173
+ //// from .sub import answer
174
+ // @filename: sample/root_sibling.py
175
+ // @library: true
176
+ //// root_value = 1
177
+ // @filename: sample/sub/__init__.py
178
+ // @library: true
179
+ //// from .core import answer
180
+ // @filename: sample/sub/core.py
181
+ // @library: true
182
+ //// def answer(): return 42
183
+ // @filename: sample/sub/unimported.py
184
+ // @library: true
185
+ //// sub_value = 2
186
+ `, '/projectRoot').state;
187
+ const service = state.workspace.service;
188
+ const result = await (0, typeStubGeneration_1.runTypeStubGeneration)(() => service.clone('full stub generation test', 'full stub generation test'), { kind: 'full', importName: 'sample.sub' }, (_generationService, generated) => generated, vscode_languageserver_1.CancellationToken.None);
189
+ assert_1.default.deepStrictEqual(result.files.map((file) => file.uri.toString()), [
190
+ 'file:///typings/sample/__init__.pyi',
191
+ 'file:///typings/sample/root_sibling.pyi',
192
+ 'file:///typings/sample/sub/__init__.pyi',
193
+ 'file:///typings/sample/sub/core.pyi',
194
+ 'file:///typings/sample/sub/unimported.pyi',
195
+ ]);
196
+ });
197
+ test('package source collection prefers adjacent stubs over Python sources', () => {
198
+ const state = (0, testState_1.parseAndGetTestState)(`
199
+ // @filename: sample/__init__.py
200
+ // @library: true
201
+ //// # package marker
202
+ // @filename: sample/module.py
203
+ // @library: true
204
+ //// value = 1
205
+ // @filename: sample/module.pyi
206
+ // @library: true
207
+ //// value: str
208
+ `, '/projectRoot').state;
209
+ const service = state.workspace.service;
210
+ const target = (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), 'sample');
211
+ const sourceFileUris = (0, typeStubGeneration_1.collectTypeStubSourceFileUris)(service.fs, target, vscode_languageserver_1.CancellationToken.None);
212
+ assert_1.default.deepStrictEqual(sourceFileUris.map((uri) => target.targetImportPath.getRelativePath(uri)), ['./__init__.py', './module.pyi']);
213
+ });
214
+ test.each([
215
+ {
216
+ title: 'top-level module',
217
+ importName: 'sample',
218
+ targetFile: '/site-packages/sample.py',
219
+ expectedOutput: '/projectRoot/typings/sample-stubs',
220
+ targetIsSingleFile: true,
221
+ },
222
+ {
223
+ title: 'package',
224
+ importName: 'sample',
225
+ targetFile: '/site-packages/sample/__init__.py',
226
+ expectedOutput: '/projectRoot/typings/sample-stubs',
227
+ targetIsSingleFile: false,
228
+ },
229
+ {
230
+ title: 'nested module',
231
+ importName: 'sample.sub.core',
232
+ targetFile: '/site-packages/sample/sub/core.py',
233
+ expectedOutput: '/projectRoot/typings/sample-stubs/sub',
234
+ targetIsSingleFile: false,
235
+ },
236
+ {
237
+ title: 'nested package',
238
+ importName: 'sample.sub',
239
+ targetFile: '/site-packages/sample/sub/__init__.py',
240
+ expectedOutput: '/projectRoot/typings/sample-stubs/sub',
241
+ targetIsSingleFile: false,
242
+ },
243
+ ])('partial planning maps a $title and adds the package marker', ({ importName, targetFile, expectedOutput, targetIsSingleFile }) => {
244
+ const state = (0, testState_1.parseAndGetTestState)('', '/projectRoot').state;
245
+ const stubPath = uri_1.Uri.file('/projectRoot/typings', state.serviceProvider);
246
+ const targetFileUri = uri_1.Uri.file(targetFile, state.serviceProvider);
247
+ const plan = (0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(state.workspace.service.fs, stubPath, {
248
+ importName,
249
+ targetFileUri,
250
+ });
251
+ assert_1.default.strictEqual(plan.source.targetImportPath.toString(), targetFileUri.toString());
252
+ assert_1.default.strictEqual(plan.source.targetIsSingleFile, targetIsSingleFile);
253
+ assert_1.default.strictEqual(plan.source.outputPath.toString(), uri_1.Uri.file(expectedOutput, state.serviceProvider).toString());
254
+ assert_1.default.strictEqual(plan.additionalFiles.length, 1);
255
+ assert_1.default.strictEqual(plan.additionalFiles[0].uri.toString(), uri_1.Uri.file('/projectRoot/typings/sample-stubs/py.typed', state.serviceProvider).toString());
256
+ assert_1.default.strictEqual(plan.additionalFiles[0].contents, 'partial\n');
257
+ assert_1.default.strictEqual(plan.additionalFiles[0].kind, 'partialMarker');
258
+ });
259
+ test.each([
260
+ '',
261
+ '.sample',
262
+ 'sample.',
263
+ '/outside/pkg',
264
+ '../outside',
265
+ 'sample..sub',
266
+ 'sample/sub',
267
+ String.raw `C:\outside\pkg`,
268
+ ])('partial planning rejects invalid import name %s', (importName) => {
269
+ const state = (0, testState_1.parseAndGetTestState)('', '/projectRoot').state;
270
+ const stubPath = uri_1.Uri.file('/projectRoot/typings', state.serviceProvider);
271
+ assert_1.default.throws(() => (0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(state.workspace.service.fs, stubPath, {
272
+ importName,
273
+ targetFileUri: uri_1.Uri.file('/site-packages/sample.py', state.serviceProvider),
274
+ }), /Invalid import name/);
275
+ });
276
+ test.each([
277
+ '',
278
+ '.sample',
279
+ 'sample.',
280
+ '/outside/pkg',
281
+ '../outside',
282
+ 'sample..sub',
283
+ 'sample/sub',
284
+ String.raw `C:\outside\pkg`,
285
+ ])('full target resolution rejects invalid import name %s before import resolution', (importName) => {
286
+ const state = (0, testState_1.parseAndGetTestState)('', '/projectRoot').state;
287
+ const service = state.workspace.service;
288
+ const resolveImport = jest.spyOn(service.getImportResolver(), 'resolveImport');
289
+ assert_1.default.throws(() => (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), importName), /Invalid import name/);
290
+ assert_1.default.strictEqual(resolveImport.mock.calls.length, 0);
291
+ });
292
+ test('partial planning preserves a valid marker and rejects an incompatible marker', () => {
293
+ const state = (0, testState_1.parseAndGetTestState)('', '/projectRoot').state;
294
+ const fs = state.workspace.service.fs;
295
+ const stubPath = uri_1.Uri.file('/projectRoot/typings', state.serviceProvider);
296
+ const markerUri = stubPath.resolvePaths('sample-stubs').pytypedUri;
297
+ const options = {
298
+ importName: 'sample',
299
+ targetFileUri: uri_1.Uri.file('/site-packages/sample.py', state.serviceProvider),
300
+ };
301
+ fs.mkdirSync(markerUri.getDirectory(), { recursive: true });
302
+ fs.writeFileSync(markerUri, 'partial\r\n', 'utf8');
303
+ assert_1.default.deepStrictEqual((0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(fs, stubPath, options).additionalFiles, []);
304
+ fs.writeFileSync(markerUri, 'partial\n', 'utf8');
305
+ assert_1.default.deepStrictEqual((0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(fs, stubPath, options).additionalFiles, []);
306
+ fs.writeFileSync(markerUri, 'full\n', 'utf8');
307
+ assert_1.default.throws(() => (0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(fs, stubPath, options), (error) => error instanceof typeStubGeneration_1.TypeStubGenerationConflictError &&
308
+ error.uri.equals(markerUri) &&
309
+ /is incompatible with partial stub generation/.test(error.message));
310
+ fs.unlinkSync(markerUri);
311
+ fs.mkdirSync(markerUri);
312
+ assert_1.default.throws(() => (0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(fs, stubPath, options), (error) => error instanceof typeStubGeneration_1.TypeStubGenerationConflictError && error.uri.equals(markerUri));
313
+ });
314
+ test('partial generation appends the package marker to generated stubs', () => {
315
+ const state = (0, testState_1.parseAndGetTestState)(`
316
+ // @filename: sample.py
317
+ //// def answer():
318
+ //// return 42
319
+ `, '/projectRoot').state;
320
+ const service = state.workspace.service;
321
+ const stubPath = uri_1.Uri.file('/projectRoot/typings', state.serviceProvider);
322
+ const sourceUri = uri_1.Uri.file('/projectRoot/sample.py', state.serviceProvider);
323
+ const plan = (0, typeStubGeneration_1.createPartialTypeStubGenerationPlan)(service.fs, stubPath, {
324
+ importName: 'sample',
325
+ targetFileUri: sourceUri,
326
+ });
327
+ const result = (0, typeStubGeneration_1.generateTypeStubFiles)(service.backgroundAnalysisProgram.program, plan, vscode_languageserver_1.CancellationToken.None);
328
+ assert_1.default.deepStrictEqual(result.files.map((file) => ({ uri: file.uri.toString(), kind: file.kind })), [
329
+ {
330
+ uri: uri_1.Uri.file('/projectRoot/typings/sample-stubs/__init__.pyi', state.serviceProvider).toString(),
331
+ kind: 'stub',
332
+ },
333
+ {
334
+ uri: uri_1.Uri.file('/projectRoot/typings/sample-stubs/py.typed', state.serviceProvider).toString(),
335
+ kind: 'partialMarker',
336
+ },
337
+ ]);
338
+ assert_1.default.match(result.files[0].contents, /def answer\(\): # -> Literal\[42\]:/);
339
+ assert_1.default.strictEqual(result.files[1].contents, 'partial\n');
340
+ });
341
+ test('package generation stops on cancellation without writing output', () => {
342
+ const code = `
343
+ // @filename: sample/a.py
344
+ //// value = 1
345
+ // @filename: sample/b.py
346
+ //// value = 2
347
+ `;
348
+ const state = (0, testState_1.parseAndGetTestState)(code, '/projectRoot').state;
349
+ const program = state.workspace.service.backgroundAnalysisProgram.program;
350
+ const targetImportPath = uri_1.Uri.file('/projectRoot/sample', state.serviceProvider);
351
+ const outputPath = uri_1.Uri.file('/projectRoot/typings/sample', state.serviceProvider);
352
+ const cancellationSource = new vscode_languageserver_1.CancellationTokenSource();
353
+ const analyzeFile = program.analyzeFile.bind(program);
354
+ jest.spyOn(program, 'analyzeFile').mockImplementation((fileUri, token) => {
355
+ const result = analyzeFile(fileUri, token);
356
+ cancellationSource.cancel();
357
+ return result;
358
+ });
359
+ assert_1.default.throws(() => (0, typeStubGeneration_1.generateTypeStubFiles)(program, {
360
+ source: {
361
+ targetImportPath,
362
+ targetIsSingleFile: false,
363
+ outputPath,
364
+ },
365
+ additionalFiles: [],
366
+ }, cancellationSource.token), (error) => cancellationUtils_1.OperationCanceledException.is(error));
367
+ assert_1.default.strictEqual(state.workspace.service.fs.existsSync(outputPath), false);
368
+ });
369
+ test('package generation rejects an empty target instead of returning false success', () => {
370
+ const state = (0, testState_1.parseAndGetTestState)(`
371
+ // @filename: sample.py
372
+ //// value = 1
373
+ `, '/projectRoot').state;
374
+ const outputPath = uri_1.Uri.file('/projectRoot/typings/empty', state.serviceProvider);
375
+ assert_1.default.throws(() => (0, typeStubGeneration_1.generateTypeStubFiles)(state.workspace.service.backgroundAnalysisProgram.program, {
376
+ source: {
377
+ targetImportPath: uri_1.Uri.file('/projectRoot/empty', state.serviceProvider),
378
+ targetIsSingleFile: false,
379
+ outputPath,
380
+ },
381
+ additionalFiles: [],
382
+ }, vscode_languageserver_1.CancellationToken.None), /No source files found/);
383
+ assert_1.default.strictEqual(state.workspace.service.fs.existsSync(outputPath), false);
384
+ });
385
+ test.each(['nspkg', 'nspkg.sub'])('target resolution rejects namespace-only target %s instead of widening source collection', (importName) => {
386
+ const state = (0, testState_1.parseAndGetTestState)(`
387
+ // @filename: nspkg/sub/a.py
388
+ // @library: true
389
+ //// value_a = 1
390
+ // @filename: nspkg/sub/b.py
391
+ // @library: true
392
+ //// value_b = 2
393
+ `, '/projectRoot').state;
394
+ const service = state.workspace.service;
395
+ assert_1.default.throws(() => (0, typeStubGeneration_1.resolveTypeStubTarget)(service.getImportResolver(), service.getConfigOptions(), importName), /namespace package/);
396
+ });
397
+ //# sourceMappingURL=typeStubGeneration.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"typeStubGeneration.test.js","sourceRoot":"","sources":["../../../../../src/tests/typeStubGeneration.test.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;AAEH,oDAA4B;AAC5B,iEAAmF;AAEnF,uEAOwC;AACxC,mEAAyE;AACzE,2CAAwC;AACxC,6DAAqE;AAErE,IAAI,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;IACrF,MAAM,IAAI,GAAG;;;;CAIhB,CAAC;IACE,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,IAAI,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC;IAC/D,MAAM,SAAS,GAAG,SAAG,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAC5E,MAAM,UAAU,GAAG,SAAG,CAAC,IAAI,CAAC,6BAA6B,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAClF,MAAM,OAAO,GAAG,SAAG,CAAC,IAAI,CAAC,0CAA0C,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAE5F,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,yBAAyB,CAAC,qBAAqB,CACxF;QACI,MAAM,EAAE;YACJ,gBAAgB,EAAE,SAAS;YAC3B,kBAAkB,EAAE,IAAI;YACxB,UAAU;SACb;QACD,eAAe,EAAE,EAAE;KACtB,EACD,yCAAiB,CAAC,IAAI,CACzB,CAAC;IAEF,gBAAM,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;IAC1E,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC3C,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IACvE,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACjD,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,qCAAqC,CAAC,CAAC;AAClF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,IAAI,CAAC;IACN;QACI,KAAK,EAAE,oBAAoB;QAC3B,QAAQ,EAAE,WAAW;QACrB,oBAAoB,EAAE,eAAe;QACrC,kBAAkB,EAAE,IAAI;KAC3B;IACD;QACI,KAAK,EAAE,SAAS;QAChB,QAAQ,EAAE,oBAAoB;QAC9B,oBAAoB,EAAE,sBAAsB;QAC5C,kBAAkB,EAAE,KAAK;KAC5B;CACJ,CAAC,CACE,iEAAiE,EACjE,CAAC,EAAE,QAAQ,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,EAAE,EAAE;IACvD,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;gBACI,QAAQ;;;CAGvB,EACW,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,MAAM,MAAM,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,QAAQ,CAAC,CAAC;IAExG,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC;IACtE,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;IAClE,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACpD,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC3D,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC;AAChE,CAAC,CACJ,CAAC;AAEF,IAAI,CAAC,yDAAyD,EAAE,GAAG,EAAE;IACjE,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;CAMP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,MAAM,MAAM,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,QAAQ,CAAC,CAAC;IACxG,MAAM,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,QAAS,CAAC;IAE/D,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7E,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;IACtG,gBAAM,CAAC,cAAc,CACjB,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAC1B,OAAO,CAAC,gBAAgB,EAAE,CAAC,WAAW,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAC5E,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,GAAG,EAAE;IACtE,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;;;;;;;;CAcP,CACI,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IACxC,MAAM,aAAa,GAAG,SAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAEzE,MAAM,aAAa,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,QAAQ,CAAC,CAAC;IAC/G,MAAM,YAAY,GAAG,IAAA,0CAAqB,EACtC,OAAO,CAAC,iBAAiB,EAAE,EAC3B,OAAO,CAAC,gBAAgB,EAAE,EAC1B,QAAQ,EACR,aAAa,CAChB,CAAC;IAEF,gBAAM,CAAC,WAAW,CAAC,aAAa,CAAC,gBAAgB,CAAC,QAAQ,EAAE,EAAE,2BAA2B,CAAC,CAAC;IAC3F,gBAAM,CAAC,WAAW,CAAC,YAAY,CAAC,gBAAgB,CAAC,QAAQ,EAAE,EAAE,6BAA6B,CAAC,CAAC;IAC5F,gBAAM,CAAC,eAAe,CAClB,YAAY,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EACxD,CAAC,yCAAyC,CAAC,CAC9C,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2EAA2E,EAAE,GAAG,EAAE;IACnF,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;;;;CAUP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,MAAM,MAAM,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,iBAAiB,CAAC,CAAC;IAEjH,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;IACrD,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC,CAAC;IACxE,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACpD,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC;IACvE,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAC5D,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;AAC3G,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kFAAkF,EAAE,GAAG,EAAE;IAC1F,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;;;;;;;CAaP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,MAAM,MAAM,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,eAAe,CAAC,CAAC;IAC/G,MAAM,cAAc,GAAG,IAAA,kDAA6B,EAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,yCAAiB,CAAC,IAAI,CAAC,CAAC;IAEjG,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,6BAA6B,CAAC,CAAC,CAAC;IAC/E,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,uBAAuB,CAAC,CAAC,CAAC;IACnE,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;IACrD,gBAAM,CAAC,eAAe,CAClB,cAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,EACzE,CAAC,eAAe,EAAE,SAAS,CAAC,CAC/B,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;IACvG,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;;;;CAUP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,MAAM,MAAM,GAAG,MAAM,IAAA,0CAAqB,EACtC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,uCAAuC,CAAC,EACrG,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,EAC9C,CAAC,kBAAkB,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAC5C,yCAAiB,CAAC,IAAI,CACzB,CAAC;IAEF,gBAAM,CAAC,eAAe,CAClB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAC/C,CAAC,oCAAoC,CAAC,CACzC,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;IAC/F,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;;;;;;;;;;CAgBP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,MAAM,MAAM,GAAG,MAAM,IAAA,0CAAqB,EACtC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,2BAA2B,CAAC,EAC7E,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,EAC1C,CAAC,kBAAkB,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAC5C,yCAAiB,CAAC,IAAI,CACzB,CAAC;IAEF,gBAAM,CAAC,eAAe,CAClB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAC/C;QACI,qCAAqC;QACrC,yCAAyC;QACzC,yCAAyC;QACzC,qCAAqC;QACrC,2CAA2C;KAC9C,CACJ,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sEAAsE,EAAE,GAAG,EAAE;IAC9E,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;;;;CAUP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IACxC,MAAM,MAAM,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,QAAQ,CAAC,CAAC;IAExG,MAAM,cAAc,GAAG,IAAA,kDAA6B,EAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,yCAAiB,CAAC,IAAI,CAAC,CAAC;IAEjG,gBAAM,CAAC,eAAe,CAClB,cAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,EACzE,CAAC,eAAe,EAAE,cAAc,CAAC,CACpC,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,IAAI,CAAC;IACN;QACI,KAAK,EAAE,kBAAkB;QACzB,UAAU,EAAE,QAAQ;QACpB,UAAU,EAAE,0BAA0B;QACtC,cAAc,EAAE,mCAAmC;QACnD,kBAAkB,EAAE,IAAI;KAC3B;IACD;QACI,KAAK,EAAE,SAAS;QAChB,UAAU,EAAE,QAAQ;QACpB,UAAU,EAAE,mCAAmC;QAC/C,cAAc,EAAE,mCAAmC;QACnD,kBAAkB,EAAE,KAAK;KAC5B;IACD;QACI,KAAK,EAAE,eAAe;QACtB,UAAU,EAAE,iBAAiB;QAC7B,UAAU,EAAE,mCAAmC;QAC/C,cAAc,EAAE,uCAAuC;QACvD,kBAAkB,EAAE,KAAK;KAC5B;IACD;QACI,KAAK,EAAE,gBAAgB;QACvB,UAAU,EAAE,YAAY;QACxB,UAAU,EAAE,uCAAuC;QACnD,cAAc,EAAE,uCAAuC;QACvD,kBAAkB,EAAE,KAAK;KAC5B;CACJ,CAAC,CACE,4DAA4D,EAC5D,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,EAAE;IAC/D,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,EAAE,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC;IAC7D,MAAM,QAAQ,GAAG,SAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IACzE,MAAM,aAAa,GAAG,SAAG,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAElE,MAAM,IAAI,GAAG,IAAA,wDAAmC,EAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,EAAE;QACnF,UAAU;QACV,aAAa;KAChB,CAAC,CAAC;IAEH,gBAAM,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,EAAE,aAAa,CAAC,QAAQ,EAAE,CAAC,CAAC;IACtF,gBAAM,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;IACvE,gBAAM,CAAC,WAAW,CACd,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,EACjC,SAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAC7D,CAAC;IACF,gBAAM,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACnD,gBAAM,CAAC,WAAW,CACd,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,EACtC,SAAG,CAAC,IAAI,CAAC,4CAA4C,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAC3F,CAAC;IACF,gBAAM,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAClE,gBAAM,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;AACtE,CAAC,CACJ,CAAC;AAEF,IAAI,CAAC,IAAI,CAAC;IACN,EAAE;IACF,SAAS;IACT,SAAS;IACT,cAAc;IACd,YAAY;IACZ,aAAa;IACb,YAAY;IACZ,MAAM,CAAC,GAAG,CAAA,gBAAgB;CAC7B,CAAC,CAAC,iDAAiD,EAAE,CAAC,UAAU,EAAE,EAAE;IACjE,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,EAAE,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC;IAC7D,MAAM,QAAQ,GAAG,SAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAEzE,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CACD,IAAA,wDAAmC,EAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,EAAE;QACtE,UAAU;QACV,aAAa,EAAE,SAAG,CAAC,IAAI,CAAC,0BAA0B,EAAE,KAAK,CAAC,eAAe,CAAC;KAC7E,CAAC,EACN,qBAAqB,CACxB,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,IAAI,CAAC;IACN,EAAE;IACF,SAAS;IACT,SAAS;IACT,cAAc;IACd,YAAY;IACZ,aAAa;IACb,YAAY;IACZ,MAAM,CAAC,GAAG,CAAA,gBAAgB;CAC7B,CAAC,CAAC,gFAAgF,EAAE,CAAC,UAAU,EAAE,EAAE;IAChG,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,EAAE,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC;IAC7D,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IACxC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,eAAe,CAAC,CAAC;IAE/E,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CAAC,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,UAAU,CAAC,EAChG,qBAAqB,CACxB,CAAC;IACF,gBAAM,CAAC,WAAW,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8EAA8E,EAAE,GAAG,EAAE;IACtF,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,EAAE,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC;IAC7D,MAAM,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,SAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,UAAU,CAAC;IACnE,MAAM,OAAO,GAAG;QACZ,UAAU,EAAE,QAAQ;QACpB,aAAa,EAAE,SAAG,CAAC,IAAI,CAAC,0BAA0B,EAAE,KAAK,CAAC,eAAe,CAAC;KAC7E,CAAC;IACF,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;IAEnD,gBAAM,CAAC,eAAe,CAAC,IAAA,wDAAmC,EAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IAEvG,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;IAEjD,gBAAM,CAAC,eAAe,CAAC,IAAA,wDAAmC,EAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IAEvG,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC9C,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CAAC,IAAA,wDAAmC,EAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,EAChE,CAAC,KAAK,EAAE,EAAE,CACN,KAAK,YAAY,oDAA+B;QAChD,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC;QAC3B,8CAA8C,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CACzE,CAAC;IAEF,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IACzB,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACxB,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CAAC,IAAA,wDAAmC,EAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,EAChE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,YAAY,oDAA+B,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAC7F,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kEAAkE,EAAE,GAAG,EAAE;IAC1E,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;CAIP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IACxC,MAAM,QAAQ,GAAG,SAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,SAAG,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAC5E,MAAM,IAAI,GAAG,IAAA,wDAAmC,EAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,EAAE;QACnE,UAAU,EAAE,QAAQ;QACpB,aAAa,EAAE,SAAS;KAC3B,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,IAAA,0CAAqB,EAAC,OAAO,CAAC,yBAAyB,CAAC,OAAO,EAAE,IAAI,EAAE,yCAAiB,CAAC,IAAI,CAAC,CAAC;IAE9G,gBAAM,CAAC,eAAe,CAClB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,EAC3E;QACI;YACI,GAAG,EAAE,SAAG,CAAC,IAAI,CAAC,gDAAgD,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;YACjG,IAAI,EAAE,MAAM;SACf;QACD;YACI,GAAG,EAAE,SAAG,CAAC,IAAI,CAAC,4CAA4C,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;YAC7F,IAAI,EAAE,eAAe;SACxB;KACJ,CACJ,CAAC;IACF,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,qCAAqC,CAAC,CAAC;IAC9E,gBAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;AAC9D,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iEAAiE,EAAE,GAAG,EAAE;IACzE,MAAM,IAAI,GAAG;;;;;CAKhB,CAAC;IACE,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,IAAI,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC;IAC/D,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,yBAAyB,CAAC,OAAO,CAAC;IAC1E,MAAM,gBAAgB,GAAG,SAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAChF,MAAM,UAAU,GAAG,SAAG,CAAC,IAAI,CAAC,6BAA6B,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAClF,MAAM,kBAAkB,GAAG,IAAI,+CAAuB,EAAE,CAAC;IACzD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtD,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC,kBAAkB,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QACrE,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC3C,kBAAkB,CAAC,MAAM,EAAE,CAAC;QAC5B,OAAO,MAAM,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CACD,IAAA,0CAAqB,EACjB,OAAO,EACP;QACI,MAAM,EAAE;YACJ,gBAAgB;YAChB,kBAAkB,EAAE,KAAK;YACzB,UAAU;SACb;QACD,eAAe,EAAE,EAAE;KACtB,EACD,kBAAkB,CAAC,KAAK,CAC3B,EACL,CAAC,KAAc,EAAE,EAAE,CAAC,8CAA0B,CAAC,EAAE,CAAC,KAAK,CAAC,CAC3D,CAAC;IACF,gBAAM,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;AACjF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+EAA+E,EAAE,GAAG,EAAE;IACvF,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;CAGP,EACO,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,UAAU,GAAG,SAAG,CAAC,IAAI,CAAC,4BAA4B,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IAEjF,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CACD,IAAA,0CAAqB,EACjB,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,yBAAyB,CAAC,OAAO,EACzD;QACI,MAAM,EAAE;YACJ,gBAAgB,EAAE,SAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE,KAAK,CAAC,eAAe,CAAC;YACvE,kBAAkB,EAAE,KAAK;YACzB,UAAU;SACb;QACD,eAAe,EAAE,EAAE;KACtB,EACD,yCAAiB,CAAC,IAAI,CACzB,EACL,uBAAuB,CAC1B,CAAC;IACF,gBAAM,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;AACjF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,CAC7B,0FAA0F,EAC1F,CAAC,UAAU,EAAE,EAAE;IACX,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAC9B;;;;;;;CAOX,EACW,cAAc,CACjB,CAAC,KAAK,CAAC;IACR,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC;IAExC,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CAAC,IAAA,0CAAqB,EAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,UAAU,CAAC,EAChG,mBAAmB,CACtB,CAAC;AACN,CAAC,CACJ,CAAC"}
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,136 @@
1
+ "use strict";
2
+ /*
3
+ * typeStubOutput.test.ts
4
+ * Copyright (c) Microsoft Corporation.
5
+ * Licensed under the MIT license.
6
+ *
7
+ * Tests for generated type stub output adapters.
8
+ */
9
+ var __importDefault = (this && this.__importDefault) || function (mod) {
10
+ return (mod && mod.__esModule) ? mod : { "default": mod };
11
+ };
12
+ Object.defineProperty(exports, "__esModule", { value: true });
13
+ const assert_1 = __importDefault(require("assert"));
14
+ const vscode_languageserver_types_1 = require("vscode-languageserver-types");
15
+ const typeStubOutput_1 = require("../analyzer/typeStubOutput");
16
+ const uri_1 = require("../common/uri/uri");
17
+ const filesystem_1 = require("./harness/vfs/filesystem");
18
+ test('writes generated type stub files and replaces existing contents', () => {
19
+ const fs = new filesystem_1.TestFileSystem(/* ignoreCase */ false, { cwd: '/' });
20
+ const existingUri = uri_1.Uri.file('/typings/sample-stubs/py.typed', fs);
21
+ const stubUri = uri_1.Uri.file('/typings/sample-stubs/sample/core.pyi', fs);
22
+ fs.mkdirSync(existingUri.getDirectory(), { recursive: true });
23
+ fs.writeFileSync(existingUri, 'old marker', 'utf8');
24
+ const files = [
25
+ { uri: stubUri, contents: 'def answer() -> int: ...\n', kind: 'stub' },
26
+ { uri: existingUri, contents: 'partial\n', kind: 'partialMarker' },
27
+ ];
28
+ (0, typeStubOutput_1.writeGeneratedTypeStubFiles)(fs, files);
29
+ assert_1.default.strictEqual(fs.readFileSync(stubUri, 'utf8'), 'def answer() -> int: ...\n');
30
+ assert_1.default.strictEqual(fs.readFileSync(existingUri, 'utf8'), 'partial\n');
31
+ });
32
+ test('converts generated type stub files to ordered create and replacement edits', () => {
33
+ const fs = new filesystem_1.TestFileSystem(/* ignoreCase */ false, { cwd: '/' });
34
+ const existingUri = uri_1.Uri.file('/typings/sample-stubs/py.typed', fs);
35
+ const stubUri = uri_1.Uri.file('/typings/sample-stubs/sample/core.pyi', fs);
36
+ fs.mkdirSync(existingUri.getDirectory(), { recursive: true });
37
+ fs.writeFileSync(existingUri, 'old marker\ncontent\n', 'utf8');
38
+ const files = [
39
+ { uri: stubUri, contents: 'def answer() -> int: ...\n', kind: 'stub' },
40
+ { uri: existingUri, contents: 'partial\n', kind: 'partialMarker' },
41
+ ];
42
+ const edit = (0, typeStubOutput_1.convertGeneratedTypeStubFilesToWorkspaceEdit)(fs, files);
43
+ assert_1.default.ok(edit.documentChanges);
44
+ assert_1.default.strictEqual(edit.documentChanges.length, 3);
45
+ assert_1.default.ok(vscode_languageserver_types_1.CreateFile.is(edit.documentChanges[0]));
46
+ assert_1.default.deepStrictEqual(edit.documentChanges[0], {
47
+ kind: 'create',
48
+ uri: stubUri.toString(),
49
+ options: { overwrite: false },
50
+ });
51
+ assert_1.default.ok(vscode_languageserver_types_1.TextDocumentEdit.is(edit.documentChanges[1]));
52
+ assert_1.default.deepStrictEqual(edit.documentChanges[1], {
53
+ textDocument: { uri: stubUri.toString(), version: null },
54
+ edits: [
55
+ {
56
+ range: {
57
+ start: { line: 0, character: 0 },
58
+ end: { line: 0, character: 0 },
59
+ },
60
+ newText: 'def answer() -> int: ...\n',
61
+ },
62
+ ],
63
+ });
64
+ assert_1.default.ok(vscode_languageserver_types_1.TextDocumentEdit.is(edit.documentChanges[2]));
65
+ assert_1.default.deepStrictEqual(edit.documentChanges[2], {
66
+ textDocument: { uri: existingUri.toString(), version: null },
67
+ edits: [
68
+ {
69
+ range: {
70
+ start: { line: 0, character: 0 },
71
+ end: { line: 2, character: 0 },
72
+ },
73
+ newText: 'partial\n',
74
+ },
75
+ ],
76
+ });
77
+ assert_1.default.strictEqual(fs.existsSync(stubUri), false);
78
+ assert_1.default.strictEqual(fs.readFileSync(existingUri, 'utf8'), 'old marker\ncontent\n');
79
+ });
80
+ test('direct output surfaces a later write failure after preserving completed writes', () => {
81
+ const fs = new filesystem_1.TestFileSystem(/* ignoreCase */ false, { cwd: '/' });
82
+ const firstUri = uri_1.Uri.file('/typings/pkg/a.pyi', fs);
83
+ const secondUri = uri_1.Uri.file('/typings/pkg/b.pyi', fs);
84
+ const writeFileSync = fs.writeFileSync.bind(fs);
85
+ jest.spyOn(fs, 'writeFileSync').mockImplementation((uri, data, encoding) => {
86
+ if (uri.equals(secondUri)) {
87
+ throw new Error('second write failed');
88
+ }
89
+ writeFileSync(uri, data, encoding);
90
+ });
91
+ assert_1.default.throws(() => (0, typeStubOutput_1.writeGeneratedTypeStubFiles)(fs, [
92
+ { uri: firstUri, contents: 'value: int\n', kind: 'stub' },
93
+ { uri: secondUri, contents: 'name: str\n', kind: 'stub' },
94
+ ]), /second write failed/);
95
+ assert_1.default.strictEqual(fs.readFileSync(firstUri, 'utf8'), 'value: int\n');
96
+ assert_1.default.strictEqual(fs.existsSync(secondUri), false);
97
+ });
98
+ test('uses the client version when replacing an open generated stub', () => {
99
+ const fs = new filesystem_1.TestFileSystem(/* ignoreCase */ false, { cwd: '/' });
100
+ const stubUri = uri_1.Uri.file('/typings/sample.pyi', fs);
101
+ const existingContents = 'value: int\n';
102
+ fs.mkdirSync(stubUri.getDirectory(), { recursive: true });
103
+ fs.writeFileSync(stubUri, existingContents, 'utf8');
104
+ const edit = (0, typeStubOutput_1.convertGeneratedTypeStubFilesToWorkspaceEdit)(fs, [{ uri: stubUri, contents: 'value: str\n', kind: 'stub' }], (uri) => (uri.equals(stubUri) ? { version: 7, contents: existingContents } : undefined));
105
+ assert_1.default.deepStrictEqual(edit.documentChanges, [
106
+ {
107
+ textDocument: { uri: stubUri.toString(), version: 7 },
108
+ edits: [
109
+ {
110
+ range: {
111
+ start: { line: 0, character: 0 },
112
+ end: { line: 1, character: 0 },
113
+ },
114
+ newText: 'value: str\n',
115
+ },
116
+ ],
117
+ },
118
+ ]);
119
+ });
120
+ test('refuses to replace an open generated stub with unsaved changes', () => {
121
+ const fs = new filesystem_1.TestFileSystem(/* ignoreCase */ false, { cwd: '/' });
122
+ const stubUri = uri_1.Uri.file('/typings/sample.pyi', fs);
123
+ fs.mkdirSync(stubUri.getDirectory(), { recursive: true });
124
+ fs.writeFileSync(stubUri, 'value: int\n', 'utf8');
125
+ assert_1.default.throws(() => (0, typeStubOutput_1.convertGeneratedTypeStubFilesToWorkspaceEdit)(fs, [{ uri: stubUri, contents: 'value: str\n', kind: 'stub' }], (uri) => (uri.equals(stubUri) ? { version: 8, contents: 'value: float\n' } : undefined)), {
126
+ message: `Cannot update generated type stub '${stubUri.toUserVisibleString()}' because it has unsaved changes`,
127
+ });
128
+ });
129
+ test('refuses to create a generated stub over an open unsaved document', () => {
130
+ const fs = new filesystem_1.TestFileSystem(/* ignoreCase */ false, { cwd: '/' });
131
+ const stubUri = uri_1.Uri.file('/typings/sample.pyi', fs);
132
+ assert_1.default.throws(() => (0, typeStubOutput_1.convertGeneratedTypeStubFilesToWorkspaceEdit)(fs, [{ uri: stubUri, contents: 'value: str\n', kind: 'stub' }], (uri) => (uri.equals(stubUri) ? { version: 1, contents: 'value: float\n' } : undefined)), {
133
+ message: `Cannot update generated type stub '${stubUri.toUserVisibleString()}' because it has unsaved changes`,
134
+ });
135
+ });
136
+ //# sourceMappingURL=typeStubOutput.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"typeStubOutput.test.js","sourceRoot":"","sources":["../../../../../src/tests/typeStubOutput.test.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;AAEH,oDAA4B;AAC5B,6EAA2E;AAG3E,+DAAuH;AACvH,2CAAwC;AACxC,yDAA0D;AAE1D,IAAI,CAAC,iEAAiE,EAAE,GAAG,EAAE;IACzE,MAAM,EAAE,GAAG,IAAI,2BAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,MAAM,WAAW,GAAG,SAAG,CAAC,IAAI,CAAC,gCAAgC,EAAE,EAAE,CAAC,CAAC;IACnE,MAAM,OAAO,GAAG,SAAG,CAAC,IAAI,CAAC,uCAAuC,EAAE,EAAE,CAAC,CAAC;IACtE,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9D,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IACpD,MAAM,KAAK,GAA4B;QACnC,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,4BAA4B,EAAE,IAAI,EAAE,MAAM,EAAE;QACtE,EAAE,GAAG,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE;KACrE,CAAC;IAEF,IAAA,4CAA2B,EAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAEvC,gBAAM,CAAC,WAAW,CAAC,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,4BAA4B,CAAC,CAAC;IACnF,gBAAM,CAAC,WAAW,CAAC,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,WAAW,CAAC,CAAC;AAC1E,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,4EAA4E,EAAE,GAAG,EAAE;IACpF,MAAM,EAAE,GAAG,IAAI,2BAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,MAAM,WAAW,GAAG,SAAG,CAAC,IAAI,CAAC,gCAAgC,EAAE,EAAE,CAAC,CAAC;IACnE,MAAM,OAAO,GAAG,SAAG,CAAC,IAAI,CAAC,uCAAuC,EAAE,EAAE,CAAC,CAAC;IACtE,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9D,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,uBAAuB,EAAE,MAAM,CAAC,CAAC;IAC/D,MAAM,KAAK,GAA4B;QACnC,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,4BAA4B,EAAE,IAAI,EAAE,MAAM,EAAE;QACtE,EAAE,GAAG,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE;KACrE,CAAC;IAEF,MAAM,IAAI,GAAG,IAAA,6DAA4C,EAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAErE,gBAAM,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChC,gBAAM,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACnD,gBAAM,CAAC,EAAE,CAAC,wCAAU,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClD,gBAAM,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;QAC5C,IAAI,EAAE,QAAQ;QACd,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE;QACvB,OAAO,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE;KAChC,CAAC,CAAC;IACH,gBAAM,CAAC,EAAE,CAAC,8CAAgB,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,gBAAM,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;QAC5C,YAAY,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;QACxD,KAAK,EAAE;YACH;gBACI,KAAK,EAAE;oBACH,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;oBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;iBACjC;gBACD,OAAO,EAAE,4BAA4B;aACxC;SACJ;KACJ,CAAC,CAAC;IACH,gBAAM,CAAC,EAAE,CAAC,8CAAgB,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,gBAAM,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;QAC5C,YAAY,EAAE,EAAE,GAAG,EAAE,WAAW,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;QAC5D,KAAK,EAAE;YACH;gBACI,KAAK,EAAE;oBACH,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;oBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;iBACjC;gBACD,OAAO,EAAE,WAAW;aACvB;SACJ;KACJ,CAAC,CAAC;IACH,gBAAM,CAAC,WAAW,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;IAClD,gBAAM,CAAC,WAAW,CAAC,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC,CAAC;AACtF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gFAAgF,EAAE,GAAG,EAAE;IACxF,MAAM,EAAE,GAAG,IAAI,2BAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,MAAM,QAAQ,GAAG,SAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,SAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;IACrD,MAAM,aAAa,GAAG,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChD,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;QACvE,IAAI,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC3C,CAAC;QACD,aAAa,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CACD,IAAA,4CAA2B,EAAC,EAAE,EAAE;QAC5B,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE;QACzD,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE;KAC5D,CAAC,EACN,qBAAqB,CACxB,CAAC;IACF,gBAAM,CAAC,WAAW,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,cAAc,CAAC,CAAC;IACtE,gBAAM,CAAC,WAAW,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;AACxD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACvE,MAAM,EAAE,GAAG,IAAI,2BAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,SAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,gBAAgB,GAAG,cAAc,CAAC;IACxC,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;IAEpD,MAAM,IAAI,GAAG,IAAA,6DAA4C,EACrD,EAAE,EACF,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAC1D,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAC1F,CAAC;IAEF,gBAAM,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,EAAE;QACzC;YACI,YAAY,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE;YACrD,KAAK,EAAE;gBACH;oBACI,KAAK,EAAE;wBACH,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;wBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;qBACjC;oBACD,OAAO,EAAE,cAAc;iBAC1B;aACJ;SACJ;KACJ,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gEAAgE,EAAE,GAAG,EAAE;IACxE,MAAM,EAAE,GAAG,IAAI,2BAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,SAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;IACpD,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC;IAElD,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CACD,IAAA,6DAA4C,EACxC,EAAE,EACF,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAC1D,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAC1F,EACL;QACI,OAAO,EAAE,sCAAsC,OAAO,CAAC,mBAAmB,EAAE,kCAAkC;KACjH,CACJ,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kEAAkE,EAAE,GAAG,EAAE;IAC1E,MAAM,EAAE,GAAG,IAAI,2BAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,SAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;IAEpD,gBAAM,CAAC,MAAM,CACT,GAAG,EAAE,CACD,IAAA,6DAA4C,EACxC,EAAE,EACF,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAC1D,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAC1F,EACL;QACI,OAAO,EAAE,sCAAsC,OAAO,CAAC,mBAAmB,EAAE,kCAAkC;KACjH,CACJ,CAAC;AACN,CAAC,CAAC,CAAC"}
@@ -0,0 +1 @@
1
+ export {};