Refactoring startup logic for upcoming FRE changes (non-Win 10).

chrome/browser/first_run/first_run_unittest.cc

 // Copyright (c) 2011 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 "base/compiler_specific.h"
 #include "base/files/file_util.h"
 #include "base/macros.h"
+#include "base/strings/utf_string_conversions.h"
 #include "base/test/scoped_path_override.h"
 #include "base/values.h"
 #include "chrome/browser/first_run/first_run.h"
 #include "chrome/browser/first_run/first_run_internal.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/url_constants.h"
 #include "chrome/installer/util/master_preferences.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
 
 namespace first_run {
 
 class FirstRunTest : public testing::Test {
  protected:
   FirstRunTest() : user_data_dir_override_(chrome::DIR_USER_DATA) {}
   ~FirstRunTest() override {}
 
  private:
   base::ScopedPathOverride user_data_dir_override_;
@@ skipping 54 matching lines @@
 
 TEST_F(FirstRunTest,
        SetupMasterPrefsFromInstallPrefs_WelcomePageOnOSUpgradeDisabled) {
   installer::MasterPreferences install_prefs(
       "{\"distribution\":{\"welcome_page_on_os_upgrade_enabled\": false}}");
   MasterPrefs out_prefs;
   internal::SetupMasterPrefsFromInstallPrefs(install_prefs, &out_prefs);
   EXPECT_FALSE(out_prefs.welcome_page_on_os_upgrade_enabled);
 }
 
+TEST_F(FirstRunTest, ProcessMasterPrefsTabs) {
+  std::vector<GURL> input;
+  input.push_back(GURL(base::ASCIIToUTF16("https://new_tab_page")));
+  input.push_back(GURL(base::ASCIIToUTF16("https://www.google.com")));
+  input.push_back(GURL(base::ASCIIToUTF16("https://welcome_page")));
+  std::vector<GURL> output = ProcessMasterPrefsTabs(input);
+  ASSERT_EQ(3U, output.size());
+  EXPECT_EQ(GURL(chrome::kChromeUINewTabURL).host(), output[0].host());
+  EXPECT_EQ("www.google.com", output[1].host());
+  EXPECT_EQ(GetWelcomePageURL().host(), output[2].host());
+}
+
+// No flags, no sentinel present. Welcome should be surfaced.
+TEST_F(FirstRunTest, GetOnboardingTabs_StandardFirstRun) {
+  FirstRunSystemStatus status;
+  status.os_class = FirstRunSystemStatus::OperatingSystemClass::STANDARD;
+  status.is_sentinel_present = false;
+  status.is_force_flag_on = false;
+  status.is_suppress_flag_on = false;
+
+  std::vector<GURL> output = GetOnboardingTabs(status);
+
+  ASSERT_EQ(1U, output.size());
+  EXPECT_EQ(GetWelcomePageURL().host(), output[0].host());
+}
+
+// Sentinel present, no flags. Nothing should be shown.
+TEST_F(FirstRunTest, GetOnboardingTabs_StandardWithSentinel) {
+  FirstRunSystemStatus status;
+  status.os_class = FirstRunSystemStatus::OperatingSystemClass::STANDARD;
+  status.is_sentinel_present = true;
+  status.is_force_flag_on = false;
+  status.is_suppress_flag_on = true;
+
+  std::vector<GURL> output = GetOnboardingTabs(status);
+
+  EXPECT_EQ(0U, output.size());
+}
+
+// Even without sentinel, the suppress flag means no tabs come back.
+TEST_F(FirstRunTest, GetOnboardingTabs_StandardSuppressed) {
+  FirstRunSystemStatus status;
+  status.os_class = FirstRunSystemStatus::OperatingSystemClass::STANDARD;
+  status.is_sentinel_present = false;
+  status.is_force_flag_on = false;
+  status.is_suppress_flag_on = true;
+
+  std::vector<GURL> output = GetOnboardingTabs(status);
+
+  EXPECT_EQ(0U, output.size());
+}
+
+// Force First Run flag overrides both the presence of the sentinel file and
+// the suppress flag, and the welcome page is still shown.
+TEST_F(FirstRunTest, GetOnboardingTabs_StandardForced) {
+  FirstRunSystemStatus status;
+  status.os_class = FirstRunSystemStatus::OperatingSystemClass::STANDARD;
+  status.is_sentinel_present = true;
+  status.is_force_flag_on = true;
+  status.is_suppress_flag_on = true;
+
+  std::vector<GURL> output = GetOnboardingTabs(status);
+
+  ASSERT_EQ(1U, output.size());
+  EXPECT_EQ(GetWelcomePageURL().host(), output[0].host());
+}
+
 }  // namespace first_run