fixed race condition while upgrading containers and site assignments.

src/js/background/assignManager.js

@@ -112,8 +112,6 @@ const assignManager = {
       return sites;
     },
 
-    // TODOkmw: Change site assignment UUID
-
     /*
      * Looks for abandoned site assignments. If there is no identity with 
      * the site assignment's userContextId (cookieStoreId), then the assignment
@@ -133,15 +131,15 @@ const assignManager = {
             await this.remove(configKey.replace(/^siteContainerMap@@_/, "https://"));
             continue;
           }
-          const updatedSideAssignment = macConfigs[configKey];
-          if (!macConfigs[configKey].identityMacAddonUUID) {
-            await this.set(
-              configKey.replace(/^siteContainerMap@@_/, "https://"),
-              updatedSideAssignment,
-              false,
-              false
-            );
-          }
+          const updatedSiteAssignment = macConfigs[configKey];
+          updatedSiteAssignment.identityMacAddonUUID = 
+            await identityState.lookupMACaddonUUID(match.cookieStoreId);
+          await this.set(
+            configKey,
+            updatedSiteAssignment,
+            false,
+            false
+          );
         }
       }
 
@@ -168,7 +166,7 @@ const assignManager = {
   // We return here so the confirm page can load the tab when exempted
   async _exemptTab(m) {
     const pageUrl = m.pageUrl;
-    this.storageArea.setExempted(pageUrl, m.tabId);
+    await this.storageArea.setExempted(pageUrl, m.tabId);
     return true;
   },
 

src/js/background/identityState.js

@@ -58,7 +58,7 @@ const identityState = {
           );
           if (cookieStoreId === "firefox-default") continue;
           if (!match) {
-            this.remove(cookieStoreId);
+            await this.remove(cookieStoreId);
             continue;
           }
           if (!macConfigs[configKey].macAddonUUID) {

src/js/background/sync.js

@@ -148,7 +148,7 @@ const sync = {
   },
 
   async errorHandledRunSync () {
-    sync.runSync().catch(async (error)=> { 
+    await sync.runSync().catch( async (error)=> { 
       console.error("Error from runSync", error);
       await sync.checkForListenersMaybeAdd();
     });