chrome/browser/browser_switcher/alternative_browser_launcher_unittest.cc

// Copyright 2018 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/browser_switcher/alternative_browser_launcher.h"

#include "base/values.h"
#include "chrome/browser/browser_switcher/browser_switcher_prefs.h"
#include "chrome/browser/browser_switcher/mock_alternative_browser_driver.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/scoped_user_pref_update.h"
#include "components/prefs/testing_pref_service.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

using ::testing::Invoke;
using ::testing::Return;
using ::testing::_;

namespace browser_switcher {

namespace {

const char kExampleDotCom[] = "http://example.com/";


}  // namespace

class AlternativeBrowserLauncherTest : public testing::Test {
 public:
  void SetUp() override {
    prefs_.registry()->RegisterStringPref(prefs::kAlternativeBrowserPath, "");
    prefs_.registry()->RegisterListPref(prefs::kAlternativeBrowserParameters);
    auto driver = std::make_unique<MockAlternativeBrowserDriver>();
    driver_ = driver.get();
    EXPECT_CALL(*driver_, SetBrowserPath(_))
        .WillOnce(Invoke([](base::StringPiece path) { EXPECT_EQ("", path); }));
    EXPECT_CALL(*driver_, SetBrowserParameters(_))
        .WillOnce(Invoke([](const base::ListValue* parameters) {
          EXPECT_TRUE(parameters->GetList().empty());
        }));
    launcher_ = std::make_unique<AlternativeBrowserLauncherImpl>(
        &prefs_, std::move(driver));
  }

  TestingPrefServiceSimple* prefs() { return &prefs_; }
  AlternativeBrowserLauncher* launcher() { return launcher_.get(); }
  MockAlternativeBrowserDriver& driver() { return *driver_; }

 private:
  TestingPrefServiceSimple prefs_;

  std::unique_ptr<AlternativeBrowserLauncher> launcher_;
  MockAlternativeBrowserDriver* driver_;
};

TEST_F(AlternativeBrowserLauncherTest, LaunchSucceeds) {
  EXPECT_CALL(driver(), TryLaunch(_)).WillOnce(Invoke([](const GURL& url) {
    EXPECT_EQ(kExampleDotCom, url.spec());
    return true;
  }));
  EXPECT_TRUE(launcher()->Launch(GURL(kExampleDotCom)));
}

TEST_F(AlternativeBrowserLauncherTest, LaunchFails) {
  EXPECT_CALL(driver(), TryLaunch(_)).WillOnce(Return(false));
  EXPECT_FALSE(launcher()->Launch(GURL(kExampleDotCom)));
}

TEST_F(AlternativeBrowserLauncherTest, LaunchPicksUpPrefChanges) {
  EXPECT_CALL(driver(), SetBrowserPath(_))
      .WillOnce(
          Invoke([](base::StringPiece path) { EXPECT_EQ("bogus.exe", path); }));
  prefs()->SetManagedPref(prefs::kAlternativeBrowserPath,
                          std::make_unique<base::Value>("bogus.exe"));
  EXPECT_CALL(driver(), SetBrowserParameters(_))
      .WillOnce(Invoke([](const base::ListValue* parameters) {
        EXPECT_EQ(1u, parameters->GetList().size());
        EXPECT_EQ(base::Value("--single-process"), parameters->GetList()[0]);
      }));
  std::vector<base::Value> params;
  params.emplace_back("--single-process");
  prefs()->SetManagedPref(prefs::kAlternativeBrowserParameters,
                          std::make_unique<base::Value>(std::move(params)));
}

TEST_F(AlternativeBrowserLauncherTest, LaunchIgnoresNonManagedPrefs) {
  EXPECT_CALL(driver(), SetBrowserPath(_))
      .WillOnce(Invoke([](base::StringPiece path) { EXPECT_EQ("", path); }));
  EXPECT_CALL(driver(), SetBrowserParameters(_))
      .WillOnce(Invoke([](const base::ListValue* parameters) {
        EXPECT_TRUE(parameters->GetList().empty());
      }));
  prefs()->Set(prefs::kAlternativeBrowserPath, base::Value("evil.exe"));
  std::vector<base::Value> params;
  params.emplace_back("--launch-missiles");
  prefs()->Set(prefs::kAlternativeBrowserParameters,
               base::Value(std::move(params)));
}

}  // namespace browser_switcher