Ci/vitest coverage

EmbraceAn · whale · commit dca8879394b7 · 2026-01-06T14:54:14.000+08:00
* test: init create vitest * ci: add vitest coverage, junit reporter and improve yarn cache strategy Ci/vitest coverage (#3253) * ci: pin node 18.18.2, add inputs display and yarn cache-dependency-path * ci: debug test discovery + set vitest root/include to ensure tests are found in CI * test: move Vitest__Test__ to __tests__ so CI includes it * fix(test): set spinner false when no editorOperationRecord
diff --git a/.github/workflows/Essential-tests.yml b/.github/workflows/Essential-tests.yml
@@ -0,0 +1,77 @@
+name: Essential tests
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+on:
+  pull_request:
+    branches: [test]
+    paths:
+      - 'app/renderer/src/main/src/release/**'
+      - '.github/workflows/**'
+
+jobs:
+  vitest:
+    name: Vitest - renderer main
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        node-version: [18.x]
+    steps:
+      - name: Display incoming configuration parameters
+        run: echo ${{ inputs.platform }} ${{ inputs.legacy }} ${{ inputs.version }} ${{ inputs.engine }} ${{ inputs.engineVersion }} ${{ inputs.devTool }}
+
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - uses: actions/setup-node@v4
+        with:
+          node-version: 18.18.2
+          cache: "yarn"
+          cache-dependency-path: |
+              yarn.lock
+              app/renderer/src/main/yarn.lock
+              app/renderer/engine-link-startup/yarn.lock
+
+      - name: Cache Yarn cache
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/yarn
+          key: ${{ runner.os }}-yarn-cache-${{ hashFiles('app/renderer/src/main/yarn.lock') }}
+          restore-keys: |
+            ${{ runner.os }}-yarn-cache-
+
+      - name: Cache node_modules
+        uses: actions/cache@v4
+        with:
+          path: app/renderer/src/main/node_modules
+          key: ${{ runner.os }}-node-modules-${{ hashFiles('app/renderer/src/main/yarn.lock') }}
+          restore-keys: |
+            ${{ runner.os }}-node-modules-
+
+      - name: Install dependencies (renderer main)
+        working-directory: app/renderer/src/main
+        run: yarn install --frozen-lockfile --prefer-offline --network-concurrency 1
+
+      - name: Verify installed packages (quick)
+        working-directory: app/renderer/src/main
+        run: |
+          echo "Yarn cache dir: $(yarn cache dir)"
+          ls -la node_modules | head -n 20 || true
+
+      - name: Debug - list candidate test files
+        working-directory: app/renderer/src/main
+        run: |
+          echo "pwd: $(pwd)"
+          echo "node: $(node -v) yarn: $(yarn -v)"
+          echo "Find test/spec files (up to depth 6):"
+          find . -maxdepth 6 -type f \( -name "*.test.*" -o -name "*.spec.*" \) -print || true
+          echo "Show specific folder (if exists):"
+          ls -la src/components/playground/Vitest__Test__ || true
+
+      - name: Run Vitest (renderer main)
+        working-directory: app/renderer/src/main
+        env:
+          CI: true
+        run: yarn test:vitest -- --run --reporter verbose
diff --git a/app/renderer/src/main/package.json b/app/renderer/src/main/package.json
@@ -143,6 +143,7 @@
         "build-test-memfit": "env-cmd -e memfit,devTool -f ./.env-cmdrc cross-env GENERATE_SOURCEMAP=false react-app-rewired build",
         "build-breachtrace": "env-cmd -e breachtrace -f ./.env-cmdrc cross-env GENERATE_SOURCEMAP=false react-app-rewired build",
         "test": "react-app-rewired test",
+        "test:vitest": "vitest",
         "eject": "react-app-rewired eject",
         "analyzer": "env-cmd -e analyzer -f cross-env react-app-rewired build",
         "postinstall": "patch-package",
@@ -180,7 +181,10 @@
         "sass": "^1.54.8",
         "sass-loader": "^12.6.0",
         "source-map-loader": "^4.0.2",
-        "webpack-bundle-analyzer": "^4.10.1"
+        "webpack-bundle-analyzer": "^4.10.1",
+        "vitest": "^0.34.6",
+        "vite": "^4.4.9",
+        "vite-tsconfig-paths": "^3.5.0"
     },
     "resolutions": {
         "//": "See https://github.com/facebook/create-react-app/issues/11773",
diff --git a/app/renderer/src/main/src/__tests__/playground/Vitest__Test__/index.test.tsx b/app/renderer/src/main/src/__tests__/playground/Vitest__Test__/index.test.tsx
@@ -0,0 +1,66 @@
+import React from "react"
+import {render, waitFor, screen} from "@testing-library/react"
+import {afterEach, beforeEach, describe, expect, it, vi} from "vitest"
+
+vi.mock("@/components/yakitUI/YakitSpin/YakitSpin", () => {
+    const React = require("react")
+    return {
+        YakitSpin: (props: any) => {
+            const {spinning, children} = props
+            return React.createElement(
+                "div",
+                {"data-spinning": spinning ? "true" : "false", "data-testid": "yakitspin"},
+                children
+            )
+        }
+    }
+})
+
+const mockGetRemoteValue = vi.fn()
+vi.mock("@/utils/kv", () => ({
+    getRemoteValue: (...args: any[]) => mockGetRemoteValue(...args)
+}))
+
+vi.mock("ahooks", () => {
+    const React = require("react")
+    return {
+        useSafeState: (init: any) => React.useState(init)
+    }
+})
+
+let Vitest__Test__: any
+beforeAll(async () => {
+    const mod = await import("@/components/playground/Vitest__Test__")
+    Vitest__Test__ = mod.Vitest__Test__
+})
+
+beforeEach(() => {
+    mockGetRemoteValue.mockReset()
+})
+
+afterEach(() => {
+    vi.restoreAllMocks()
+})
+
+describe("Vitest__Test__ component - loading behavior", () => {
+    it("sets spinning=false immediately when no editorOperationRecord is provided", async () => {
+        render(React.createElement(Vitest__Test__))
+        const spin = screen.getByTestId("yakitspin")
+        expect(spin).toBeInTheDocument()
+        expect(spin).toHaveAttribute("data-spinning", "false")
+    })
+
+    it("shows", async () => {
+        mockGetRemoteValue.mockImplementation(() => Promise.resolve(JSON.stringify({fontSize: 14})))
+
+        render(React.createElement(Vitest__Test__, {editorOperationRecord: "my-key"}))
+
+        const spin = screen.getByTestId("yakitspin")
+        expect(spin).toBeInTheDocument()
+        expect(spin).toHaveAttribute("data-spinning", "true")
+
+        await waitFor(() => {
+            expect(spin).toHaveAttribute("data-spinning", "false")
+        })
+    })
+})
diff --git a/app/renderer/src/main/src/components/playground/Vitest__Test__/index.tsx b/app/renderer/src/main/src/components/playground/Vitest__Test__/index.tsx
@@ -0,0 +1,57 @@
+import {OperationRecordRes} from "@/components/yakitUI/YakitEditor/YakitEditorType"
+import {YakitSpin} from "@/components/yakitUI/YakitSpin/YakitSpin"
+import {getRemoteValue} from "@/utils/kv"
+import {useSafeState} from "ahooks"
+import {FC, useEffect} from "react"
+
+interface VitestTestProps {
+    editorOperationRecord?: string
+}
+
+const Vitest__Test__: FC<VitestTestProps> = (props) => {
+    const {editorOperationRecord} = props
+    const [typeLoading, setTypeLoading] = useSafeState<boolean>(true)
+    const [fontSize, setFontSize] = useSafeState<undefined | number>(12)
+    const [showLineBreaks, setShowLineBreaks] = useSafeState<boolean>(true)
+    const [noWordwrap, setNoWordwrap] = useSafeState(false)
+
+    useEffect(() => {
+        if (!editorOperationRecord) {
+            setTypeLoading(false)
+            return
+        }
+        setTypeLoading(true)
+        getRemoteValue(editorOperationRecord)
+            .then((data) => {
+                try {
+                    setTypeLoading(false)
+                    if (!data) return
+                    let obj: OperationRecordRes = JSON.parse(data)
+                    if (obj?.fontSize) {
+                        setFontSize(obj?.fontSize)
+                    }
+                    if (typeof obj?.showBreak === "boolean") {
+                        setShowLineBreaks(obj?.showBreak)
+                    }
+                    if (typeof obj?.noWordWrap === "boolean") {
+                        setNoWordwrap(obj?.noWordWrap)
+                    }
+                } catch (error) {
+                    setTypeLoading(false)
+                    fail(error + "")
+                }
+            })
+            .finally(() => {
+                setTypeLoading(false)
+            })
+        // eslint-disable-next-line react-hooks/exhaustive-deps
+    }, [editorOperationRecord])
+
+    return (
+        <YakitSpin spinning={typeLoading}>
+            <div>Vitest__Test__</div>
+        </YakitSpin>
+    )
+}
+
+export {Vitest__Test__}
diff --git a/app/renderer/src/main/src/pages/debugMonaco/DebugMonacoEditorPage.tsx b/app/renderer/src/main/src/pages/debugMonaco/DebugMonacoEditorPage.tsx
@@ -5,8 +5,8 @@ import {info} from "@/utils/notification"
 import {SelectOne} from "@/utils/inputUtil"
 import {YakitEditor} from "@/components/yakitUI/YakitEditor/YakitEditor"
 import {YakURLTree} from "@/pages/yakURLTree/YakURLTree"
-import {TrafficDemo} from "@/components/playground/TrafficDemo"
 import {PcapXDemo} from "@/components/playground/PcapXDemo"
+import {Vitest__Test__} from "@/components/playground/Vitest__Test__"
 import {DemoItemSelectOne} from "@/demoComponents/itemSelect/ItemSelect"
 import {RiskTableDemo} from "@/components/playground/RiskTableDemo"
 import {ChaosMakerRulesDemo} from "@/components/playground/ChaosMakerRulesDemo"
@@ -18,7 +18,7 @@ import {JavaDecompilerOperator} from "@/components/playground/javadecompiler/Jav
 import {KnowledgeBaseDemo} from "@/components/playground/knowlegeBase/KnowledgeBaseDemo"
 import {RagManagerDemo} from "@/components/playground/ragManager/RagManagerDemo"
 import {ThirdPartyBinaryManager} from "@/components/playground/thirdPartyBinary/ThirdPartyBinaryManager"
-import { EntityRepositoryPage } from "@/components/playground/entityRepository/EntityRepositoryPage"
+import {EntityRepositoryPage} from "@/components/playground/entityRepository/EntityRepositoryPage"
 export interface DebugMonacoEditorPageProp {}
 
 const TAG = "DEBUG_PLAYGROUND_DEFAULT_MODE"
@@ -77,6 +77,7 @@ a=1&b=2 Content-Length: a
                             {value: "rag-manager", label: "RAG 向量存储管理"},
                             {value: "third-party-binary", label: "第三方应用管理"},
                             {value: "entity-repository", label: "实体仓库"},
+                            {value: "vitest__test__", label: "Vitest 测试组件"}
                         ]}
                         formItemStyle={{margin: 0}}
                         value={mode}
@@ -112,7 +113,9 @@ a=1&b=2 Content-Length: a
                             return <ThirdPartyBinaryManager />
                         case "entity-repository":
                             return <EntityRepositoryPage />
-                        }
+                        case "vitest__test__":
+                            return <Vitest__Test__ />
+                    }
                     return <div>NO PLUGIN DEMO</div>
                 })()}
             </AutoCard>
diff --git a/app/renderer/src/main/vitest.config.ts b/app/renderer/src/main/vitest.config.ts
@@ -0,0 +1,27 @@
+import {defineConfig} from "vitest/config"
+import path from "path"
+
+export default defineConfig({
+    resolve: {
+        alias: {
+            "@": path.resolve(__dirname, "src")
+        }
+    },
+    // Set test root to `src` and explicitly include test file patterns to ensure files are found in CI
+    test: {
+        environment: "jsdom",
+        globals: true,
+        root: path.resolve(__dirname),
+        include: ["src/**/*.test.{ts,tsx,js,jsx}", "src/**/*.spec.{ts,tsx,js,jsx}"],
+        setupFiles: ["src/setupTests.ts"],
+        // Add JUnit reporter (writes `reports/junit.xml`) and keep default reporter
+        // Cast the tuple to `any` to avoid type mismatch on this Vitest version
+        reporters: [["junit", {outputFile: "reports/junit.xml"}] as unknown as any, "default"],
+        // Use a supported coverage provider for Node (v8 or istanbul)
+        coverage: {
+            provider: "v8",
+            reporter: ["lcov", "text"],
+            reportsDirectory: "coverage"
+        }
+    }
+})
diff --git a/app/renderer/src/main/yarn.lock b/app/renderer/src/main/yarn.lock
