Skip to content

Commit ec05b3a

Browse files
committed
feat(context): find app.schema.addToContext calls
closes #1169
1 parent ca53f35 commit ec05b3a

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/lib/add-to-context-extractor/extractor.spec.ts

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,13 @@ describe('syntax cases', () => {
1111
`).types.length
1212
).not.toEqual(0)
1313
})
14+
it('will extract from app.schema.addToContext', () => {
15+
expect(
16+
extractOrThrow(`
17+
app.schema.addToContext(req => ({ a: 1 }))
18+
`).types.length
19+
).not.toEqual(0)
20+
})
1421
describe('does not extract when not relevant AST pattern', () => {
1522
it('case 1', () => {
1623
expect(

src/lib/add-to-context-extractor/extractor.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ export function extractContextTypes(program: ts.Program): Either<Exception, Extr
9090
const expText = exp.getExpression().getText()
9191
const propName = exp.getName()
9292

93-
if (expText !== 'schema' || propName !== 'addToContext') {
93+
if (!((expText === 'schema' || expText === 'app.schema') && propName === 'addToContext')) {
9494
n.forEachChild(visit)
9595
return
9696
}

0 commit comments

Comments
 (0)