Fix testcode

Author: owjs3901

src/codegen/props/__tests__/position.test.ts

@@ -1,16 +1,20 @@
-import { afterAll, describe, expect, it, mock } from 'bun:test'
+import { afterAll, describe, expect, it, spyOn } from 'bun:test'
+import * as checkAssetNodeModule from '../../utils/check-asset-node'
+import * as isPageRootModule from '../../utils/is-page-root'
 import { canBeAbsolute, getPositionProps } from '../position'
 
-mock.module('../../utils/check-asset-node', () => ({
-  checkAssetNode: () => null,
-}))
+const checkAssetNodeSpy = spyOn(
+  checkAssetNodeModule,
+  'checkAssetNode',
+).mockReturnValue(null)
 
-mock.module('../../utils/is-page-root', () => ({
-  isPageRoot: () => false,
-}))
+const isPageRootSpy = spyOn(isPageRootModule, 'isPageRoot').mockReturnValue(
+  false,
+)
 
 afterAll(() => {
-  mock.restore()
+  checkAssetNodeSpy.mockRestore()
+  isPageRootSpy.mockRestore()
 })
 
 describe('position', () => {

src/codegen/utils/__tests__/paint-to-css.test.ts

@@ -1,14 +1,24 @@
-import { afterAll, beforeEach, describe, expect, mock, test } from 'bun:test'
+import {
+  afterAll,
+  beforeEach,
+  describe,
+  expect,
+  mock,
+  spyOn,
+  test,
+} from 'bun:test'
+import * as checkAssetNodeModule from '../check-asset-node'
 import { paintToCSS } from '../paint-to-css'
 import { resetVariableCache } from '../variable-cache'
 
 // mock asset checker to avoid real node handling
-mock.module('../check-asset-node', () => ({
-  checkAssetNode: () => 'png',
-}))
+const checkAssetNodeSpy = spyOn(
+  checkAssetNodeModule,
+  'checkAssetNode',
+).mockReturnValue('png' as const)
 
 afterAll(() => {
-  mock.restore()
+  checkAssetNodeSpy.mockRestore()
 })
 
 describe('paintToCSS', () => {

src/commands/__tests__/exportPagesAndComponents.test.ts

@@ -1,4 +1,5 @@
-import { afterAll, describe, expect, mock, test } from 'bun:test'
+import { afterAll, describe, expect, spyOn, test } from 'bun:test'
+import * as checkAssetNodeModule from '../../codegen/utils/check-asset-node'
 import {
   collectAssetNodes,
   DEVUP_COMPONENTS,
@@ -7,21 +8,21 @@ import {
   generateImportStatements,
 } from '../exportPagesAndComponents'
 
-mock.module('../../codegen/utils/check-asset-node', () => ({
-  checkAssetNode: (node: {
-    type: string
-    isAsset?: boolean
-  }): string | null => {
+const checkAssetNodeSpy = spyOn(
+  checkAssetNodeModule,
+  'checkAssetNode',
+).mockImplementation(
+  (node: { type: string; isAsset?: boolean }): 'svg' | 'png' | null => {
     if (node.type === 'VECTOR') return 'svg'
     if (node.type === 'STAR') return 'svg'
     if (node.type === 'POLYGON') return 'svg'
     if (node.isAsset && node.type === 'RECTANGLE') return 'png'
     return null
   },
-}))
+)
 
 afterAll(() => {
-  mock.restore()
+  checkAssetNodeSpy.mockRestore()
 })
 
 describe('DEVUP_COMPONENTS', () => {