Add a test case for StoreCreationCountryQuestionViewModel.

Author: jaclync

WooCommerce/WooCommerceTests/Authentication/Store Creation/Profiler/StoreCreationCountryQuestionViewModelTests.swift

@@ -8,15 +8,15 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
 
     func test_topHeader_is_set_to_store_name() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store 🌟") { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store 🌟") { _ in } onSupport: {}
 
         // Then
         XCTAssertEqual(viewModel.topHeader, "store 🌟")
     }
 
     func test_currentCountryCode_and_initial_selectedCountryCode_are_set_by_locale() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "fr_FR")) { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "fr_FR")) { _ in } onSupport: {}
 
         // Then
         XCTAssertEqual(viewModel.currentCountryCode, .FR)
@@ -25,7 +25,7 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
 
     func test_currentCountryCode_and_initial_selectedCountryCode_are_nil_with_an_invalid_locale() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "zzzz")) { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "zzzz")) { _ in } onSupport: {}
 
         // Then
         XCTAssertNil(viewModel.currentCountryCode)
@@ -34,15 +34,15 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
 
     func test_countryCodes_include_all_countries_with_an_invalid_locale() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "zzzz")) { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "zzzz")) { _ in } onSupport: {}
 
         // Then
         XCTAssertEqual(viewModel.countryCodes, SiteAddress.CountryCode.allCases.sorted(by: { $0.readableCountry < $1.readableCountry }))
     }
 
     func test_countryCodes_do_not_include_currentCountryCode_from_locale() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "fr_FR")) { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "fr_FR")) { _ in } onSupport: {}
 
         // Then
         XCTAssertFalse(viewModel.countryCodes.contains(.FR))
@@ -51,7 +51,7 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
 
     func test_selecting_a_country_updates_selectedCountryCode() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store") { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store") { _ in } onSupport: {}
 
         // When
         viewModel.selectCountry(.FJ)
@@ -62,7 +62,7 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
 
     func test_selecting_a_country_sets_isContinueButtonEnabled_to_true_with_an_invalid_locale() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "zzzz")) { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "zzzz")) { _ in } onSupport: {}
         var isContinueButtonEnabledValues = [Bool]()
         viewModel.isContinueButtonEnabled.removeDuplicates().sink { isEnabled in
             isContinueButtonEnabledValues.append(isEnabled)
@@ -79,7 +79,7 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
 
     func test_isContinueButtonEnabled_stays_true_with_a_valid_locale() throws {
         // Given
-        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "fr_FR")) { _ in }
+        let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store", currentLocale: .init(identifier: "fr_FR")) { _ in } onSupport: {}
         var isContinueButtonEnabledValues = [Bool]()
         viewModel.isContinueButtonEnabled.removeDuplicates().sink { isEnabled in
             isContinueButtonEnabledValues.append(isEnabled)
@@ -102,7 +102,7 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
                 // Then
                 XCTAssertEqual(countryCode, .JP)
                 promise(())
-            }
+            } onSupport: {}
 
             // When
             viewModel.selectCountry(.JP)
@@ -118,10 +118,24 @@ final class StoreCreationCountryQuestionViewModelTests: XCTestCase {
                                                               currentLocale: .init(identifier: "zzzz")) { countryCode in
             // Then
             XCTFail("Should not be invoked without selecting a country.")
-        }
+        } onSupport: {}
         // When
         Task { @MainActor in
             await viewModel.continueButtonTapped()
         }
     }
+
+    func test_supportButtonTapped_invokes_onSupport() throws {
+        waitFor { promise in
+            // Given
+            let viewModel = StoreCreationCountryQuestionViewModel(storeName: "store",
+                                                                  currentLocale: .init(identifier: "")) { _ in } onSupport: {
+                // Then
+                promise(())
+            }
+
+            // When
+            viewModel.supportButtonTapped()
+        }
+    }
 }