wehub-resource-sync
50 lines
1.7 KiB
JavaScript
50 lines
1.7 KiB
JavaScript
const test = require('node:test');
|
|
const assert = require('node:assert/strict');
|
|
const path = require('node:path');
|
|
|
|
const {
|
|
LOCAL_AI_DIR_ENV,
|
|
resolveLocalAiPaths,
|
|
} = require('../electron/lib/localInferencePaths');
|
|
|
|
test('resolveLocalAiPaths defaults under the Electron user data directory', () => {
|
|
const userDataPath = path.join(process.cwd(), 'fixtures', 'user-data');
|
|
const paths = resolveLocalAiPaths({ userDataPath, env: {} });
|
|
const dataDir = path.resolve(userDataPath, 'local-ai');
|
|
|
|
assert.deepEqual(paths, {
|
|
dataDir,
|
|
binDir: path.join(dataDir, 'bin'),
|
|
modelsDir: path.join(dataDir, 'models'),
|
|
tmpDir: path.join(dataDir, 'tmp'),
|
|
});
|
|
});
|
|
|
|
test('resolveLocalAiPaths honors a custom local AI directory', () => {
|
|
const customDir = path.join(process.cwd(), 'fixtures', 'custom-ai-store');
|
|
const paths = resolveLocalAiPaths({
|
|
userDataPath: path.join(process.cwd(), 'fixtures', 'ignored-user-data'),
|
|
env: { [LOCAL_AI_DIR_ENV]: customDir },
|
|
});
|
|
const dataDir = path.resolve(customDir);
|
|
|
|
assert.equal(paths.dataDir, dataDir);
|
|
assert.equal(paths.modelsDir, path.join(dataDir, 'models'));
|
|
});
|
|
|
|
test('resolveLocalAiPaths trims accidental whitespace around the override', () => {
|
|
const customDir = path.join(process.cwd(), 'fixtures', 'spaced-ai-store');
|
|
const paths = resolveLocalAiPaths({
|
|
env: { [LOCAL_AI_DIR_ENV]: ` ${customDir} ` },
|
|
});
|
|
|
|
assert.equal(paths.dataDir, path.resolve(customDir));
|
|
});
|
|
|
|
test('resolveLocalAiPaths requires userDataPath when no override is set', () => {
|
|
assert.throws(
|
|
() => resolveLocalAiPaths({ env: {} }),
|
|
/userDataPath is required/
|
|
);
|
|
});
|