Add UI tests for performance [SwiftUI] (#1246)

* Add UI tests

* Fix styling

* Add TODO

Author: jrcrespoh

performance/SwiftUIPerformanceQuickstart/PerformanceExample/PerformanceExample.xcodeproj/project.pbxproj

@@ -10,8 +10,8 @@
 		4102056426A631F5004B24B3 /* PerformanceExampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 4102055626A631F4004B24B3 /* PerformanceExampleApp.swift */; };
 		4102056626A631F5004B24B3 /* MainView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 4102055726A631F4004B24B3 /* MainView.swift */; };
 		4102056826A631F5004B24B3 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 4102055826A631F5004B24B3 /* Assets.xcassets */; };
-		4102057926A63210004B24B3 /* Tests_iOS.swift in Sources */ = {isa = PBXBuildFile; fileRef = 4102057826A63210004B24B3 /* Tests_iOS.swift */; };
-		414C55A726B87EFB00114089 /* Tests_tvOS.swift in Sources */ = {isa = PBXBuildFile; fileRef = 414C55A626B87EFB00114089 /* Tests_tvOS.swift */; };
+		419700ED26C30DA500F186C3 /* UITests.swift in Sources */ = {isa = PBXBuildFile; fileRef = 419700EC26C30DA500F186C3 /* UITests.swift */; };
+		419700EE26C30DA500F186C3 /* UITests.swift in Sources */ = {isa = PBXBuildFile; fileRef = 419700EC26C30DA500F186C3 /* UITests.swift */; };
 		419AADCE26B86A94005C4817 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 419AADCD26B86A94005C4817 /* GoogleService-Info.plist */; };
 		419AADCF26B86A94005C4817 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 419AADCD26B86A94005C4817 /* GoogleService-Info.plist */; };
 		41BFA1EA26B1D36C0013EB90 /* PerformanceExampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 4102055626A631F4004B24B3 /* PerformanceExampleApp.swift */; };
@@ -44,11 +44,10 @@
 		4102055826A631F5004B24B3 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		4102055D26A631F5004B24B3 /* PerformanceExample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = PerformanceExample.app; sourceTree = BUILT_PRODUCTS_DIR; };
 		4102057626A63210004B24B3 /* Tests_iOS.xctest */ = {isa = PBXFileReference; explicitFileType = wrapper.cfbundle; includeInIndex = 0; path = Tests_iOS.xctest; sourceTree = BUILT_PRODUCTS_DIR; };
-		4102057826A63210004B24B3 /* Tests_iOS.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Tests_iOS.swift; sourceTree = "<group>"; };
 		410FF76926B1CEB500CA084F /* PerformanceExample (tvOS).app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = "PerformanceExample (tvOS).app"; sourceTree = BUILT_PRODUCTS_DIR; };
 		413CB73826B4AE8200E995E2 /* PerformanceExample--tvOS--Info.plist */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text.plist.xml; path = "PerformanceExample--tvOS--Info.plist"; sourceTree = "<group>"; };
 		414C55A426B87EFB00114089 /* Tests_tvOS.xctest */ = {isa = PBXFileReference; explicitFileType = wrapper.cfbundle; includeInIndex = 0; path = Tests_tvOS.xctest; sourceTree = BUILT_PRODUCTS_DIR; };
-		414C55A626B87EFB00114089 /* Tests_tvOS.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Tests_tvOS.swift; sourceTree = "<group>"; };
+		419700EC26C30DA500F186C3 /* UITests.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = UITests.swift; sourceTree = "<group>"; };
 		419AADCD26B86A94005C4817 /* GoogleService-Info.plist */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text.plist.xml; name = "GoogleService-Info.plist"; path = "../../GoogleService-Info.plist"; sourceTree = "<group>"; };
 		41BFA1EF26B1D8B80013EB90 /* Process.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Process.swift; sourceTree = "<group>"; };
 /* End PBXFileReference section */
@@ -91,8 +90,6 @@
 				419AADCD26B86A94005C4817 /* GoogleService-Info.plist */,
 				413CB73826B4AE8200E995E2 /* PerformanceExample--tvOS--Info.plist */,
 				4102055526A631F4004B24B3 /* Shared */,
-				4102057726A63210004B24B3 /* Tests_iOS */,
-				414C55A526B87EFB00114089 /* Tests_tvOS */,
 				4102055E26A631F5004B24B3 /* Products */,
 			);
 			sourceTree = "<group>";
@@ -103,6 +100,7 @@
 				4102055626A631F4004B24B3 /* PerformanceExampleApp.swift */,
 				4102055726A631F4004B24B3 /* MainView.swift */,
 				41BFA1EF26B1D8B80013EB90 /* Process.swift */,
+				419700EC26C30DA500F186C3 /* UITests.swift */,
 				4102055826A631F5004B24B3 /* Assets.xcassets */,
 			);
 			path = Shared;
@@ -223,13 +221,15 @@
 					};
 					4102057526A63210004B24B3 = {
 						CreatedOnToolsVersion = 13.0;
+						LastSwiftMigration = 1300;
 						TestTargetID = 4102055C26A631F5004B24B3;
 					};
 					410FF76826B1CEB500CA084F = {
 						CreatedOnToolsVersion = 13.0;
 					};
 					414C55A326B87EFB00114089 = {
 						CreatedOnToolsVersion = 13.0;
+						LastSwiftMigration = 1300;
 						TestTargetID = 410FF76826B1CEB500CA084F;
 					};
 				};
@@ -305,7 +305,7 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				4102057926A63210004B24B3 /* Tests_iOS.swift in Sources */,
+				419700ED26C30DA500F186C3 /* UITests.swift in Sources */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
@@ -323,7 +323,7 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				414C55A726B87EFB00114089 /* Tests_tvOS.swift in Sources */,
+				419700EE26C30DA500F186C3 /* UITests.swift in Sources */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
@@ -516,6 +516,7 @@
 		4102057D26A63210004B24B3 /* Debug */ = {
 			isa = XCBuildConfiguration;
 			buildSettings = {
+				CLANG_ENABLE_MODULES = YES;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				GENERATE_INFOPLIST_FILE = YES;
@@ -530,6 +531,7 @@
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				SDKROOT = iphoneos;
 				SWIFT_EMIT_LOC_STRINGS = NO;
+				SWIFT_OPTIMIZATION_LEVEL = "-Onone";
 				SWIFT_VERSION = 5.0;
 				TARGETED_DEVICE_FAMILY = "1,2";
 				TEST_TARGET_NAME = "PerformanceExample (iOS)";
@@ -539,6 +541,7 @@
 		4102057E26A63210004B24B3 /* Release */ = {
 			isa = XCBuildConfiguration;
 			buildSettings = {
+				CLANG_ENABLE_MODULES = YES;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				GENERATE_INFOPLIST_FILE = YES;
@@ -629,6 +632,7 @@
 			isa = XCBuildConfiguration;
 			buildSettings = {
 				CLANG_CXX_LANGUAGE_STANDARD = "gnu++17";
+				CLANG_ENABLE_MODULES = YES;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				GENERATE_INFOPLIST_FILE = YES;
@@ -642,6 +646,7 @@
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				SDKROOT = appletvos;
 				SWIFT_EMIT_LOC_STRINGS = NO;
+				SWIFT_OPTIMIZATION_LEVEL = "-Onone";
 				SWIFT_VERSION = 5.0;
 				TARGETED_DEVICE_FAMILY = 3;
 				TEST_TARGET_NAME = "PerformanceExample (tvOS)";
@@ -653,6 +658,7 @@
 			isa = XCBuildConfiguration;
 			buildSettings = {
 				CLANG_CXX_LANGUAGE_STANDARD = "gnu++17";
+				CLANG_ENABLE_MODULES = YES;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				GENERATE_INFOPLIST_FILE = YES;

performance/SwiftUIPerformanceQuickstart/PerformanceExample/Shared/Process.swift

@@ -23,7 +23,7 @@ class Process: ObservableObject {
   @Published var action: ProcessAction?
   @Published var image: UIImage?
   var categories: [(category: String, confidence: VNConfidence)]?
-  let precision: Float = 0.4
+  let precision: Float = 0.2
   let site = "https://firebase.google.com/downloads/brand-guidelines/PNG/logo-logomark.png"
 
   #if swift(>=5.5)
@@ -217,5 +217,6 @@ enum ProcessStatus {
 enum ProcessAction: String, CaseIterable {
   case download = "Download"
   case classify = "Classify"
-  case upload = "Upload"
+  // TODO: implement upload functionality
+//  case upload = "Upload"
 }

performance/SwiftUIPerformanceQuickstart/PerformanceExample/Shared/UITests.swift

@@ -0,0 +1,150 @@
+//
+//  Copyright 2021 Google LLC
+//
+//  Licensed under the Apache License, Version 2.0 (the "License");
+//  you may not use this file except in compliance with the License.
+//  You may obtain a copy of the License at
+//
+//  http://www.apache.org/licenses/LICENSE-2.0
+//
+//  Unless required by applicable law or agreed to in writing, software
+//  distributed under the License is distributed on an "AS IS" BASIS,
+//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//  See the License for the specific language governing permissions and
+//  limitations under the License.
+//
+
+import XCTest
+
+class UITests: XCTestCase {
+  var app: XCUIApplication = XCUIApplication()
+
+  override func setUpWithError() throws {
+    #if !os(iOS) && !os(tvOS)
+      throw Error("Unsupported platform.")
+    #endif
+
+    continueAfterFailure = false
+
+    app.launch()
+  }
+
+  override func tearDownWithError() throws {
+    app.terminate()
+  }
+
+  func checkButtons(_ functions: [String], timeout: TimeInterval = 1) throws {
+    for function in functions {
+      let button = app.buttons["\(function) Image"]
+      XCTAssert(button.waitForExistence(timeout: timeout), "Missing \(function) button.")
+      XCTAssert(button.isEnabled, "\(function) button not enabled.")
+      XCTAssert(button.isHittable, "\(function) button not in view.")
+    }
+  }
+
+  func checkText(_ text: String, timeout: TimeInterval = 1) throws {
+    let element = app.staticTexts[text]
+    XCTAssert(element.waitForExistence(timeout: timeout), "Missing text '\(text)'.")
+    XCTAssert(element.isHittable, "Text '\(text)' not in view.")
+  }
+
+  func checkStatus(_ status: String, timeout: TimeInterval = 1) throws {
+    #if os(iOS)
+      try checkText(status, timeout: timeout)
+    #endif
+  }
+
+  func goBack(timeout: TimeInterval = 1) throws {
+    #if os(iOS)
+      XCTAssert(app.buttons["Performance"].isHittable, "Reached invalid state.")
+      app.buttons["Performance"].tap()
+    #else
+      XCUIRemote.shared.press(.menu)
+      XCTAssert(app.navigationBars["Performance"].waitForExistence(timeout: timeout), "Timeout.")
+    #endif
+  }
+
+  func classify(image: Bool = false) throws {
+    #if os(iOS)
+      XCTAssert(app.buttons["Classify Image"].isHittable, "Reached invalid state.")
+      app.buttons["Classify Image"].tap()
+    #else
+      if !image {
+        XCUIRemote.shared.press(.down)
+        sleep(1)
+        XCTAssert(app.cells["Classify Image"].hasFocus, "Reached invalid state.")
+      }
+      XCUIRemote.shared.press(.select)
+    #endif
+  }
+
+  func download(timeout: TimeInterval = 10) throws {
+    #if os(iOS)
+      XCTAssert(app.buttons["Download Image"].isHittable, "Reached invalid state.")
+      app.buttons["Download Image"].tap()
+    #else
+      XCUIRemote.shared.press(.up)
+      sleep(1)
+      XCTAssert(app.cells["Download Image"].hasFocus, "Reached invalid state.")
+      XCUIRemote.shared.press(.select)
+    #endif
+
+    XCTAssert(app.images.firstMatch.waitForExistence(timeout: timeout), "Failed to retrieve image.")
+  }
+
+  func testAllViews() throws {
+    let buttons = ["Download", "Classify"]
+
+    XCTAssert(app.navigationBars["Performance"].exists, "Missing navigation title.")
+    XCTAssert(app.navigationBars["Performance"].isHittable, "Navigation title not in view.")
+
+    // Classify - Empty
+    try checkStatus("⏸ Idle")
+    try checkButtons(buttons)
+    try classify()
+    try checkText("No image found!\nPlease download an image first.")
+    try checkStatus("⏸ Idle")
+    try goBack()
+
+    // Download - Image
+    try checkButtons(buttons)
+    try checkStatus("⏸ Idle")
+    try download()
+    try checkStatus("✅ Success")
+    try goBack()
+
+    // Classify - Image
+    try checkStatus("✅ Success")
+    try checkButtons(buttons)
+    try classify()
+    try checkButtons(["Classify"])
+    try classify(image: true)
+    try checkText("Categories found:", timeout: 5)
+    try checkStatus("✅ Success")
+    try goBack()
+
+    // Download - Done
+    try checkStatus("✅ Success")
+    try checkButtons(buttons)
+    try download()
+    try checkStatus("✅ Success")
+    try goBack()
+
+    // Classify - Done
+    try checkStatus("✅ Success")
+    try checkButtons(buttons)
+    try classify()
+    try checkText("Categories found:")
+    try checkStatus("✅ Success")
+    try goBack()
+  }
+
+  func testLaunchPerformance() throws {
+    if #available(macOS 10.15, iOS 13.0, tvOS 13.0, watchOS 7.0, *) {
+      // This measures how long it takes to launch your application.
+      measure(metrics: [XCTApplicationLaunchMetric()]) {
+        XCUIApplication().launch()
+      }
+    }
+  }
+}