XCTestCaseExtension.swift

import XCTest

extension XCTestCase {
    
    func waitForKeyboard() {
        expectationForPredicate(Predicate.countEqual1, evaluatedWithObject: XCUIApplication().keyboards, handler: nil)
        waitForExpectationsWithTimeout(10, handler: nil)
    }
    
    func waitForElementExist(element: XCUIElement, timeout: NSTimeInterval = 10) {
        expectationForPredicate(Predicate.exists, evaluatedWithObject: element, handler: nil)
        waitForExpectationsWithTimeout(timeout, handler: nil)
    }
    
    func waitForElementNotExist(element: XCUIElement, timeout: NSTimeInterval = 10) {
        expectationForPredicate(Predicate.notExists, evaluatedWithObject: element, handler: nil)
        waitForExpectationsWithTimeout(timeout, handler: nil)
    }
    
}

struct Predicate {
    
    static let countEqual1 = NSPredicate(format: "count == 1")
    static let exists = NSPredicate(format: "exists == 1")
    static let notExists = NSPredicate(format: "exists == 0")
    
}