package com.peachpasswords.security

import java.io.File
import org.junit.Assert.assertTrue
import org.junit.Test

class SecureFlagCoverageTest {
    @Test fun secret_surfaces_keep_flag_secure_coverage() {
        val protectedScreens = listOf(
            "app/src/main/java/com/peachpasswords/ui/vault/EntryDetailScreen.kt",
            "app/src/main/java/com/peachpasswords/ui/vault/EntryEditScreen.kt",
            "app/src/main/java/com/peachpasswords/ui/vault/NotesModeScreen.kt",
            "app/src/main/java/com/peachpasswords/ui/vault/ExportScreen.kt",
            "app/src/main/java/com/peachpasswords/ui/vault/RecoveryWordsScreen.kt",
            "app/src/main/java/com/peachpasswords/ui/recovery/RecoveryWordEntryScreen.kt",
            "app/src/main/java/com/peachpasswords/ui/codex/CodexGenerateScreen.kt",
        )

        protectedScreens.forEach { relativePath ->
            val file = File(relativePath).takeIf { it.exists() } ?: File("../$relativePath")
            val source = file.readText()
            assertTrue(
                "$relativePath must protect screenshots while visible",
                source.contains("SecureFlagEffect()") || source.contains("WindowManager.LayoutParams.FLAG_SECURE"),
            )
        }
    }
}