Completing refactor of startup_browser_creator_impl

chrome/browser/ui/startup/startup_tab_provider_unittest.cc

 // 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/startup_tab_provider.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/common/url_constants.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
@@ skipping 50 matching lines @@
   EXPECT_FALSE(output[0].is_pinned);
 }
 
 TEST(StartupTabProviderTest, CheckResetTriggerTabPolicy_Negative) {
   StartupTabs output =
       StartupTabProviderImpl::CheckResetTriggerTabPolicy(false);
 
   ASSERT_TRUE(output.empty());
 }
 
+TEST(StartupTabProviderTest, CheckPinnedTabPolicy) {
+  StartupTabs pinned = {StartupTab(GURL("https://www.google.com"), true)};
+  SessionStartupPref pref_default(SessionStartupPref::Type::DEFAULT);
+  SessionStartupPref pref_urls(SessionStartupPref::Type::URLS);
+
+  StartupTabs output =
+      StartupTabProviderImpl::CheckPinnedTabPolicy(pref_default, pinned);
+
+  ASSERT_EQ(1U, output.size());
+  EXPECT_EQ("www.google.com", output[0].url.host());
+
+  output = StartupTabProviderImpl::CheckPinnedTabPolicy(pref_urls, pinned);
+
+  ASSERT_EQ(1U, output.size());
+  EXPECT_EQ("www.google.com", output[0].url.host());
+}
+
+TEST(StartupTabProviderTest, CheckPinnedTabPolicy_Negative) {
+  StartupTabs pinned = {StartupTab(GURL("https://www.google.com"), true)};
+  SessionStartupPref pref_last(SessionStartupPref::Type::LAST);
+
+  StartupTabs output =
+      StartupTabProviderImpl::CheckPinnedTabPolicy(pref_last, pinned);
+
+  ASSERT_TRUE(output.empty());
+}
+
 TEST(StartupTabProviderTest, CheckPreferencesTabPolicy) {
   SessionStartupPref pref(SessionStartupPref::Type::URLS);
   pref.urls = {GURL(base::ASCIIToUTF16("https://www.google.com"))};
 
   StartupTabs output = StartupTabProviderImpl::CheckPreferencesTabPolicy(pref);
 
   ASSERT_EQ(1U, output.size());
   EXPECT_EQ("www.google.com", output[0].url.host());
 }
 
 TEST(StartupTabProviderTest, CheckPreferencesTabPolicy_Negative) {
   SessionStartupPref pref_default(SessionStartupPref::Type::DEFAULT);
   pref_default.urls = {GURL(base::ASCIIToUTF16("https://www.google.com"))};
 
   StartupTabs output =
       StartupTabProviderImpl::CheckPreferencesTabPolicy(pref_default);
 
   EXPECT_TRUE(output.empty());
 
   SessionStartupPref pref_last(SessionStartupPref::Type::LAST);
   pref_last.urls = {GURL(base::ASCIIToUTF16("https://www.google.com"))};
 
   output = StartupTabProviderImpl::CheckPreferencesTabPolicy(pref_last);
 
   EXPECT_TRUE(output.empty());
 }
+
+TEST(StartupTabProviderTest, CheckNewTabPageTabPolicy) {
+  SessionStartupPref pref_default(SessionStartupPref::Type::DEFAULT);
+  SessionStartupPref pref_urls(SessionStartupPref::Type::URLS);
+
+  StartupTabs output =
+      StartupTabProviderImpl::CheckNewTabPageTabPolicy(pref_default);
+
+  ASSERT_EQ(1U, output.size());
+  EXPECT_EQ(GURL(chrome::kChromeUINewTabURL), output[0].url);
+
+  output = StartupTabProviderImpl::CheckNewTabPageTabPolicy(pref_urls);
+
+  ASSERT_EQ(1U, output.size());
+  EXPECT_EQ(GURL(chrome::kChromeUINewTabURL), output[0].url);
+}
+
+TEST(StartupTabProviderTest, CheckNewTabPageTabPolicy_Negative) {
+  SessionStartupPref pref_last(SessionStartupPref::Type::LAST);
+
+  StartupTabs output =
+      StartupTabProviderImpl::CheckNewTabPageTabPolicy(pref_last);
+
+  ASSERT_TRUE(output.empty());
+}