code block 2 for UITest medium blog

RawRepresentableUITest.swift

public extension RawRepresentable {
    var element: XCUIElement {
        if query.count > 1 {
            fatalError("There are \(query.count) elements with identifier \(self.rawValue as? String ?? "") found!")
        }
        return query.firstMatch
    }
    var query: XCUIElementQuery {
        return XCUIApplication().descendants(matching: .any).matching(identifier: self.rawValue as? String ?? "")
    }
    var count: Int {
        return Int(query.count)
    }
    subscript(i: Int) -> XCUIElement {
        return query.allElementsBoundByIndex[i]
    }
    var exists: Bool {
        return element.exists
    }
    func tap() {
        element.tap()
    }
}