import test from 'node:test'
import assert from 'node:assert/strict'
import { stableMemoryArgs } from '../scripts/comfy-memory-policy.mjs'

test('memory policy preserves user launch flags and never forces a different VRAM loader', () => {
  const args = ['main.py', '--enable-dynamic-vram', '--fast-disk', '--enable-manager', '--fp16-vae']
  assert.deepEqual(stableMemoryArgs(args, { platform: 'win32', cliSource: '--disable-dynamic-vram', dynamicVram: 'false' }), args)
  assert.deepEqual(stableMemoryArgs(args, { platform: 'win32', cliSource: '' }), args)
  assert.deepEqual(stableMemoryArgs(args, { platform: 'linux', cliSource: '--disable-dynamic-vram' }), args)
  assert.deepEqual(stableMemoryArgs(args, { platform: 'win32', cliSource: '--disable-dynamic-vram', dynamicVram: 'true' }), args)
  const defaults = ['main.py', '--enable-manager']
  assert.deepEqual(stableMemoryArgs(defaults), defaults)
  assert.notEqual(stableMemoryArgs(defaults), defaults)
  assert.deepEqual(stableMemoryArgs(['main.py', '--disable-dynamic-vram']), ['main.py', '--disable-dynamic-vram'])
})