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',
    expiresAt: null,
    features: ['*'],
    loaded: true,
    wall: { open: false },
  })
}

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