Update element queries for navigation bar buttons

carousell · Sep 17, 2017 · d0f5622 · d0f5622
1 parent 58e4a55
commit d0f5622
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Example/UITests/UITests.swift b/Example/UITests/UITests.swift
@@ -15,11 +15,11 @@ class UITests: XCTestCase {
     private lazy var app: XCUIApplication = XCUIApplication()
 
     private var cancelButton: XCUIElement {
-        return app.navigationBars["Camera Roll"].buttons["Cancel"]
+        return app.navigationBars.buttons["Cancel"]
     }
 
     private var doneButton: XCUIElement {
-        return app.navigationBars["Camera Roll"].buttons["Done"]
+        return app.navigationBars.buttons["Done"]
     }
 
     // MARK: -
@@ -101,7 +101,7 @@ class UITests: XCTestCase {
     func testSwitchingAlbums() {
         showImagePicker(named: "Default appearance")
 
-        app.navigationBars["Camera Roll"].staticTexts["Camera Roll"].tap()
+        app.navigationBars.staticTexts["Camera Roll"].tap()
         app.tables.cells.staticTexts["Favorites"].tap()
         XCTAssert(app.collectionViews.cells.count == 0)