OpenAppLock/SeveredUITests/RuleManagementUITests.swift

//
//  RuleManagementUITests.swift
//  SeveredUITests
//

import XCTest

/// Detail sheet, editing, disabling, deleting, and unblocking — seeded with
/// one actively-blocking rule ("Work Time") and one upcoming rule ("Sleep").
final class RuleManagementUITests: XCTestCase {
    override func setUpWithError() throws {
        continueAfterFailure = false
    }

    func testDetailShowsLiveStatusAndFacts() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "standard")

        app.buttons["ruleCard-Work Time"].waitToAppear().tap()

        XCTAssertEqual(app.staticTexts["detailRuleName"].waitToAppear().label, "Work Time")
        XCTAssertTrue(app.staticTexts["detailStatusLabel"].label.contains("left"))
        app.element("detailRow-During this time").waitToAppear()
        XCTAssertTrue(app.element("detailRow-On these days").exists)
        XCTAssertTrue(app.element("detailRow-Unblocks allowed").exists)
        app.buttons["editRuleButton"].waitToAppear()

        app.buttons["closeDetailButton"].tap()
        app.buttons["newRuleButton"].waitToAppear()
    }

    func testEditRuleTogglesHardModeOn() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "standard")

        app.buttons["ruleCard-Sleep"].waitToAppear().tap()
        app.buttons["editRuleButton"].waitToAppear().tap()

        app.switches["hardModeToggle"].waitToAppear().tap()
        app.buttons["doneButton"].tap()

        // Back on the detail view, unblocks are no longer allowed.
        let row = app.element("detailRow-Unblocks allowed").waitToAppear()
        XCTAssertTrue(row.label.contains("No"), "Expected 'Unblocks allowed: No', got: \(row.label)")
    }

    func testDisableRule() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "standard")

        app.buttons["ruleCard-Sleep"].waitToAppear().tap()
        app.buttons["editRuleButton"].waitToAppear().tap()

        // Disable lives in the ellipsis menu in the navigation bar.
        let actionsMenu = app.navigationBars.buttons["ruleActionsMenu"].waitToAppear()
        XCTAssertEqual(actionsMenu.label, "Rule Actions")
        actionsMenu.tap()
        app.buttons["Disable Rule"].waitToAppear().tap()

        // The detail caption now reports the rule as disabled.
        let status = app.staticTexts["detailStatusLabel"].waitToAppear()
        XCTAssertTrue(status.label.contains("Disabled"), "Got: \(status.label)")

        app.buttons["closeDetailButton"].tap()
        let cardStatus = app.staticTexts["ruleStatus-Sleep"].waitToAppear()
        XCTAssertEqual(cardStatus.label, "Disabled")
    }

    func testDeleteRuleRemovesCard() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "standard")

        app.buttons["ruleCard-Sleep"].waitToAppear().tap()
        app.buttons["editRuleButton"].waitToAppear().tap()

        // Delete lives in the ellipsis menu in the navigation bar.
        app.navigationBars.buttons["ruleActionsMenu"].waitToAppear().tap()
        app.buttons["Delete Rule"].waitToAppear().tap()

        app.buttons["newRuleButton"].waitToAppear()
        XCTAssertFalse(
            app.buttons["ruleCard-Sleep"].waitForExistence(timeout: 2),
            "Deleted rule's card should disappear"
        )
        XCTAssertTrue(app.buttons["ruleCard-Work Time"].exists)
    }

    func testUnblockActiveSoftRule() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "standard")

        // The active rule surfaces in Blocked Apps; unblocking pauses it.
        app.buttons["blockedTile-Work Time"].waitToAppear().tap()
        app.sheets.buttons["Unblock"].waitToAppear().tap()

        app.staticTexts["nothingBlockedLabel"].waitToAppear()
        XCTAssertEqual(app.staticTexts["ruleStatus-Work Time"].waitToAppear().label, "Paused")
    }
}

/// Hard block behavior — seeded with an actively-blocking Hard Mode rule.
final class HardModeUITests: XCTestCase {
    override func setUpWithError() throws {
        continueAfterFailure = false
    }

    func testHardLockedRuleCannotBeEdited() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "hard-mode-active")

        app.buttons["ruleCard-Locked In"].waitToAppear().tap()

        // The lock notice replaces Edit Rule entirely.
        app.element("hardModeLockedNotice").waitToAppear()
        XCTAssertFalse(app.buttons["editRuleButton"].exists)
        XCTAssertTrue(app.element("detailRow-Unblocks allowed").label.contains("No"))
    }

    func testHardLockedRuleCannotBeUnblocked() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "hard-mode-active")

        app.buttons["blockedTile-Locked In"].waitToAppear().tap()

        // No unblock dialog — just the refusal alert.
        let alert = app.alerts["Hard Mode is on"].waitToAppear()
        XCTAssertTrue(alert.staticTexts["This block can't be lifted until it ends."].exists)
        alert.buttons["OK"].tap()

        // Still blocked.
        XCTAssertTrue(app.buttons["blockedTile-Locked In"].exists)
        XCTAssertFalse(app.staticTexts["nothingBlockedLabel"].exists)
    }

    func testSoftRuleUnblockOfferedButHardRuleRefused() throws {
        let app = XCUIApplication.launchSevered(seedScenario: "standard")

        app.buttons["blockedTile-Work Time"].waitToAppear().tap()
        // Soft rule: the confirmation dialog appears instead of the refusal alert.
        app.sheets.buttons["Unblock"].waitToAppear()
        XCTAssertFalse(app.alerts["Hard Mode is on"].exists)
    }
}