From e191255c472ef47dd39c8d70f7cf584fa947c019 Mon Sep 17 00:00:00 2001
From: groovecoder <luke.crouch@gmail.com>
Date: Thu, 22 Jun 2017 09:26:30 -0500
Subject: [PATCH] fix #608: re-"render" badge on window focus change

---
 webextension/background.js | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/webextension/background.js b/webextension/background.js
index d1935f7..6e19420 100644
--- a/webextension/background.js
+++ b/webextension/background.js
@@ -505,6 +505,11 @@ const messageHandler = {
 
     browser.windows.onFocusChanged.addListener((windowId) => {
       assignManager.removeContextMenu();
+      // browserAction loses background color in new windows ...
+      // https://bugzil.la/1314674
+      // https://github.com/mozilla/testpilot-containers/issues/608
+      // ... so re-call displayBrowserActionBadge on window changes
+      displayBrowserActionBadge();
       browser.tabs.query({active: true, windowId}).then((tabs) => {
         if (tabs && tabs[0]) {
           tabPageCounter.initTabCounter(tabs[0]);