Add unit test for coupons menu on the hub menu

Author: itsmeichigo

WooCommerce/Classes/ViewRelated/Hub Menu/HubMenuViewModel.swift

@@ -86,7 +86,7 @@ final class HubMenuViewModel: ObservableObject {
                 self.menuElements.append(.coupons)
             }
         }
-        ServiceLocator.stores.dispatch(action)
+        stores.dispatch(action)
     }
 
     /// Present the `StorePickerViewController` using the `StorePickerCoordinator`, passing the navigation controller from the entry point.

WooCommerce/WooCommerceTests/ViewRelated/HubMenu/HubMenuViewModelTests.swift

@@ -1,6 +1,7 @@
 import XCTest
 
 @testable import WooCommerce
+@testable import Yosemite
 
 final class HubMenuViewModelTests: XCTestCase {
     private let sampleSiteID: Int64 = 606
@@ -22,8 +23,51 @@ final class HubMenuViewModelTests: XCTestCase {
 
         // When
         let viewModel = HubMenuViewModel(siteID: sampleSiteID, featureFlagService: featureFlagService)
+        viewModel.setupMenuElements()
 
         // Then
         XCTAssertEqual(viewModel.menuElements, [.woocommerceAdmin, .viewStore, .inbox, .reviews])
     }
+
+    func test_menuElements_include_coupons_when_couponManagement_is_enabled_in_app_settings() {
+        // Given
+        let stores = MockStoresManager(sessionManager: .makeForTesting())
+        let featureFlagService = MockFeatureFlagService(isInboxOn: true)
+
+        // When
+        stores.whenReceivingAction(ofType: AppSettingsAction.self) { action in
+            switch action {
+            case .loadCouponManagementFeatureSwitchState(let onCompletion):
+                onCompletion(.success(true))
+            default:
+                break
+            }
+        }
+        let viewModel = HubMenuViewModel(siteID: sampleSiteID, featureFlagService: featureFlagService, stores: stores)
+        viewModel.setupMenuElements()
+
+        // Then
+        XCTAssertEqual(viewModel.menuElements, [.woocommerceAdmin, .viewStore, .inbox, .coupons, .reviews])
+    }
+
+    func test_menuElements_does_not_include_coupons_when_couponManagement_is_not_enabled_in_app_settings() {
+        // Given
+        let stores = MockStoresManager(sessionManager: .makeForTesting())
+        let featureFlagService = MockFeatureFlagService(isInboxOn: false)
+
+        // When
+        stores.whenReceivingAction(ofType: AppSettingsAction.self) { action in
+            switch action {
+            case .loadCouponManagementFeatureSwitchState(let onCompletion):
+                onCompletion(.success(false))
+            default:
+                break
+            }
+        }
+        let viewModel = HubMenuViewModel(siteID: sampleSiteID, featureFlagService: featureFlagService, stores: stores)
+        viewModel.setupMenuElements()
+
+        // Then
+        XCTAssertEqual(viewModel.menuElements, [.woocommerceAdmin, .viewStore, .reviews])
+    }
 }