wehub-resource-sync
61 lines
1.8 KiB
JavaScript
61 lines
1.8 KiB
JavaScript
const test = require('node:test');
|
|
const assert = require('node:assert/strict');
|
|
|
|
const { withWan2gpAvailability } = require('../electron/lib/wan2gpModelAvailability');
|
|
|
|
const fluxModel = {
|
|
id: 'wan2gp:flux-dev',
|
|
name: 'Flux.1 Dev (Wan2GP)',
|
|
fn: 'flux',
|
|
};
|
|
|
|
test('withWan2gpAvailability marks models unavailable when the server probe fails', () => {
|
|
const model = withWan2gpAvailability(fluxModel, { ok: false, error: 'Request timed out' });
|
|
|
|
assert.equal(model.ready, false);
|
|
assert.equal(model.unavailableReason, 'Request timed out');
|
|
});
|
|
|
|
test('withWan2gpAvailability uses resolved api_name when endpoint metadata matches', () => {
|
|
const model = withWan2gpAvailability(
|
|
fluxModel,
|
|
{ ok: true },
|
|
{
|
|
apiNames: ['flux_image'],
|
|
resolved: new Map([['wan2gp:flux-dev', 'flux_image']]),
|
|
}
|
|
);
|
|
|
|
assert.equal(model.ready, true);
|
|
assert.equal(model.fn, 'flux_image');
|
|
});
|
|
|
|
test('withWan2gpAvailability keeps default api_name available when Gradio omits endpoint metadata', () => {
|
|
const model = withWan2gpAvailability(
|
|
fluxModel,
|
|
{ ok: true },
|
|
{
|
|
apiNames: [],
|
|
resolved: new Map([['wan2gp:flux-dev', null]]),
|
|
}
|
|
);
|
|
|
|
assert.equal(model.ready, true);
|
|
assert.equal(model.fn, 'flux');
|
|
assert.match(model.availabilityNote, /default api_name/);
|
|
});
|
|
|
|
test('withWan2gpAvailability rejects unmatched models when endpoint metadata is present', () => {
|
|
const model = withWan2gpAvailability(
|
|
fluxModel,
|
|
{ ok: true },
|
|
{
|
|
apiNames: ['qwen_image'],
|
|
resolved: new Map([['wan2gp:flux-dev', null]]),
|
|
}
|
|
);
|
|
|
|
assert.equal(model.ready, false);
|
|
assert.match(model.unavailableReason, /no api_name matching "flux"/);
|
|
});
|