chrome/browser/extensions/api/tabs/windows_util_unittest.cc

// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/extensions/api/tabs/windows_util.h"

#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/bookmarks/bookmark_model_factory.h"
#include "chrome/browser/extensions/extension_service_test_base.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/test_browser_window.h"
#include "chrome/test/base/testing_profile.h"
#include "components/bookmarks/browser/bookmark_model.h"
#include "components/bookmarks/test/bookmark_test_helpers.h"
#include "components/policy/core/common/policy_pref_names.h"
#include "extensions/browser/api_test_utils.h"

namespace windows_util {

class WindowsUtilUnitTest : public extensions::ExtensionServiceTestBase {
 public:
  void SetUp() override {
    ExtensionServiceTestBase::SetUp();
    InitializeExtensionService(ExtensionServiceInitParams());
  }
};

TEST_F(WindowsUtilUnitTest, ShouldOpenIncognitoWindowIncognitoDisabled) {
  // Incognito disabled.
  IncognitoModePrefs::SetAvailability(
      profile()->GetPrefs(), policy::IncognitoModeAvailability::kDisabled);

  std::string error;
  std::vector<GURL> urls;
  urls.emplace_back(GURL("https://google.com"));
  EXPECT_EQ(
      IncognitoResult::kError,
      ShouldOpenIncognitoWindow(profile(), /*incognito=*/true, &urls, &error));
  EXPECT_EQ("Incognito mode is disabled.", error);
}

TEST_F(WindowsUtilUnitTest, ShouldOpenIncognitoWindowIncognitoForced) {
  // Incognito forced.
  IncognitoModePrefs::SetAvailability(
      profile()->GetPrefs(), policy::IncognitoModeAvailability::kForced);

  std::string error;
  std::vector<GURL> urls;
  urls.emplace_back(GURL("https://google.com"));
  EXPECT_EQ(
      IncognitoResult::kError,
      ShouldOpenIncognitoWindow(profile(), /*incognito=*/false, &urls, &error));
  EXPECT_EQ("Incognito mode is forced. Cannot open normal windows.", error);
}

TEST_F(WindowsUtilUnitTest, ShouldOpenIncognitoWindowIncompatibleURL) {
  std::string error;
  std::vector<GURL> urls;
  urls.emplace_back(GURL("chrome://history"));
  EXPECT_EQ(
      IncognitoResult::kError,
      ShouldOpenIncognitoWindow(profile(), /*incognito=*/true, &urls, &error));
  EXPECT_EQ("Cannot open URL \"chrome://history/\" in an incognito window.",
            error);
  EXPECT_TRUE(urls.empty());
}

TEST_F(WindowsUtilUnitTest,
       ShouldOpenIncognitoWindowIncompatibleURLWithSomeLeft) {
  std::string error;
  std::vector<GURL> urls;
  urls.emplace_back(GURL("chrome://history"));
  urls.emplace_back(GURL("https://google.com"));
  EXPECT_EQ(
      IncognitoResult::kIncognito,
      ShouldOpenIncognitoWindow(profile(), /*incognito=*/true, &urls, &error));

  size_t expected_size = 1;
  EXPECT_EQ(expected_size, urls.size());
}

}  // namespace windows_util