Files
2026-07-13 13:02:44 +08:00

37 lines
1.1 KiB
JavaScript

const test = require('node:test');
const assert = require('node:assert/strict');
const {
pickBinaryAssetForPlatform,
getBundledBinaryResourceDir,
} = require('../electron/lib/localInferenceAssets');
test('pickBinaryAssetForPlatform prefers native linux arm64 assets', () => {
const zipNames = [
'sd-master-abc-bin-Linux-Ubuntu-24.04-x86_64.zip',
'sd-master-abc-bin-Linux-Ubuntu-24.04-aarch64.zip',
'sd-master-abc-bin-Linux-Ubuntu-24.04-aarch64-vulkan.zip',
];
const picked = pickBinaryAssetForPlatform({
platform: 'linux',
arch: 'arm64',
zipNames,
});
assert.equal(picked, 'sd-master-abc-bin-Linux-Ubuntu-24.04-aarch64.zip');
});
test('getBundledBinaryResourceDir resolves linux arm64 bundled path', () => {
const bundledDir = getBundledBinaryResourceDir({
resourcesPath: '/opt/Open Generative AI/resources',
platform: 'linux',
arch: 'arm64',
});
assert.equal(
bundledDir,
'/opt/Open Generative AI/resources/local-ai/linux-arm64/bin'
);
});