import { useEntitlementStore } from '../../src/stores/entitlementStore'

/** Enable Pro features for acceptance tests that exercise gated flows. */
export function unlockProForTests(): void {
  useEntitlementStore.setState({
    plan: 'pro',
    deviceId: 'test-device',
    email: '',
    licenseKey: 'LIC-TEST',
    expiresAt: null,
    maxDevices: 5,
    planType: 'perpetual',
    features: ['*'],
    loaded: true,
    wall: { open: false },
  })
}

export function resetEntitlementForTests(): void {
  useEntitlementStore.setState({
    plan: 'free',
    deviceId: 'test-device',
    email: '',
    licenseKey: '',
    expiresAt: null,
    maxDevices: null,
    planType: null,
    features: [],
    loaded: true,
    wall: { open: false },
  })
}