95ad629558
* Add number with units formatting as KCL * Change type assertion helper to check what we need * Fix rectangle unit test
93 lines
2.9 KiB
TypeScript
93 lines
2.9 KiB
TypeScript
import { expect } from 'vitest'
|
|
import {
|
|
recast,
|
|
assertParse,
|
|
topLevelRange,
|
|
VariableDeclaration,
|
|
initPromise,
|
|
} from 'lang/wasm'
|
|
import { updateCenterRectangleSketch } from './rectangleTool'
|
|
import { getNodePathFromSourceRange } from 'lang/queryAstNodePathUtils'
|
|
import { getNodeFromPath } from 'lang/queryAst'
|
|
import { findUniqueName } from 'lang/modifyAst'
|
|
import { err, trap } from './trap'
|
|
|
|
beforeAll(async () => {
|
|
await initPromise
|
|
})
|
|
|
|
describe('library rectangleTool helper functions', () => {
|
|
describe('updateCenterRectangleSketch', () => {
|
|
// regression test for https://github.com/KittyCAD/modeling-app/issues/5157
|
|
test('should update AST and source code', async () => {
|
|
// Base source code that will be edited in place
|
|
const sourceCode = `sketch001 = startSketchOn('XZ')
|
|
|> startProfileAt([120.37, 162.76], %)
|
|
|> angledLine([0, 0], %, $rectangleSegmentA001)
|
|
|> angledLine([segAng(rectangleSegmentA001) + 90, 0], %, $rectangleSegmentB001)
|
|
|> angledLine([
|
|
segAng(rectangleSegmentA001),
|
|
-segLen(rectangleSegmentA001)
|
|
], %, $rectangleSegmentC001)
|
|
|> lineTo([profileStartX(%), profileStartY(%)], %)
|
|
|> close(%)
|
|
`
|
|
// Create ast
|
|
const _ast = assertParse(sourceCode)
|
|
let ast = structuredClone(_ast)
|
|
|
|
// Find some nodes and paths to reference
|
|
const sketchSnippet = `startProfileAt([120.37, 162.76], %)`
|
|
const sketchRange = topLevelRange(
|
|
sourceCode.indexOf(sketchSnippet),
|
|
sourceCode.indexOf(sketchSnippet) + sketchSnippet.length
|
|
)
|
|
const sketchPathToNode = getNodePathFromSourceRange(ast, sketchRange)
|
|
const _node = getNodeFromPath<VariableDeclaration>(
|
|
ast,
|
|
sketchPathToNode || [],
|
|
'VariableDeclaration'
|
|
)
|
|
if (trap(_node)) return
|
|
const sketchInit = _node.node?.declaration.init
|
|
|
|
// Hard code inputs that a user would have taken with their mouse
|
|
const x = 40
|
|
const y = 60
|
|
const rectangleOrigin = [120, 180]
|
|
const tags: [string, string, string] = [
|
|
'rectangleSegmentA001',
|
|
'rectangleSegmentB001',
|
|
'rectangleSegmentC001',
|
|
]
|
|
|
|
// Update the ast
|
|
if (sketchInit.type === 'PipeExpression') {
|
|
updateCenterRectangleSketch(
|
|
sketchInit,
|
|
x,
|
|
y,
|
|
tags[0],
|
|
rectangleOrigin[0],
|
|
rectangleOrigin[1]
|
|
)
|
|
}
|
|
|
|
// ast is edited in place from the updateCenterRectangleSketch
|
|
const expectedSourceCode = `sketch001 = startSketchOn('XZ')
|
|
|> startProfileAt([80, 120], %)
|
|
|> angledLine([0, 80], %, $rectangleSegmentA001)
|
|
|> angledLine([segAng(rectangleSegmentA001) + 90, 120], %, $rectangleSegmentB001)
|
|
|> angledLine([
|
|
segAng(rectangleSegmentA001),
|
|
-segLen(rectangleSegmentA001)
|
|
], %, $rectangleSegmentC001)
|
|
|> lineTo([profileStartX(%), profileStartY(%)], %)
|
|
|> close(%)
|
|
`
|
|
const recasted = recast(ast)
|
|
expect(recasted).toEqual(expectedSourceCode)
|
|
})
|
|
})
|
|
})
|