package com.peachpasswords.ui.codex import org.junit.Assert.assertEquals import org.junit.Test class CodexGenerateScreenTest { @Test fun codexGenerateGate_requiresProBeforeRecoveryWords() { assertEquals(CodexGenerateGate.PRO_REQUIRED, codexGenerateGate(isPro = false, recoveryWordsSet = false)) assertEquals(CodexGenerateGate.PRO_REQUIRED, codexGenerateGate(isPro = false, recoveryWordsSet = true)) assertEquals(CodexGenerateGate.RECOVERY_REQUIRED, codexGenerateGate(isPro = true, recoveryWordsSet = false)) assertEquals(CodexGenerateGate.READY, codexGenerateGate(isPro = true, recoveryWordsSet = true)) } }