From 7570461594a74be471e732f4e8ecaddfdc0cf251 Mon Sep 17 00:00:00 2001
From: alpavanoglu <alp.avanoglu@automattic.com>
Date: Fri, 7 Jul 2023 18:51:50 +0200
Subject: [PATCH] Apply change requests

---
 .../CompliancePopoverCoordinator.swift        |  2 +-
 .../CompliancePopoverViewModel.swift          |  2 +-
 .../CompliancePopoverViewModelTests.swift     | 34 +++++++------------
 3 files changed, 15 insertions(+), 23 deletions(-)

diff --git a/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverCoordinator.swift b/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverCoordinator.swift
index b15871acd7a3..f679555b3335 100644
--- a/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverCoordinator.swift
+++ b/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverCoordinator.swift
@@ -102,7 +102,7 @@ private extension CompliancePopoverCoordinator {
 extension UserDefaults {
     static let didShowCompliancePopupKey = "didShowCompliancePopup"
 
-    @objc var didShowCompliancePopup: Bool {
+    var didShowCompliancePopup: Bool {
         get {
             bool(forKey: Self.didShowCompliancePopupKey)
         } set {
diff --git a/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverViewModel.swift b/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverViewModel.swift
index 08b024e16435..af31b107e8de 100644
--- a/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverViewModel.swift
+++ b/WordPress/Classes/ViewRelated/EEUUSCompliance/CompliancePopoverViewModel.swift
@@ -25,7 +25,7 @@ final class CompliancePopoverViewModel: ObservableObject {
         appAnalytics?.setUserHasOptedOut(!isAnalyticsEnabled)
 
         let (accountID, restAPI) = contextManager.performQuery { context -> (NSNumber?, WordPressComRestApi?) in
-           let account = try? WPAccount.lookupDefaultWordPressComAccount(in: context)
+            let account = try? WPAccount.lookupDefaultWordPressComAccount(in: context)
             return (account?.userID, account?.wordPressComRestApi)
         }
 
diff --git a/WordPress/WordPressTest/EUUSCompliance/CompliancePopoverViewModelTests.swift b/WordPress/WordPressTest/EUUSCompliance/CompliancePopoverViewModelTests.swift
index b22ba697d3a0..c3cd2bff3ce2 100644
--- a/WordPress/WordPressTest/EUUSCompliance/CompliancePopoverViewModelTests.swift
+++ b/WordPress/WordPressTest/EUUSCompliance/CompliancePopoverViewModelTests.swift
@@ -20,19 +20,15 @@ final class CompliancePopoverViewModelTests: CoreDataTestCase {
         testDefaults?.removeObject(forKey: UserDefaults.didShowCompliancePopupKey)
     }
 
-    func testDidTapSettingsUpdatesDefaults() {
-        guard let defaults = try? XCTUnwrap(testDefaults) else {
-            return
-        }
+    func testDidTapSettingsUpdatesDefaults() throws {
+        let defaults = try XCTUnwrap(testDefaults)
         let sut = CompliancePopoverViewModel(defaults: defaults, contextManager: contextManager)
         sut.didTapSettings()
         XCTAssert(defaults.didShowCompliancePopup)
     }
 
-    func testDidTapSettingsInvokesCoordinatorNavigation() {
-        guard let defaults = try? XCTUnwrap(testDefaults) else {
-            return
-        }
+    func testDidTapSettingsInvokesCoordinatorNavigation() throws {
+        let defaults = try XCTUnwrap(testDefaults)
         let mockCoordinator = MockCompliancePopoverCoordinator()
         let sut = CompliancePopoverViewModel(defaults: defaults, contextManager: contextManager)
         sut.coordinator = mockCoordinator
@@ -40,14 +36,13 @@ final class CompliancePopoverViewModelTests: CoreDataTestCase {
         XCTAssertEqual(mockCoordinator.navigateToSettingsCallCount, 1)
     }
 
-    func testDidTapSaveInvokesDismissWhenAccountIDExists() {
-        guard let defaults = try? XCTUnwrap(testDefaults) else {
-            return
-        }
+    func testDidTapSaveInvokesDismissWhenAccountIDExists() throws {
+        let defaults = try XCTUnwrap(testDefaults)
         let mockCoordinator = MockCompliancePopoverCoordinator()
+        UserSettings.defaultDotComUUID = account().uuid
         let sut = CompliancePopoverViewModel(
             defaults: defaults,
-            contextManager: makeCoreDataStack()
+            contextManager: contextManager
         )
         sut.coordinator = mockCoordinator
         sut.didTapSave()
@@ -55,23 +50,20 @@ final class CompliancePopoverViewModelTests: CoreDataTestCase {
         XCTAssert(defaults.didShowCompliancePopup)
     }
 
-    private func makeCoreDataStack() -> ContextManager {
-        let contextManager = ContextManager.forTesting()
-        let account = AccountBuilder(contextManager)
+    private func account() -> WPAccount {
+        return AccountBuilder(contextManager)
             .with(id: 1229)
             .with(username: "foobar")
             .with(email: "foo@automattic.com")
             .with(authToken: "9384rj398t34j98")
             .build()
-        UserSettings.defaultDotComUUID = account.uuid
-        return contextManager
     }
 }
 
 private class MockCompliancePopoverCoordinator: CompliancePopoverCoordinatorProtocol {
-    var navigateToSettingsCallCount = 0
-    var presentIfNeededCallCount = 0
-    var dismissCallCount = 0
+    private(set) var navigateToSettingsCallCount = 0
+    private(set) var presentIfNeededCallCount = 0
+    private(set) var dismissCallCount = 0
 
     func presentIfNeeded() {
         presentIfNeededCallCount += 1