Fix testcode

Author: owjs3901

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

@@ -1,14 +1,18 @@
-import { describe, expect, it, vi } from 'bun:test'
+import { afterAll, describe, expect, it, mock } from 'bun:test'
 import { canBeAbsolute, getPositionProps } from '../position'
 
-vi.mock('../../utils/check-asset-node', () => ({
+mock.module('../../utils/check-asset-node', () => ({
   checkAssetNode: () => null,
 }))
 
-vi.mock('../../utils/is-page-root', () => ({
+mock.module('../../utils/is-page-root', () => ({
   isPageRoot: () => false,
 }))
 
+afterAll(() => {
+  mock.restore()
+})
+
 describe('position', () => {
   describe('canBeAbsolute', () => {
     it('should return true for ABSOLUTE positioned node', () => {

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

@@ -1,12 +1,16 @@
-import { beforeEach, describe, expect, mock, test, vi } from 'bun:test'
+import { afterAll, beforeEach, describe, expect, mock, test } from 'bun:test'
 import { paintToCSS } from '../paint-to-css'
 import { resetVariableCache } from '../variable-cache'
 
 // mock asset checker to avoid real node handling
-vi.mock('../check-asset-node', () => ({
+mock.module('../check-asset-node', () => ({
   checkAssetNode: () => 'png',
 }))
 
+afterAll(() => {
+  mock.restore()
+})
+
 describe('paintToCSS', () => {
   beforeEach(() => {
     resetVariableCache()

src/commands/__tests__/exportPagesAndComponents.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, test, vi } from 'bun:test'
+import { afterAll, describe, expect, mock, test } from 'bun:test'
 import {
   collectAssetNodes,
   DEVUP_COMPONENTS,
@@ -7,7 +7,7 @@ import {
   generateImportStatements,
 } from '../exportPagesAndComponents'
 
-vi.mock('../../codegen/utils/check-asset-node', () => ({
+mock.module('../../codegen/utils/check-asset-node', () => ({
   checkAssetNode: (node: {
     type: string
     isAsset?: boolean
@@ -20,6 +20,10 @@ vi.mock('../../codegen/utils/check-asset-node', () => ({
   },
 }))
 
+afterAll(() => {
+  mock.restore()
+})
+
 describe('DEVUP_COMPONENTS', () => {
   test('should contain expected devup-ui components', () => {
     expect(DEVUP_COMPONENTS).toContain('Box')