ZenEcho
489f3d4f1f
- 新增 BattleNet、WeGame、GOG 等游戏平台扫描支持 - 添加系统代理绕过列表和 Bootstrap DNS 配置 - 实现应用重置功能，支持清理数据并重启 - 增加主题模式选择（亮色/暗色/自动） - 优化本地游戏扫描性能，支持并行扫描 - 改进代理监控，支持链式代理子进程检测 - 修复日志发送时的窗口销毁检查 - 更新图标和侧边栏布局 - 移除 prepare-singbox-core 脚本，简化构建流程
92 lines
2.5 KiB
TypeScript
92 lines
2.5 KiB
TypeScript
import { describe, expect, it, vi } from 'vitest'
|
|
import {
|
|
parsePowerShellProcessJson,
|
|
parseWmicProcessCsv,
|
|
type ProcessNode
|
|
} from '../../electron/services/process'
|
|
|
|
vi.mock('electron', () => ({
|
|
ipcMain: {
|
|
handle: vi.fn()
|
|
}
|
|
}))
|
|
|
|
function findNodeByPid(nodes: ProcessNode[], pid: number): ProcessNode | null {
|
|
const stack = [...nodes]
|
|
while (stack.length > 0) {
|
|
const node = stack.pop()
|
|
if (!node) continue
|
|
if (node.pid === pid) return node
|
|
if (node.children?.length) {
|
|
stack.push(...node.children)
|
|
}
|
|
}
|
|
return null
|
|
}
|
|
|
|
describe('process tree parser', () => {
|
|
it('parses PowerShell/CIM json array output', () => {
|
|
const json = JSON.stringify([
|
|
{
|
|
ProcessId: 100,
|
|
ParentProcessId: 4,
|
|
Name: 'launcher.exe',
|
|
ExecutablePath: 'C:\\Games\\Launcher\\launcher.exe'
|
|
},
|
|
{
|
|
ProcessId: 200,
|
|
ParentProcessId: 100,
|
|
Name: 'game.exe',
|
|
ExecutablePath: 'C:\\Games\\Launcher\\game.exe'
|
|
},
|
|
{
|
|
ProcessId: 300,
|
|
ParentProcessId: 0,
|
|
Name: 'other.exe',
|
|
ExecutablePath: 'C:\\Tools\\other.exe'
|
|
}
|
|
])
|
|
|
|
const tree = parsePowerShellProcessJson(json)
|
|
const rootPids = tree.map((node) => node.pid).sort((a, b) => a - b)
|
|
expect(rootPids).toEqual([100, 300])
|
|
|
|
const launcher = findNodeByPid(tree, 100)
|
|
expect(launcher).not.toBeNull()
|
|
expect(launcher!.children.map((child) => child.pid)).toEqual([200])
|
|
})
|
|
|
|
it('parses PowerShell/CIM json single-object output', () => {
|
|
const json = JSON.stringify({
|
|
ProcessId: 501,
|
|
ParentProcessId: 1,
|
|
Name: 'single.exe',
|
|
ExecutablePath: 'C:\\single.exe'
|
|
})
|
|
|
|
const tree = parsePowerShellProcessJson(json)
|
|
expect(tree.length).toBe(1)
|
|
expect(tree[0]?.pid).toBe(501)
|
|
expect(tree[0]?.name).toBe('single.exe')
|
|
})
|
|
|
|
it('parses WMIC csv output with comma in executable path', () => {
|
|
const csv = [
|
|
'Node,ExecutablePath,Name,ParentProcessId,ProcessId',
|
|
'DESKTOP,C:\\Games\\A,B\\launcher.exe,launcher.exe,4,100',
|
|
'DESKTOP,C:\\Games\\A,B\\game.exe,game.exe,100,200'
|
|
].join('\r\n')
|
|
|
|
const tree = parseWmicProcessCsv(csv)
|
|
const launcher = findNodeByPid(tree, 100)
|
|
const game = findNodeByPid(tree, 200)
|
|
|
|
expect(launcher).not.toBeNull()
|
|
expect(game).not.toBeNull()
|
|
expect(launcher?.path).toBe('C:\\Games\\A,B\\launcher.exe')
|
|
expect(game?.path).toBe('C:\\Games\\A,B\\game.exe')
|
|
expect(launcher?.children.some((child) => child.pid === 200)).toBe(true)
|
|
})
|
|
})
|
|
|