From 65c63009c9f63ea2bccf8af9cfbeb2cc776f8351 Mon Sep 17 00:00:00 2001
From: Mark Striemer <mstriemer@mozilla.com>
Date: Sat, 10 Apr 2021 00:30:41 +0000
Subject: [PATCH] Bug 1702330 - Fix
 browser_aboutNewTab_defaultBrowserNotification.js proton infobars r=mconley

Differential Revision: https://phabricator.services.mozilla.com/D111356
---
 .../browser_aboutNewTab_defaultBrowserNotification.js      | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/browser/base/content/test/about/browser_aboutNewTab_defaultBrowserNotification.js b/browser/base/content/test/about/browser_aboutNewTab_defaultBrowserNotification.js
index dbda75a21b7d..5cff68d9d73c 100644
--- a/browser/base/content/test/about/browser_aboutNewTab_defaultBrowserNotification.js
+++ b/browser/base/content/test/about/browser_aboutNewTab_defaultBrowserNotification.js
@@ -154,7 +154,9 @@ add_task(async function clicking_button_on_notification_calls_setAsDefault() {
       !shellService.isDefaultBrowser(),
       "should not be default prior to clicking button"
     );
-    let button = notification.querySelector(".notification-button");
+    let button = notification.buttonContainer.querySelector(
+      ".notification-button"
+    );
     button.click();
     ok(
       shellService.isDefaultBrowser(),
@@ -248,8 +250,7 @@ add_task(async function clicking_dismiss_disables_default_browser_checking() {
       "Notification should be default browser"
     );
 
-    let closeButton = notification.querySelector(".close-icon");
-    closeButton.click();
+    notification.closeButton.click();
     ok(
       !Services.prefs.getBoolPref("browser.shell.checkDefaultBrowser"),
       "checkDefaultBrowser bar pref should be false after dismissing notification"