forked from reactjs/react-docgen
/
resolveHOC-test.js
57 lines (49 loc) · 1.57 KB
/
resolveHOC-test.js
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
/**
* Copyright (c) Facebook, Inc. and its affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*
*/
import types from 'ast-types';
import resolveHOC from '../resolveHOC';
import * as utils from '../../../tests/utils';
const { builders } = types;
describe('resolveHOC', () => {
function parse(src) {
const root = utils.parse(src);
return root.get('body', root.node.body.length - 1, 'expression');
}
it('resolves simple hoc', () => {
const path = parse(['hoc(42);'].join('\n'));
expect(resolveHOC(path)).toEqualASTNode(builders.literal(42));
});
it('resolves simple hoc w/ multiple args', () => {
const path = parse(['hoc1(arg1a, arg1b)(42);'].join('\n'));
expect(resolveHOC(path)).toEqualASTNode(builders.literal(42));
});
it('resolves nested hocs', () => {
const path = parse(
['hoc2(arg2b, arg2b)(', ' hoc1(arg1a, arg2a)(42)', ');'].join('\n'),
);
expect(resolveHOC(path)).toEqualASTNode(builders.literal(42));
});
it('resolves really nested hocs', () => {
const path = parse(
[
'hoc3(arg3a, arg3b)(',
' hoc2(arg2b, arg2b)(',
' hoc1(arg1a, arg2a)(42)',
' )',
');',
].join('\n'),
);
expect(resolveHOC(path)).toEqualASTNode(builders.literal(42));
});
it('resolves intermediate hocs', () => {
const path = parse(
['const Component = React.memo(42);', 'hoc()(Component);'].join('\n'),
);
expect(resolveHOC(path)).toEqualASTNode(builders.literal(42));
});
});