Fix another test, allow it to use callExpressionKw

This commit is contained in:
Adam Chalmers
2025-01-14 16:45:49 -06:00
committed by Nick Cameron
parent 9e778280b3
commit 9dc06f6b92

View File

@ -6,6 +6,7 @@ import {
getXComponent, getXComponent,
addCloseToPipe, addCloseToPipe,
getConstraintInfo, getConstraintInfo,
getConstraintInfoKw,
} from './sketch' } from './sketch'
import { import {
assertParse, assertParse,
@ -13,6 +14,7 @@ import {
initPromise, initPromise,
CallExpression, CallExpression,
topLevelRange, topLevelRange,
CallExpressionKw,
} from '../wasm' } from '../wasm'
import { getNodeFromPath, getNodePathFromSourceRange } from '../queryAst' import { getNodeFromPath, getNodePathFromSourceRange } from '../queryAst'
import { enginelessExecutor } from '../../lib/testHelpers' import { enginelessExecutor } from '../../lib/testHelpers'
@ -689,13 +691,16 @@ describe('testing getConstraintInfo', () => {
) )
if (err(ast)) return ast if (err(ast)) return ast
const pathToNode = getNodePathFromSourceRange(ast, sourceRange) const pathToNode = getNodePathFromSourceRange(ast, sourceRange)
const callExp = getNodeFromPath<Node<CallExpression>>( const callExp = getNodeFromPath<Node<CallExpression | CallExpressionKw>>(
ast, ast,
pathToNode, pathToNode,
'CallExpression' ['CallExpression', 'CallExpressionKw']
) )
if (err(callExp)) return callExp if (err(callExp)) return callExp
const result = getConstraintInfo(callExp.node, code, pathToNode) const result =
callExp.node.type === 'CallExpression'
? getConstraintInfo(callExp.node, code, pathToNode)
: getConstraintInfoKw(callExp.node, code, pathToNode)
expect(result).toEqual(expected) expect(result).toEqual(expected)
}) })
}) })
@ -843,13 +848,16 @@ describe('testing getConstraintInfo', () => {
) )
if (err(ast)) return ast if (err(ast)) return ast
const pathToNode = getNodePathFromSourceRange(ast, sourceRange) const pathToNode = getNodePathFromSourceRange(ast, sourceRange)
const callExp = getNodeFromPath<Node<CallExpression>>( const callExp = getNodeFromPath<Node<CallExpression | CallExpressionKw>>(
ast, ast,
pathToNode, pathToNode,
'CallExpression' ['CallExpression', 'CallExpressionKw']
) )
if (err(callExp)) return callExp if (err(callExp)) return callExp
const result = getConstraintInfo(callExp.node, code, pathToNode) const result =
callExp.node.type === 'CallExpression'
? getConstraintInfo(callExp.node, code, pathToNode)
: getConstraintInfoKw(callExp.node, code, pathToNode)
expect(result).toEqual(expected) expect(result).toEqual(expected)
}) })
}) })
@ -1199,14 +1207,17 @@ describe('testing getConstraintInfo', () => {
) )
if (err(ast)) return ast if (err(ast)) return ast
const pathToNode = getNodePathFromSourceRange(ast, sourceRange) const pathToNode = getNodePathFromSourceRange(ast, sourceRange)
const callExp = getNodeFromPath<Node<CallExpression>>( const callExp = getNodeFromPath<Node<CallExpression | CallExpressionKw>>(
ast, ast,
pathToNode, pathToNode,
'CallExpression' ['CallExpression', 'CallExpressionKw']
) )
if (err(callExp)) return callExp if (err(callExp)) return callExp
const result = getConstraintInfo(callExp.node, code, pathToNode) const result =
callExp.node.type === 'CallExpression'
? getConstraintInfo(callExp.node, code, pathToNode)
: getConstraintInfoKw(callExp.node, code, pathToNode)
expect(result).toEqual(expected) expect(result).toEqual(expected)
}) })
}) })