import XCTest
@testable import MaestroDriverLib
final class PermissionButtonFinderTests: XCTestCase {
var sut: PermissionButtonFinder!
override func setUp() {
super.setUp()
sut = PermissionButtonFinder()
}
override func tearDown() {
sut = nil
super.tearDown()
}
// MARK: - findButtons Tests
func testFindButtons_withNoButtons_returnsEmptyArray() {
let hierarchy = AXElement(
elementType: 0, // Not a button
children: []
)
let buttons = sut.findButtons(in: hierarchy)
XCTAssertTrue(buttons.isEmpty)
}
func testFindButtons_withSingleButton_returnsButton() {
let button = AXElement(
label: "Allow",
elementType: ElementType.button,
children: nil
)
let buttons = sut.findButtons(in: button)
XCTAssertEqual(buttons.count, 1)
XCTAssertEqual(buttons.first?.label, "Allow")
}
func testFindButtons_withNestedButtons_returnsAllButtons() {
let allowButton = AXElement(
label: "Allow",
elementType: ElementType.button,
children: nil
)
let denyButton = AXElement(
label: "Don't Allow",
elementType: ElementType.button,
children: nil
)
let container = AXElement(
elementType: 0, // Container, not a button
children: [allowButton, denyButton]
)
let hierarchy = AXElement(
elementType: 0,
children: [container]
)
let buttons = sut.findButtons(in: hierarchy)
XCTAssertEqual(buttons.count, 2)
XCTAssertEqual(buttons[0].label, "Allow")
XCTAssertEqual(buttons[1].label, "Don't Allow")
}
func testFindButtons_withDeeplyNestedButtons_findsAllButtons() {
let deepButton = AXElement(
label: "Deep Button",
elementType: ElementType.button,
children: nil
)
let level2 = AXElement(
elementType: 0,
children: [deepButton]
)
let level1 = AXElement(
elementType: 0,
children: [level2]
)
let hierarchy = AXElement(
elementType: 0,
children: [level1]
)
let buttons = sut.findButtons(in: hierarchy)
XCTAssertEqual(buttons.count, 1)
XCTAssertEqual(buttons.first?.label, "Deep Button")
}
// MARK: - findButtonToTap Tests (Allow Permission)
func testFindButtonToTap_allowPermission_findsAllowButtonByLabel() {
let allowButton = AXElement(
frame: ["X": 100, "Y": 200, "Width": 80, "Height": 40],
label: "Allow",
elementType: ElementType.button,
children: nil
)
let denyButton = AXElement(
frame: ["X": 10, "Y": 200, "Width": 80, "Height": 40],
label: "Don't Allow",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [denyButton, allowButton])
let result = sut.findButtonToTap(for: .allow, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 100)
XCTAssertEqual(frame["Y"], 200)
} else {
XCTFail("Expected .found result")
}
}
func testFindButtonToTap_allowPermission_findsContinueButton() {
let continueButton = AXElement(
frame: ["X": 100, "Y": 200, "Width": 80, "Height": 40],
label: "Continue",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [continueButton])
let result = sut.findButtonToTap(for: .allow, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 100)
} else {
XCTFail("Expected .found result")
}
}
func testFindButtonToTap_allowPermission_fallsBackToSecondButton() {
let firstButton = AXElement(
frame: ["X": 10, "Y": 200, "Width": 80, "Height": 40],
label: "Nope",
elementType: ElementType.button,
children: nil
)
let secondButton = AXElement(
frame: ["X": 100, "Y": 200, "Width": 80, "Height": 40],
label: "OK",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [firstButton, secondButton])
let result = sut.findButtonToTap(for: .allow, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 100)
} else {
XCTFail("Expected .found result")
}
}
// MARK: - findButtonToTap Tests (Deny Permission)
func testFindButtonToTap_denyPermission_findsDontAllowButtonByLabel() {
let allowButton = AXElement(
frame: ["X": 100, "Y": 200, "Width": 80, "Height": 40],
label: "Allow",
elementType: ElementType.button,
children: nil
)
let denyButton = AXElement(
frame: ["X": 10, "Y": 200, "Width": 80, "Height": 40],
label: "Don't Allow",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [allowButton, denyButton])
let result = sut.findButtonToTap(for: .deny, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 10)
} else {
XCTFail("Expected .found result")
}
}
func testFindButtonToTap_denyPermission_findsCancelButton() {
let cancelButton = AXElement(
frame: ["X": 10, "Y": 200, "Width": 80, "Height": 40],
label: "Cancel",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [cancelButton])
let result = sut.findButtonToTap(for: .deny, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 10)
} else {
XCTFail("Expected .found result")
}
}
func testFindButtonToTap_denyPermission_fallsBackToFirstButton() {
let firstButton = AXElement(
frame: ["X": 10, "Y": 200, "Width": 80, "Height": 40],
label: "Nope",
elementType: ElementType.button,
children: nil
)
let secondButton = AXElement(
frame: ["X": 100, "Y": 200, "Width": 80, "Height": 40],
label: "OK",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [firstButton, secondButton])
let result = sut.findButtonToTap(for: .deny, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 10)
} else {
XCTFail("Expected .found result")
}
}
// MARK: - findButtonToTap Tests (No Action Required)
func testFindButtonToTap_unsetPermission_returnsNoActionRequired() {
let hierarchy = AXElement(children: [])
let result = sut.findButtonToTap(for: .unset, in: hierarchy)
XCTAssertEqual(result, .noActionRequired)
}
func testFindButtonToTap_unknownPermission_returnsNoActionRequired() {
let hierarchy = AXElement(children: [])
let result = sut.findButtonToTap(for: .unknown, in: hierarchy)
XCTAssertEqual(result, .noActionRequired)
}
// MARK: - findButtonToTap Tests (No Buttons Found)
func testFindButtonToTap_noButtonsInHierarchy_returnsNoButtonsFound() {
let textElement = AXElement(
elementType: 0,
children: nil
)
let hierarchy = AXElement(children: [textElement])
let result = sut.findButtonToTap(for: .allow, in: hierarchy)
XCTAssertEqual(result, .noButtonsFound)
}
// MARK: - Case Insensitivity Tests
func testFindButtonToTap_allowPermission_isCaseInsensitive() {
let allowButton = AXElement(
frame: ["X": 100, "Y": 200, "Width": 80, "Height": 40],
label: "ALLOW",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [allowButton])
let result = sut.findButtonToTap(for: .allow, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 100)
} else {
XCTFail("Expected .found result")
}
}
func testFindButtonToTap_denyPermission_isCaseInsensitive() {
let denyButton = AXElement(
frame: ["X": 10, "Y": 200, "Width": 80, "Height": 40],
label: "DON'T ALLOW",
elementType: ElementType.button,
children: nil
)
let hierarchy = AXElement(children: [denyButton])
let result = sut.findButtonToTap(for: .deny, in: hierarchy)
if case .found(let frame) = result {
XCTAssertEqual(frame["X"], 10)
} else {
XCTFail("Expected .found result")
}
}
}
