Query.swift

//
//  Query.swift
//  CodeEditUITests
//
//  Created by Khan Winter on 7/10/24.
//

import XCTest

/// Query helpers for querying for specific UI elements. Organized by category in the app.
/// Queries should not evaluate if an element exists. This allows for tests to expect an element to exist,
/// perform an action, and then wait for that element to exist.
enum Query {
    static func getWindow(_ application: XCUIApplication, named: String? = nil) -> XCUIElement {
        if let named {
            return application.windows[named]
        } else {
            return application.windows.element(matching: .window, identifier: "workspace")
        }
    }

    static func getSettingsWindow(_ application: XCUIApplication) -> XCUIElement {
        return application.windows.element(matching: .window, identifier: "settings")
    }

    static func getWelcomeWindow(_ application: XCUIApplication) -> XCUIElement {
        return application.windows.element(matching: .window, identifier: "welcome")
    }

    static func getAboutWindow(_ application: XCUIApplication) -> XCUIElement {
        return application.windows.element(matching: .window, identifier: "about")
    }

    enum Window {
        static func getProjectNavigator(_ window: XCUIElement) -> XCUIElement {
            return window.descendants(matching: .any).matching(identifier: "ProjectNavigator").element
        }

        static func getTabBar(_ window: XCUIElement) -> XCUIElement {
            return window.descendants(matching: .any).matching(identifier: "TabBar").element
        }
    }

    enum Navigator {
        static func getRows(_ navigator: XCUIElement) -> XCUIElementQuery {
            navigator.descendants(matching: .outlineRow)
        }

        static func getSelectedRows(_ navigator: XCUIElement) -> XCUIElementQuery {
            getRows(navigator).matching(NSPredicate(format: "selected = true"))
        }

        static func getProjectNavigatorRow(fileTitle: String, index: Int = 0, _ navigator: XCUIElement) -> XCUIElement {
            return getRows(navigator)
                .containing(.textField, identifier: "ProjectNavigatorTableViewCell-\(fileTitle)")
                .element(boundBy: index)
        }

        static func disclosureIndicatorForRow(_ row: XCUIElement) -> XCUIElement {
            row.descendants(matching: .disclosureTriangle).element
        }

        static func rowContainsDisclosureIndicator(_ row: XCUIElement) -> Bool {
            disclosureIndicatorForRow(row).exists
        }
    }

    enum TabBar {
        static func getTab(labeled title: String, _ tabBar: XCUIElement) -> XCUIElement {
            tabBar.descendants(matching: .group).containing(NSPredicate(format: "value = %@", title)).firstMatch
        }
    }
}