Fetch the correct invite code.

Author: samsymons

DuckDuckGo/AppDelegate/AppDelegate.swift

@@ -221,7 +221,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, FileDownloadManagerDel
         syncService?.initializeIfNeeded(isInternalUser: internalUserDecider?.isInternalUser ?? false)
         syncService?.scheduler.notifyAppLifecycleEvent()
 
-        NetworkProtectionWaitlist.shared.fetchInviteCodeIfAvailable { _ in
+        NetworkProtectionWaitlist.shared.fetchNetworkProtectionInviteCodeIfAvailable { _ in
             // Do nothing when code fetching fails, as the app will try again later
         }
     }

DuckDuckGo/Menus/MainMenuActions.swift

@@ -742,10 +742,12 @@ extension MainViewController {
     @IBAction func resetNetworkProtectionWaitlistState(_ sender: Any?) {
         NetworkProtectionWaitlist.shared.waitlistStorage.deleteWaitlistState()
         UserDefaults().removeObject(forKey: UserDefaultsWrapper<Bool>.Key.networkProtectionTermsAndConditionsAccepted.rawValue)
+        NotificationCenter.default.post(name: .networkProtectionWaitlistAccessChanged, object: nil)
     }
 
     @IBAction func resetNetworkProtectionTermsAndConditionsAcceptance(_ sender: Any?) {
         UserDefaults().removeObject(forKey: UserDefaultsWrapper<Bool>.Key.networkProtectionTermsAndConditionsAccepted.rawValue)
+        NotificationCenter.default.post(name: .networkProtectionWaitlistAccessChanged, object: nil)
     }
 
     @IBAction func showNetworkProtectionInviteCodePrompt(_ sender: Any?) {

DuckDuckGo/Waitlist/Waitlist.swift

@@ -203,6 +203,8 @@ struct NetworkProtectionWaitlist: Waitlist {
                         assertionFailure("Failed to redeem invite code")
                     }
                 }
+            } else {
+                assertionFailure("Didn't get error or invite code")
             }
         }
     }