66 lines
1.8 KiB
Swift
66 lines
1.8 KiB
Swift
//
|
|
// XCUIElement+Extensions.swift
|
|
// Passepartout
|
|
//
|
|
// Created by Davide De Rosa on 11/28/24.
|
|
// Copyright (c) 2024 Davide De Rosa. All rights reserved.
|
|
//
|
|
// https://github.com/passepartoutvpn
|
|
//
|
|
// This file is part of Passepartout.
|
|
//
|
|
// Passepartout is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// Passepartout is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Passepartout. If not, see <http://www.gnu.org/licenses/>.
|
|
//
|
|
|
|
import Foundation
|
|
import UITesting
|
|
import XCTest
|
|
|
|
extension XCUIElement {
|
|
|
|
@discardableResult
|
|
func get(_ info: AccessibilityInfo, at index: Int = 0, timeout: TimeInterval = 1.0) -> XCUIElement {
|
|
let element = query(for: info.elementType)
|
|
.matching(identifier: info.id)
|
|
.element(boundBy: index)
|
|
|
|
XCTAssertTrue(element.waitForExistence(timeout: timeout))
|
|
return element
|
|
}
|
|
}
|
|
|
|
private extension XCUIElement {
|
|
func query(for elementType: AccessibilityInfo.ElementType) -> XCUIElementQuery {
|
|
#if os(iOS)
|
|
switch elementType {
|
|
case .button, .menu, .menuItem:
|
|
return buttons
|
|
case .text:
|
|
return staticTexts
|
|
}
|
|
#else
|
|
switch elementType {
|
|
case .button:
|
|
return buttons
|
|
case .menu:
|
|
return menuButtons
|
|
case .menuItem:
|
|
return menuItems
|
|
case .text:
|
|
return staticTexts
|
|
}
|
|
#endif
|
|
}
|
|
}
|