Add an experiment for the default browser infobar on Windows 10

chrome/browser/ui/startup/default_browser_infobar_delegate_unittest.cc

Index: chrome/browser/ui/startup/default_browser_infobar_delegate_unittest.cc
diff --git a/chrome/browser/ui/startup/default_browser_infobar_delegate_unittest.cc b/chrome/browser/ui/startup/default_browser_infobar_delegate_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b6cd55e17d7492ce3526b7783ac14d3d1c58cb65
--- /dev/null
+++ b/chrome/browser/ui/startup/default_browser_infobar_delegate_unittest.cc
@@ -0,0 +1,147 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/ui/startup/default_browser_infobar_delegate.h"
+
+#include <memory>
+#include <string>
+
+#include "base/feature_list.h"
+#include "base/macros.h"
+#include "base/memory/ptr_util.h"
+#include "base/run_loop.h"
+#include "chrome/browser/infobars/infobar_service.h"
+#include "chrome/test/base/testing_profile.h"
+#include "components/infobars/core/infobar.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "content/public/test/test_web_contents_factory.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace chrome {
+
+// An implementation of DefaultBrowserWorker that will simply invoke the
+// 'on_finished_callback' immediately.
+class FakeDefaultBrowserWorker
+    : public shell_integration::DefaultBrowserWorker {
+ public:
+  FakeDefaultBrowserWorker(
+      const shell_integration::DefaultWebClientWorkerCallback& callback)
+      : shell_integration::DefaultBrowserWorker(callback) {}
+
+ private:
+  // Check if Chrome is the default browser.
+  shell_integration::DefaultWebClientState CheckIsDefaultImpl() override {
+    return shell_integration::NOT_DEFAULT;
+  }
+
+  // Set Chrome as the default browser.
+  void SetAsDefaultImpl(const base::Closure& on_finished_callback) override {
+    on_finished_callback.Run();
+  }
+
+  DISALLOW_COPY_AND_ASSIGN(FakeDefaultBrowserWorker);
+};
+
+class FakeDefaultBrowserInfoBarDelegate : public DefaultBrowserInfoBarDelegate {
+ public:
+  FakeDefaultBrowserInfoBarDelegate()
+      : DefaultBrowserInfoBarDelegate(nullptr) {}
+
+ private:
+  scoped_refptr<shell_integration::DefaultBrowserWorker>
+  CreateDefaultBrowserWorker(
+      const shell_integration::DefaultWebClientWorkerCallback& callback)
+      override {
+    return new FakeDefaultBrowserWorker(callback);
+  }
+
+  DISALLOW_COPY_AND_ASSIGN(FakeDefaultBrowserInfoBarDelegate);
+};
+
+class DefaultBrowserInfoBarDelegateTest : public ::testing::Test {
+ public:
+  DefaultBrowserInfoBarDelegateTest()
+      : profile_(new TestingProfile()),
+        web_contents_(factory_.CreateWebContents(profile_.get())) {
+    InfoBarService::CreateForWebContents(web_contents_);
+    infobar_service_ = InfoBarService::FromWebContents(web_contents_);

[Peter Kasting, 2016/06/01 19:32:10]

I think this class leaks |web_contents_| and |infobar_service_|.  If they are
safe to delete directly make them unique_ptrs, otherwise add a destructor that
cleans up.

[Patrick Monette, 2016/06/01 22:07:40]

The TestWebContentsFactory's documentation is pretty clear about keeping
ownership of the WebContents that it creates. It will get cleaned up in the
factory's destructor.

And since the InfoBarService derives WebContentsUserData, it is scoped to the
WebContents.

+  }
+
+ protected:
+  void EnableStickyDefaultBrowserPrompt() {
+    base::FeatureList::ClearInstanceForTesting();
+    std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);
+    feature_list->InitializeFromCommandLine(kStickyDefaultBrowserPrompt.name,
+                                            std::string());
+    base::FeatureList::SetInstance(std::move(feature_list));
+  }
+
+  void AddDefaultBrowserInfoBar() {
+    infobar_service_->AddInfoBar(infobar_service_->CreateConfirmInfoBar(
+        base::WrapUnique(new FakeDefaultBrowserInfoBarDelegate())));
+  }
+
+  InfoBarService* infobar_service() { return infobar_service_; }
+
+ private:
+  // The DefaultBrowserWorker requires a FILE thread. Also provides a
+  // SingleThreadTaskRunner for the test profile and the default browser prompt.
+  content::TestBrowserThreadBundle thread_bundle;
+
+  // Required to get an InfoBarService.
+  std::unique_ptr<TestingProfile> profile_;
+  content::TestWebContentsFactory factory_;
+  content::WebContents* web_contents_;
+
+  // Manages the default browser prompt.
+  InfoBarService* infobar_service_;

[Peter Kasting, 2016/06/01 19:32:10]

Heh, I was suggesting subclassing this, but actually using it directly without
subclassing it is better -- didn't think that'd work!

[Patrick Monette, 2016/06/01 22:07:40]

I was pleasantly surprised as well!

+
+  DISALLOW_COPY_AND_ASSIGN(DefaultBrowserInfoBarDelegateTest);
+};
+
+TEST_F(DefaultBrowserInfoBarDelegateTest, DefaultBehavior) {
+  AddDefaultBrowserInfoBar();
+  ASSERT_EQ(1U, infobar_service()->infobar_count());
+
+  infobars::InfoBar* default_browser_infobar = infobar_service()->infobar_at(0);
+  ConfirmInfoBarDelegate* default_browser_infobar_delegate =
+      default_browser_infobar->delegate()->AsConfirmInfoBarDelegate();
+
+  // The infobar delegate should allow itself to be destroyed immediately if the
+  // user activates the default action. Accept() should return true to indicate
+  // this.
+  EXPECT_TRUE(default_browser_infobar_delegate->Accept());
+}
+
+#if defined(OS_WIN)
+TEST_F(DefaultBrowserInfoBarDelegateTest, StickyDefaultBrowserPrompt) {
+  EnableStickyDefaultBrowserPrompt();
+
+  // For most Windows versions, this experiment is disabled.
+  if (!IsStickyDefaultBrowserPromptEnabled())
+    return;
+
+  AddDefaultBrowserInfoBar();
+  ASSERT_EQ(1U, infobar_service()->infobar_count());
+
+  infobars::InfoBar* default_browser_infobar = infobar_service()->infobar_at(0);
+  ConfirmInfoBarDelegate* default_browser_infobar_delegate =
+      default_browser_infobar->delegate()->AsConfirmInfoBarDelegate();
+
+  // The sticky default browser prompt experiment should mean activating the
+  // infobar's default action does not result in the infobar's destruction.
+  // Instead, the infobar will ultimately be destroyed when the
+  // DefaultBrowserWorker is done. To indicate this Accept() should return
+  // false.
+  EXPECT_FALSE(default_browser_infobar_delegate->Accept());
+  EXPECT_EQ(1U, infobar_service()->infobar_count());
+
+  // Spin the message loop to allow the FakeDefaultBrowserWorker to complete,
+  // which should destroy the infobar.
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(0U, infobar_service()->infobar_count());
+}
+#endif  // defined(OS_WIN)
+
+}  // namespace chrome