honeyDueKMP/iosApp/HoneyDueUITests/Tests/MultiUserSharingUITests.swift

import XCTest

/// XCUITests for multi-user residence sharing.
///
/// Pattern: User A's data is seeded via API before app launch.
/// The app launches logged in as User B (via AuthenticatedUITestCase with UI-driven login).
/// User B joins User A's residence through the UI and verifies shared data.
///
/// ALL assertions check UI elements only. If the UI doesn't show the expected
/// data, that indicates a real app bug and the test should fail.
final class MultiUserSharingUITests: AuthenticatedUITestCase {

    /// User A's session (API-only, set up before app launch)
    private var userASession: TestSession!
    /// User B's session (fresh account, logged in via UI)
    private var userBSession: TestSession!
    /// The shared residence ID
    private var sharedResidenceId: Int!
    /// The share code User B will enter in the UI
    private var shareCode: String!
    /// The residence name (to verify in UI)
    private var sharedResidenceName: String!
    /// Titles of tasks/documents seeded by User A (to verify in UI)
    private var userATaskTitle: String!
    private var userADocTitle: String!

    /// Stored credentials for User B, set before super.setUpWithError() calls loginToMainApp()
    private var _userBUsername: String = ""
    private var _userBPassword: String = ""

    /// Dynamic credentials — returns User B's freshly created account
    override var testCredentials: (username: String, password: String) {
        (_userBUsername, _userBPassword)
    }

    override func setUpWithError() throws {
        guard TestAccountAPIClient.isBackendReachable() else {
            throw XCTSkip("Local backend not reachable")
        }

        // ── Create User A via API ──
        let runId = UUID().uuidString.prefix(6)
        guard let a = TestAccountAPIClient.createVerifiedAccount(
            username: "owner_\(runId)",
            email: "owner_\(runId)@test.com",
            password: "TestPass123!"
        ) else {
            XCTFail("Could not create User A (owner)"); return
        }
        userASession = a

        // ── User A creates a residence ──
        sharedResidenceName = "Shared House \(runId)"
        guard let residence = TestAccountAPIClient.createResidence(
            token: userASession.token,
            name: sharedResidenceName
        ) else {
            XCTFail("Could not create residence for User A"); return
        }
        sharedResidenceId = residence.id

        // ── User A generates a share code ──
        guard let code = TestAccountAPIClient.generateShareCode(
            token: userASession.token,
            residenceId: sharedResidenceId
        ) else {
            XCTFail("Could not generate share code"); return
        }
        shareCode = code.code

        // ── User A seeds data on the residence ──
        userATaskTitle = "Fix Roof \(runId)"
        _ = TestAccountAPIClient.createTask(
            token: userASession.token,
            residenceId: sharedResidenceId,
            title: userATaskTitle
        )

        userADocTitle = "Home Warranty \(runId)"
        _ = TestAccountAPIClient.createDocument(
            token: userASession.token,
            residenceId: sharedResidenceId,
            title: userADocTitle,
            documentType: "warranty"
        )

        // ── Create User B via API (fresh account) ──
        guard let b = TestAccountManager.createVerifiedAccount() else {
            XCTFail("Could not create User B (fresh account)"); return
        }
        userBSession = b

        // Set User B's credentials BEFORE super.setUpWithError() calls loginToMainApp()
        _userBUsername = b.username
        _userBPassword = b.password

        // ── Now launch the app and login as User B via base class ──
        try super.setUpWithError()
    }

    override func tearDownWithError() throws {
        // Clean up User A's data
        if let id = sharedResidenceId, let token = userASession?.token {
            _ = TestAccountAPIClient.deleteResidence(token: token, id: id)
        }
        try super.tearDownWithError()
    }

    // MARK: - Test 01: Join Residence via UI Share Code

    func test01_joinResidenceWithShareCode() {
        navigateToResidences()

        // Tap the join button (person.badge.plus icon in toolbar)
        let joinButton = findJoinButton()
        XCTAssertTrue(joinButton.waitForExistence(timeout: defaultTimeout), "Join button should exist")
        joinButton.tap()

        // Verify JoinResidenceView appeared
        let codeField = app.textFields["JoinResidence.ShareCodeField"]
        XCTAssertTrue(codeField.waitForExistence(timeout: defaultTimeout),
                       "Share code field should appear")

        // Type the share code
        codeField.tap()
        _ = app.keyboards.firstMatch.waitForExistence(timeout: 3)
        codeField.typeText(shareCode)

        // Tap Join
        let joinAction = app.buttons["JoinResidence.JoinButton"]
        XCTAssertTrue(joinAction.waitForExistence(timeout: defaultTimeout), "Join button should exist")
        XCTAssertTrue(joinAction.isEnabled, "Join button should be enabled with 6-char code")
        joinAction.tap()

        // Wait for join to complete — the sheet should dismiss

        // Verify the join screen dismissed (code field should be gone)
        let codeFieldGone = codeField.waitForNonExistence(timeout: 10)
        XCTAssertTrue(codeFieldGone || !codeField.exists,
                       "Join sheet should dismiss after successful join")

        // Verify the shared residence name appears in the Residences list
        let residenceText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", sharedResidenceName)
        ).firstMatch
        pullToRefreshUntilVisible(residenceText, maxRetries: 3)
        XCTAssertTrue(residenceText.exists,
                       "Shared residence '\(sharedResidenceName!)' should appear in Residences list after joining")
    }

    // MARK: - Test 02: Joined Residence Shows Data in UI

    func test02_joinedResidenceShowsSharedDocuments() {
        // Join via UI
        joinResidenceViaUI()

        // Verify residence appears in Residences tab
        let residenceText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", sharedResidenceName)
        ).firstMatch
        pullToRefreshUntilVisible(residenceText, maxRetries: 3)
        XCTAssertTrue(residenceText.exists,
                       "Shared residence '\(sharedResidenceName!)' should appear in Residences list")

        // Navigate to Documents tab and verify User A's document title appears
        navigateToDocuments()

        let docText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", userADocTitle)
        ).firstMatch
        pullToRefreshUntilVisible(docText, maxRetries: 3)
        XCTAssertTrue(docText.exists,
                       "User A's document '\(userADocTitle!)' should be visible in Documents tab after joining the shared residence")
    }

    // MARK: - Test 03: Shared Tasks Visible in UI

    /// Known issue: After joining a shared residence, the Tasks tab doesn't show
    /// the shared tasks. The AllTasksView's residenceViewModel uses cached (empty)
    /// data, which disables the refresh button and prevents task loading.
    /// Fix: AllTasksView.onAppear should detect residence list changes or use
    /// DataManager's already-refreshed cache.
    func test03_sharedTasksVisibleInTasksTab() {
        // Join via UI — this lands on Residences tab which triggers forceRefresh
        joinResidenceViaUI()

        // Verify the residence appeared (confirms join + refresh worked)
        let sharedRes = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", sharedResidenceName)
        ).firstMatch
        XCTAssertTrue(sharedRes.waitForExistence(timeout: defaultTimeout),
                       "Shared residence should be visible before navigating to Tasks")

        // Navigate to Tasks tab
        navigateToTasks()

        // Tap the refresh button (arrow.clockwise) to force-reload tasks
        let refreshButton = app.navigationBars.buttons.containing(
            NSPredicate(format: "label CONTAINS 'arrow.clockwise'")
        ).firstMatch
        for _ in 0..<5 {
            if refreshButton.waitForExistence(timeout: 3) && refreshButton.isEnabled {
                refreshButton.tap()
                // Wait for task data to load
                _ = app.staticTexts.containing(
                    NSPredicate(format: "label CONTAINS[c] %@", userATaskTitle)
                ).firstMatch.waitForExistence(timeout: defaultTimeout)
                break
            }
            // If disabled, wait for residence data to propagate
            _ = refreshButton.waitForExistence(timeout: 3)
        }

        // Search for User A's task title — it may be in any kanban column
        let taskText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", userATaskTitle)
        ).firstMatch

        // Kanban is a horizontal scroll — swipe left through columns to find the task
        for _ in 0..<5 {
            if taskText.exists { break }
            app.swipeLeft()
        }

        XCTAssertTrue(taskText.waitForExistence(timeout: defaultTimeout),
                       "User A's task '\(userATaskTitle!)' should be visible in Tasks tab after joining the shared residence")
    }

    // MARK: - Test 04: Shared Residence Shows in Documents Tab

    func test04_sharedResidenceShowsInDocumentsTab() {
        joinResidenceViaUI()
        navigateToDocuments()

        // Look for User A's document
        let docText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] 'Home Warranty'")
        ).firstMatch
        pullToRefreshUntilVisible(docText, maxRetries: 3)

        // Document may or may not show depending on filtering — verify the tab loaded
        let documentsTab = app.tabBars.buttons.containing(
            NSPredicate(format: "label CONTAINS[c] 'Doc'")
        ).firstMatch
        XCTAssertTrue(documentsTab.isSelected, "Documents tab should be selected")
    }

    // MARK: - Test 05: Cross-User Document Visibility in UI

    func test05_crossUserDocumentVisibleInUI() {
        // Join via UI
        joinResidenceViaUI()

        // Navigate to Documents tab
        navigateToDocuments()

        // Verify User A's seeded document appears
        let docText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", userADocTitle)
        ).firstMatch
        pullToRefreshUntilVisible(docText, maxRetries: 3)
        XCTAssertTrue(docText.exists,
                       "User A's document '\(userADocTitle!)' should be visible to User B in the Documents tab")
    }

    // MARK: - Test 06: Join Button Disabled With Short Code

    func test06_joinResidenceButtonDisabledWithShortCode() {
        navigateToResidences()

        let joinButton = findJoinButton()
        guard joinButton.waitForExistence(timeout: defaultTimeout) else {
            XCTFail("Join button should exist"); return
        }
        joinButton.tap()

        let codeField = app.textFields["JoinResidence.ShareCodeField"]
        guard codeField.waitForExistence(timeout: defaultTimeout) else {
            XCTFail("Share code field should appear"); return
        }

        // Type only 3 characters
        codeField.tap()
        _ = app.keyboards.firstMatch.waitForExistence(timeout: 3)
        codeField.typeText("ABC")

        let joinAction = app.buttons["JoinResidence.JoinButton"]
        XCTAssertTrue(joinAction.exists, "Join button should exist")
        XCTAssertFalse(joinAction.isEnabled, "Join button should be disabled with < 6 chars")

        // Dismiss
        let dismissButton = app.buttons.containing(
            NSPredicate(format: "label CONTAINS[c] 'Cancel' OR label CONTAINS[c] 'xmark'")
        ).firstMatch
        if dismissButton.exists { dismissButton.tap() }
    }

    // MARK: - Test 07: Invalid Code Shows Error

    func test07_joinWithInvalidCodeShowsError() {
        navigateToResidences()

        let joinButton = findJoinButton()
        guard joinButton.waitForExistence(timeout: defaultTimeout) else {
            XCTFail("Join button should exist"); return
        }
        joinButton.tap()

        let codeField = app.textFields["JoinResidence.ShareCodeField"]
        guard codeField.waitForExistence(timeout: defaultTimeout) else {
            XCTFail("Share code field should appear"); return
        }

        // Type an invalid 6-char code
        codeField.tap()
        _ = app.keyboards.firstMatch.waitForExistence(timeout: 3)
        codeField.typeText("ZZZZZZ")

        let joinAction = app.buttons["JoinResidence.JoinButton"]
        joinAction.tap()

        // Wait for API response - either error text appears or we stay on join screen
        let errorText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] 'error' OR label CONTAINS[c] 'invalid' OR label CONTAINS[c] 'not found' OR label CONTAINS[c] 'expired'")
        ).firstMatch
        _ = errorText.waitForExistence(timeout: defaultTimeout)

        // Should show an error message (code field should still be visible = still on join screen)
        let stillOnJoinScreen = codeField.exists

        XCTAssertTrue(errorText.exists || stillOnJoinScreen,
                       "Should show error or remain on join screen with invalid code")

        // Dismiss
        let dismissButton = app.buttons.containing(
            NSPredicate(format: "label CONTAINS[c] 'Cancel' OR label CONTAINS[c] 'xmark'")
        ).firstMatch
        if dismissButton.exists { dismissButton.tap() }
    }

    // MARK: - Test 08: Residence Detail Shows After Join

    func test08_residenceDetailAccessibleAfterJoin() {
        // Join via UI
        joinResidenceViaUI()

        // Find and tap the shared residence in the list
        let residenceText = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", sharedResidenceName)
        ).firstMatch
        pullToRefreshUntilVisible(residenceText, maxRetries: 3)
        XCTAssertTrue(residenceText.exists,
                       "Shared residence '\(sharedResidenceName!)' should appear in Residences list")
        residenceText.tap()

        // Verify the residence detail view loads and shows the residence name
        let detailTitle = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] %@", sharedResidenceName)
        ).firstMatch
        XCTAssertTrue(detailTitle.waitForExistence(timeout: defaultTimeout),
                       "Residence detail should display the residence name '\(sharedResidenceName!)'")

        // Look for indicators of multiple users (e.g. "2 users", "Members", user list)
        let multiUserIndicator = app.staticTexts.containing(
            NSPredicate(format: "label CONTAINS[c] '2 user' OR label CONTAINS[c] '2 member' OR label CONTAINS[c] 'Members' OR label CONTAINS[c] 'Manage Users' OR label CONTAINS[c] 'Users'")
        ).firstMatch

        // If a user count or members section is visible, verify it
        if multiUserIndicator.waitForExistence(timeout: 5) {
            XCTAssertTrue(multiUserIndicator.exists,
                           "Residence detail should show information about multiple users")
        }
        // If no explicit user indicator is visible (non-owner may not see Manage Users),
        // the test still passes because we verified the residence detail loaded successfully.
    }

    // MARK: - Helpers

    /// Find the join residence button in the toolbar
    private func findJoinButton() -> XCUIElement {
        // Look for the person.badge.plus button in the navigation bar
        let navButtons = app.navigationBars.buttons
        for i in 0..<navButtons.count {
            let button = navButtons.element(boundBy: i)
            if button.label.contains("person.badge.plus") || button.label.contains("Join") {
                return button
            }
        }
        // Fallback: any button with person.badge.plus
        return app.buttons.containing(
            NSPredicate(format: "label CONTAINS 'person.badge.plus'")
        ).firstMatch
    }

    /// Join the shared residence via the UI (type share code, tap join).
    /// After joining, verifies the join sheet dismissed and returns to the Residences list.
    private func joinResidenceViaUI() {
        navigateToResidences()

        let joinButton = findJoinButton()
        guard joinButton.waitForExistence(timeout: defaultTimeout) else {
            XCTFail("Join button not found"); return
        }
        joinButton.tap()

        let codeField = app.textFields["JoinResidence.ShareCodeField"]
        guard codeField.waitForExistence(timeout: defaultTimeout) else {
            XCTFail("Share code field not found"); return
        }
        codeField.tap()
        _ = app.keyboards.firstMatch.waitForExistence(timeout: 3)
        codeField.typeText(shareCode)

        let joinAction = app.buttons["JoinResidence.JoinButton"]
        guard joinAction.waitForExistence(timeout: defaultTimeout), joinAction.isEnabled else {
            XCTFail("Join button not enabled"); return
        }
        joinAction.tap()

        // After join, wait for the sheet to dismiss
        _ = codeField.waitForNonExistence(timeout: loginTimeout)

        // List should refresh
        pullToRefresh()
    }
}