Skip to content

Commit 5224201

Browse files
fix: Make tests more robust & fix path bug
1 parent a29134b commit 5224201

File tree

10 files changed

+538
-55
lines changed

10 files changed

+538
-55
lines changed

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,9 @@ yarn-debug.log*
66
yarn-error.log*
77
lerna-debug.log*
88

9+
# Editor files
10+
.idea/
11+
912
# Diagnostic reports (https://nodejs.org/api/report.html)
1013
report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json
1114

packages/pyright-internal/src/analyzer/program.ts

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,15 @@ export class Program {
331331

332332
addTrackedFile(filePath: string, isThirdPartyImport = false, isInPyTypedPackage = false): SourceFile {
333333
let sourceFileInfo = this.getSourceFileInfo(filePath);
334-
const importName = this._getImportNameForFile(filePath);
334+
let importName = this._getImportNameForFile(filePath);
335+
// HACK(scip-python): When adding tracked files for imports, we end up passing
336+
// normalized paths as the argument. However, _getImportNameForFile seemingly
337+
// needs a non-normalized path, which cannot be recovered directly from a
338+
// normalized path. However, in practice, the non-normalized path seems to
339+
// be stored on the sourceFileInfo, so attempt to use that instead.
340+
if (importName === '' && sourceFileInfo) {
341+
importName = this._getImportNameForFile(sourceFileInfo.sourceFile.getFilePath());
342+
}
335343

336344
if (sourceFileInfo) {
337345
// The module name may have changed based on updates to the

packages/pyright-scip/CONTRIBUTING.md

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,32 @@ node ./index.js <other args>
7373
npm run check-snapshots
7474
```
7575

76+
#### Filter specific snapshot tests
77+
78+
Use the `--filter-tests` flag to run only specific snapshot tests:
79+
```bash
80+
# Using npm scripts (note the -- to pass arguments)
81+
npm run check-snapshots -- --filter-tests test1,test2,test3
82+
```
83+
84+
Available snapshot tests can be found in `snapshots/input/`.
85+
7686
Using a different Python version other than the one specified
7787
in `.tool-versions` may also lead to errors.
7888

89+
## Making changes to Pyright internals
90+
91+
When modifying code in the `pyright-internal` package:
92+
93+
1. Keep changes minimal: Every change introduces a risk of
94+
merge conflicts. Adding doc comments is fine, but avoid
95+
changing functionality if possible. Instead of changing
96+
access modifiers, prefer copying small functions into
97+
scip-pyright logic.
98+
2. Use a `NOTE(scip-python):` prefix when adding comments to
99+
make it clearer which comments were added by upstream
100+
maintainers vs us.
101+
79102
## Publishing releases
80103

81104
1. Change the version in `packages/pyright-scip/package.json`
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
# < definition scip-python python snapshot-util 0.1 `src.long_importer`/__init__:
22

33
import foo.bar.baz.mod
4-
# ^^^^^^^^^^^^^^^ reference snapshot-util 0.1 `foo.bar.baz.mod`/__init__:
4+
# ^^^^^^^^^^^^^^^ reference local 0
55

66
print(foo.bar.baz.mod.SuchNestedMuchWow)
77
#^^^^ reference python-stdlib 3.11 builtins/print().
8-
# ^^^^^^^^^^^^^^^ reference snapshot-util 0.1 `foo.bar.baz.mod`/__init__:
8+
# ^^^ reference local 0
99
# ^^^^^^^^^^^^^^^^^ reference snapshot-util 0.1 `src.foo.bar.baz.mod`/SuchNestedMuchWow#
1010

packages/pyright-scip/src/indexer.ts

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -123,9 +123,9 @@ export class Indexer {
123123
this.importResolver = new ImportResolver(fs, this.pyrightConfig, host);
124124

125125
this.program = new Program(this.importResolver, this.pyrightConfig);
126-
// Normalize paths to ensure consistency with other code paths.
127-
const normalizedProjectFiles = [...this.projectFiles].map((path: string) => normalizePathCase(fs, path));
128-
this.program.setTrackedFiles(normalizedProjectFiles);
126+
// setTrackedFiles internally handles path normalization, so we don't normalize
127+
// paths here.
128+
this.program.setTrackedFiles([...this.projectFiles]);
129129

130130
if (scipConfig.projectNamespace) {
131131
setProjectNamespace(scipConfig.projectName, this.scipConfig.projectNamespace!);
@@ -194,7 +194,9 @@ export class Indexer {
194194
let projectSourceFiles: SourceFile[] = [];
195195
withStatus('Index workspace and track project files', () => {
196196
this.program.indexWorkspace((filepath: string) => {
197-
// Filter out filepaths not part of this project
197+
// Do not index files outside the project because SCIP doesn't support it.
198+
//
199+
// Both filepath and this.scipConfig.projectRoot are NOT normalized.
198200
if (filepath.indexOf(this.scipConfig.projectRoot) != 0) {
199201
return;
200202
}
@@ -204,6 +206,7 @@ export class Indexer {
204206

205207
let requestsImport = sourceFile.getImports();
206208
requestsImport.forEach((entry) =>
209+
// entry.resolvedPaths are all normalized.
207210
entry.resolvedPaths.forEach((value) => {
208211
this.program.addTrackedFile(value, true, false);
209212
})

packages/pyright-scip/src/lib.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -310,10 +310,10 @@ export function writeSnapshot(outputPath: string, obtained: string): void {
310310
fs.writeFileSync(outputPath, obtained, { flag: 'w' });
311311
}
312312

313-
export function diffSnapshot(outputPath: string, obtained: string): void {
313+
export function diffSnapshot(outputPath: string, obtained: string): 'equal' | 'different' {
314314
let existing = fs.readFileSync(outputPath, { encoding: 'utf8' });
315315
if (obtained === existing) {
316-
return;
316+
return 'equal';
317317
}
318318

319319
console.error(
@@ -326,7 +326,7 @@ export function diffSnapshot(outputPath: string, obtained: string): void {
326326
'(what the current code produces). Run the command "npm run update-snapshots" to accept the new behavior.'
327327
)
328328
);
329-
exit(1);
329+
return 'different';
330330
}
331331

332332
function occurrencesByLine(a: scip.Occurrence, b: scip.Occurrence): number {

packages/pyright-scip/src/main-impl.ts

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,9 @@ import { IndexOptions, SnapshotOptions, mainCommand } from './MainCommand';
99
import { sendStatus, setQuiet, setShowProgressRateLimit } from './status';
1010
import { Indexer } from './indexer';
1111
import { exit } from 'process';
12+
import { TestFailure, TestError, ValidationResults } from './test-runner';
1213

13-
function indexAction(options: IndexOptions): void {
14+
export function indexAction(options: IndexOptions): void {
1415
setQuiet(options.quiet);
1516
if (options.showProgressRateLimit !== undefined) {
1617
setShowProgressRateLimit(options.showProgressRateLimit);
@@ -91,6 +92,8 @@ function snapshotAction(snapshotRoot: string, options: SnapshotOptions): void {
9192

9293
const scipIndexPath = path.join(projectRoot, options.output);
9394
const scipIndex = scip.Index.deserializeBinary(fs.readFileSync(scipIndexPath));
95+
96+
let hasDiff = false;
9497
for (const doc of scipIndex.documents) {
9598
if (doc.relative_path.startsWith('..')) {
9699
continue;
@@ -103,11 +106,15 @@ function snapshotAction(snapshotRoot: string, options: SnapshotOptions): void {
103106
const outputPath = path.resolve(outputDirectory, snapshotDir, relativeToInputDirectory);
104107

105108
if (options.check) {
106-
diffSnapshot(outputPath, obtained);
109+
const diffResult = diffSnapshot(outputPath, obtained);
110+
hasDiff = hasDiff || diffResult === 'different';
107111
} else {
108112
writeSnapshot(outputPath, obtained);
109113
}
110114
}
115+
if (hasDiff) {
116+
exit(1);
117+
}
111118
}
112119
}
113120

Lines changed: 179 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,179 @@
1+
import * as fs from 'fs';
2+
import * as path from 'path';
3+
import { join } from 'path';
4+
5+
export interface TestFailure {
6+
testName: string;
7+
type: 'empty-scip-index' | 'missing-output' | 'content-mismatch' | 'orphaned-output';
8+
message: string;
9+
}
10+
11+
export interface ValidationResults {
12+
passed: string[];
13+
failed: TestFailure[];
14+
skipped: string[];
15+
}
16+
17+
export interface TestRunnerOptions {
18+
snapshotRoot: string;
19+
filterTests?: string;
20+
failFast: boolean;
21+
quiet: boolean;
22+
mode: 'check' | 'update';
23+
}
24+
25+
export interface SingleTestOptions {
26+
check: boolean;
27+
quiet: boolean;
28+
}
29+
30+
function validateFilterTestNames(inputDirectory: string, filterTestNames: string[]): void {
31+
const availableTests = fs.readdirSync(inputDirectory);
32+
const missingTests = filterTestNames.filter(name => !availableTests.includes(name));
33+
34+
if (missingTests.length > 0) {
35+
console.error(`ERROR: The following test names were not found: ${missingTests.join(', ')}. Available tests: ${availableTests.join(', ')}`);
36+
process.exit(1);
37+
}
38+
}
39+
40+
function handleOrphanedOutputs(inputDirectory: string, outputDirectory: string, mode: 'check' | 'update'): TestFailure[] {
41+
if (!fs.existsSync(outputDirectory)) {
42+
return [];
43+
}
44+
45+
const inputTests = new Set(fs.readdirSync(inputDirectory));
46+
const outputTests = fs.readdirSync(outputDirectory);
47+
const orphanedOutputs: TestFailure[] = [];
48+
49+
for (const outputTest of outputTests) {
50+
if (!inputTests.has(outputTest)) {
51+
if (mode === 'update') {
52+
// Delete orphaned output directory in update mode
53+
const orphanedPath = path.join(outputDirectory, outputTest);
54+
fs.rmSync(orphanedPath, { recursive: true, force: true });
55+
console.log(`Delete output folder with no corresponding input folder: ${outputTest}`);
56+
} else {
57+
// Report as failure in check mode
58+
orphanedOutputs.push({
59+
testName: outputTest,
60+
type: 'orphaned-output',
61+
message: `Output folder exists but no corresponding input folder found`
62+
});
63+
}
64+
}
65+
}
66+
67+
return orphanedOutputs;
68+
}
69+
70+
function reportResults(results: ValidationResults): void {
71+
const totalTests = results.passed.length + results.failed.length + results.skipped.length;
72+
console.assert(totalTests > 0, 'No tests found');
73+
74+
for (const failure of results.failed) {
75+
console.error(`FAIL [${failure.testName}]: ${failure.message}`);
76+
}
77+
78+
let summaryStr = `\n${results.passed.length}/${totalTests} tests passed, ${results.failed.length} failed`;
79+
if (results.skipped.length > 0) {
80+
summaryStr += `, ${results.skipped.length} skipped`;
81+
}
82+
console.log(summaryStr);
83+
84+
if (results.failed.length > 0) {
85+
process.exit(1);
86+
}
87+
}
88+
89+
export class TestRunner {
90+
constructor(private options: TestRunnerOptions) {}
91+
92+
runTests(
93+
runSingleTest: (testName: string, inputDir: string, outputDir: string) => ValidationResults
94+
): void {
95+
const inputDirectory = path.resolve(join(this.options.snapshotRoot, 'input'));
96+
const outputDirectory = path.resolve(join(this.options.snapshotRoot, 'output'));
97+
98+
const results: ValidationResults = {
99+
passed: [],
100+
failed: [],
101+
skipped: []
102+
};
103+
104+
// Pre-execution validation: determine test directories to process
105+
let snapshotDirectories = fs.readdirSync(inputDirectory);
106+
let isFilterMode = false;
107+
108+
if (this.options.filterTests) {
109+
// Filter to specific tests
110+
const filterTestNames = this.options.filterTests.split(',').map(name => name.trim());
111+
isFilterMode = true;
112+
113+
// Validate filter test names exist
114+
validateFilterTestNames(inputDirectory, filterTestNames);
115+
116+
snapshotDirectories = snapshotDirectories.filter(dir => filterTestNames.includes(dir));
117+
}
118+
119+
// In non-filtering mode, handle orphaned outputs
120+
if (!isFilterMode) {
121+
const orphanedOutputs = handleOrphanedOutputs(inputDirectory, outputDirectory, this.options.mode);
122+
123+
// In check mode, orphaned outputs are reported as failures
124+
if (orphanedOutputs.length > 0) {
125+
results.failed.push(...orphanedOutputs);
126+
127+
if (this.options.failFast) {
128+
reportResults(results);
129+
return;
130+
}
131+
}
132+
}
133+
134+
for (let i = 0; i < snapshotDirectories.length; i++) {
135+
const testName = snapshotDirectories[i];
136+
if (!this.options.quiet) {
137+
console.log(`Processing test: ${testName}`);
138+
}
139+
140+
try {
141+
const testResults = runSingleTest(
142+
testName,
143+
inputDirectory,
144+
outputDirectory,
145+
);
146+
147+
// Merge results
148+
results.passed.push(...testResults.passed);
149+
results.failed.push(...testResults.failed);
150+
151+
// Check for fail-fast condition
152+
if (this.options.failFast && testResults.failed.length > 0) {
153+
// Track remaining tests as skipped
154+
for (let j = i + 1; j < snapshotDirectories.length; j++) {
155+
results.skipped.push(snapshotDirectories[j]);
156+
}
157+
reportResults(results);
158+
return;
159+
}
160+
} catch (error) {
161+
results.failed.push({
162+
testName,
163+
type: 'empty-scip-index',
164+
message: `Test runner failed: ${error}`
165+
});
166+
167+
if (this.options.failFast) {
168+
for (let j = i + 1; j < snapshotDirectories.length; j++) {
169+
results.skipped.push(snapshotDirectories[j]);
170+
}
171+
reportResults(results);
172+
return;
173+
}
174+
}
175+
}
176+
177+
reportResults(results);
178+
}
179+
}

0 commit comments

Comments
 (0)