add ui tests for sign up and select first factor

Author: harsh62

Tests/AuthenticatorHostApp/AuthenticatorHostApp/AuthenticatorHostApp.swift

@@ -17,6 +17,7 @@ struct AuthenticatorHostApp: App {
     private var hidesSignUpButton = false
     private var initialStep = AuthenticatorInitialStep.signIn
     private var authSignInNextStep = AuthSignInStep.done
+    private var passwordlessFlow: Bool = false
     private var shouldUsePickerForTestingSteps = true
 
     var body: some Scene {
@@ -25,7 +26,8 @@ struct AuthenticatorHostApp: App {
                 hidesSignUpButton: hidesSignUpButton,
                 initialStep: initialStep,
                 authSignInStep: authSignInNextStep,
-                shouldUsePickerForTestingSteps: shouldUsePickerForTestingSteps)
+                shouldUsePickerForTestingSteps: shouldUsePickerForTestingSteps,
+                passwordlessFlow: passwordlessFlow)
         }
     }
 
@@ -54,6 +56,8 @@ struct AuthenticatorHostApp: App {
             factory.setUserAtributes(userAttributes)
         case .authSignInStep(let authUITestNextStep):
             self.authSignInNextStep = getMockedNextStepResult(from: authUITestNextStep)
+        case .passwordlessFlow(let isPasswordlessFlow):
+            self.passwordlessFlow = isPasswordlessFlow
         }
     }
 

Tests/AuthenticatorHostApp/AuthenticatorHostApp/ContentView.swift

@@ -100,15 +100,23 @@ struct ContentView: View {
     private let hidesSignUpButton: Bool
     private let initialStep: AuthenticatorInitialStep
     private let shouldUsePickerForTestingSteps: Bool
+    private let signUpFields: [SignUpField]
 
     init(hidesSignUpButton: Bool,
          initialStep: AuthenticatorInitialStep,
          authSignInStep: AuthSignInStep,
-         shouldUsePickerForTestingSteps: Bool = false) {
+         shouldUsePickerForTestingSteps: Bool = false,
+         passwordlessFlow: Bool = false) {
         self.hidesSignUpButton = hidesSignUpButton
         self.initialStep = initialStep
         self.shouldUsePickerForTestingSteps = shouldUsePickerForTestingSteps
 
+        if passwordlessFlow {
+            self.signUpFields = []
+        } else {
+            self.signUpFields = Self.defaultSignUpFields
+        }
+        
         // Configure mocks for testing
         configureMocksForPasswordlessTesting()
 
@@ -203,7 +211,11 @@ struct ContentView: View {
 
         Authenticator(
             initialStep: initialStep,
-            authenticationFlow: .userChoice(preferredAuthFactor: .password(), passkeyPrompts: .init(afterSignUp: .always, afterSignIn: .always)) // Testing UserChoice with no preferred auth factor
+            authenticationFlow: .userChoice(
+                preferredAuthFactor: .password(),
+                passkeyPrompts: .init(
+                    afterSignUp: .always,
+                    afterSignIn: .always))
         ) { state in
             VStack {
                 Text("Hello, \(state.user.username)")
@@ -225,7 +237,7 @@ struct ContentView: View {
 
     }
 
-    private var signUpFields: [SignUpField] {
+    private static var defaultSignUpFields: [SignUpField] {
         return [
             .password(isRequired: true),
             .confirmPassword(isRequired: true)

Tests/AuthenticatorHostApp/AuthenticatorHostAppUITests/AuthenticatorBaseTestCase.swift

@@ -58,7 +58,7 @@ class AuthenticatorBaseTestCase: XCTestCase {
         app.launch()
     }
 
-    func launchAppAndLogin(with args: [ProcessArgument]) {
+    func launchAppAndLogin(with args: [ProcessArgument], shouldEnterPassword: Bool = true) {
 
         // Launch Application
         launchApp(with: args)
@@ -69,9 +69,11 @@ class AuthenticatorBaseTestCase: XCTestCase {
         app.textFields.firstMatch.tap()
         app.textFields.firstMatch.typeText("username")
 
-        // Enter some password
-        app.secureTextFields.firstMatch.tap()
-        app.secureTextFields.firstMatch.typeText("password")
+        if shouldEnterPassword {
+            // Enter some password
+            app.secureTextFields.firstMatch.tap()
+            app.secureTextFields.firstMatch.typeText("password")
+        }
 
         // Tap Sign in button
         app.buttons["Sign In"].firstMatch.tap()

Tests/AuthenticatorHostApp/AuthenticatorHostAppUITests/AuthenticatorUITestUtils.swift

@@ -15,6 +15,7 @@ enum ProcessArgument: Codable {
     case initialStep(AuthenticatorInitialStep)
     case authSignInStep(AuthUITestSignInStep)
     case userAttributes([UserAttribute])
+    case passwordlessFlow(Bool)
 }
 
 enum UserAttribute: String, Codable {

Tests/AuthenticatorHostApp/AuthenticatorHostAppUITests/TestCases/ContinueSignInWithFirstFactorSelectionTests.swift

@@ -0,0 +1,20 @@
+//
+// Copyright Amazon.com Inc. or its affiliates.
+// All Rights Reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+import XCTest
+
+final class ContinueSignInWithFirstFactorSelectionTests: AuthenticatorBaseTestCase {
+
+    func testContinueSignInWithFirstFactorSelection() throws {
+        launchAppAndLogin(with: [
+            .hidesSignUpButton(false),
+            .initialStep(.signIn),
+            .authSignInStep(.continueSignInWithFirstFactorSelection)
+        ])
+        assertSnapshot()
+    }
+}

Tests/AuthenticatorHostApp/AuthenticatorHostAppUITests/TestCases/SignUpViewTests.swift

@@ -15,4 +15,12 @@ final class SignUpViewTests: AuthenticatorBaseTestCase {
         ])
         assertSnapshot()
     }
+    
+    func testPasswordlessSignUpView() throws {
+        launchApp(with: [
+            .initialStep(.signUp),
+            .passwordlessFlow(true)
+        ])
+        assertSnapshot()
+    }
 }