Fix initialization for a few more unit_tests under PlzNavigate

chrome/browser/extensions/extension_context_menu_model_unittest.cc

Index: chrome/browser/extensions/extension_context_menu_model_unittest.cc
diff --git a/chrome/browser/extensions/extension_context_menu_model_unittest.cc b/chrome/browser/extensions/extension_context_menu_model_unittest.cc
index 721a1967a489bcd144ae67ab8d90a800c0ec09e0..0fff54e42261f6fe87e73297cf795039730e97dd 100644
--- a/chrome/browser/extensions/extension_context_menu_model_unittest.cc
+++ b/chrome/browser/extensions/extension_context_menu_model_unittest.cc
@@ -26,6 +26,8 @@
 #include "chrome/test/base/test_browser_window.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/crx_file/id_util.h"
+#include "content/public/common/browser_side_navigation_policy.h"
+#include "content/public/test/browser_side_navigation_test_utils.h"
 #include "content/public/test/web_contents_tester.h"
 #include "extensions/browser/extension_dialog_auto_confirm.h"
 #include "extensions/browser/extension_registry.h"
@@ -133,6 +135,9 @@ class ExtensionContextMenuModelTest : public ExtensionServiceTestBase {
 
   Browser* GetBrowser();
 
+  void SetUp() override;
+  void TearDown() override;
+
  private:
   std::unique_ptr<TestBrowserWindow> test_window_;
   std::unique_ptr<Browser> browser_;
@@ -185,6 +190,18 @@ Browser* ExtensionContextMenuModelTest::GetBrowser() {
   return browser_.get();
 }
 
+void ExtensionContextMenuModelTest::SetUp() {
+  ExtensionServiceTestBase::SetUp();
+  if (content::IsBrowserSideNavigationEnabled())
+    content::BrowserSideNavigationSetUp();
+}
+
+void ExtensionContextMenuModelTest::TearDown() {
+  if (content::IsBrowserSideNavigationEnabled())
+    content::BrowserSideNavigationTearDown();
+  ExtensionServiceTestBase::TearDown();
+}
+
 // Tests that applicable menu items are disabled when a ManagementPolicy
 // prohibits them.
 TEST_F(ExtensionContextMenuModelTest, RequiredInstallationsDisablesItems) {