-
Notifications
You must be signed in to change notification settings - Fork 174
/
index.ts
76 lines (65 loc) · 2.28 KB
/
index.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import * as fs from 'fs';
import { ICompilerResult, LLParse } from 'llparse';
import { Dot } from 'llparse-dot';
import {
Fixture, FixtureResult, IFixtureBuildOptions,
} from 'llparse-test-fixture';
import * as path from 'path';
import * as llhttp from '../../src/llhttp';
export type TestType = 'request' | 'response' | 'request-lenient' |
'request-finish' | 'response-finish' | 'none' | 'url';
export { FixtureResult };
const BUILD_DIR = path.join(__dirname, '..', 'tmp');
const CHEADERS_FILE = path.join(BUILD_DIR, 'cheaders.h');
const cheaders = new llhttp.CHeaders().build();
try {
fs.mkdirSync(BUILD_DIR);
} catch (e) {
// no-op
}
fs.writeFileSync(CHEADERS_FILE, cheaders);
const fixtures = new Fixture({
buildDir: path.join(__dirname, '..', 'tmp'),
extra: [
'-msse4.2',
'-DLLHTTP__TEST',
'-DLLPARSE__ERROR_PAUSE=' + llhttp.constants.ERROR.PAUSED,
'-include', CHEADERS_FILE,
path.join(__dirname, 'extra.c'),
],
maxParallel: process.env.LLPARSE_DEBUG ? 1 : undefined,
});
const cache: Map<any, ICompilerResult> = new Map();
export function build(llparse: LLParse, node: any, outFile: string,
options: IFixtureBuildOptions = {},
ty: TestType = 'none'): FixtureResult {
const dot = new Dot();
fs.writeFileSync(path.join(BUILD_DIR, outFile + '.dot'),
dot.build(node));
let artifacts: ICompilerResult;
if (cache.has(node)) {
artifacts = cache.get(node)!;
} else {
artifacts = llparse.build(node, {
c: { header: outFile },
debug: process.env.LLPARSE_DEBUG ? 'llparse__debug' : undefined,
generateJS: false,
});
cache.set(node, artifacts);
}
const extra = options.extra === undefined ? [] : options.extra.slice();
if (ty === 'request' || ty === 'response' || ty === 'request-lenient') {
extra.push(
`-DLLPARSE__TEST_INIT=llhttp__test_init_${ty.replace(/-/g, '_')}`);
} else if (ty === 'request-finish' || ty === 'response-finish') {
if (ty === 'request-finish') {
extra.push('-DLLPARSE__TEST_INIT=llhttp__test_init_request');
} else {
extra.push('-DLLPARSE__TEST_INIT=llhttp__test_init_response');
}
extra.push('-DLLPARSE__TEST_FINISH=llhttp__test_finish');
}
return fixtures.build(artifacts, outFile, Object.assign(options, {
extra,
}));
}