1 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/prefs/session_startup_pref.h"
6 #include "chrome/common/pref_names.h"
7 #include "chrome/test/testing_pref_service.h"
8 #include "testing/gmock/include/gmock/gmock.h"
9 #include "testing/gtest/include/gtest/gtest.h"
10
11 // Unit tests for SessionStartupPref.
12 class SessionStartupPrefTest : public testing::Test {
13 public:
SetUp()14 virtual void SetUp() {
15 pref_service_.reset(new TestingPrefService);
16 SessionStartupPref::RegisterUserPrefs(pref_service_.get());
17 }
18
19 scoped_ptr<TestingPrefService> pref_service_;
20 };
21
TEST_F(SessionStartupPrefTest,URLListIsFixedUp)22 TEST_F(SessionStartupPrefTest, URLListIsFixedUp) {
23 ListValue* url_pref_list = new ListValue;
24 url_pref_list->Set(0, Value::CreateStringValue("google.com"));
25 url_pref_list->Set(1, Value::CreateStringValue("chromium.org"));
26 pref_service_->SetUserPref(prefs::kURLsToRestoreOnStartup, url_pref_list);
27
28 SessionStartupPref result =
29 SessionStartupPref::GetStartupPref(pref_service_.get());
30 EXPECT_EQ(2u, result.urls.size());
31 EXPECT_EQ("http://google.com/", result.urls[0].spec());
32 EXPECT_EQ("http://chromium.org/", result.urls[1].spec());
33 }
34
TEST_F(SessionStartupPrefTest,URLListManagedOverridesUser)35 TEST_F(SessionStartupPrefTest, URLListManagedOverridesUser) {
36 ListValue* url_pref_list1 = new ListValue;
37 url_pref_list1->Set(0, Value::CreateStringValue("chromium.org"));
38 pref_service_->SetUserPref(prefs::kURLsToRestoreOnStartup, url_pref_list1);
39
40 ListValue* url_pref_list2 = new ListValue;
41 url_pref_list2->Set(0, Value::CreateStringValue("chromium.org"));
42 url_pref_list2->Set(1, Value::CreateStringValue("chromium.org"));
43 url_pref_list2->Set(2, Value::CreateStringValue("chromium.org"));
44 pref_service_->SetManagedPref(prefs::kURLsToRestoreOnStartup,
45 url_pref_list2);
46
47 SessionStartupPref result =
48 SessionStartupPref::GetStartupPref(pref_service_.get());
49 EXPECT_EQ(3u, result.urls.size());
50
51 SessionStartupPref override_test =
52 SessionStartupPref(SessionStartupPref::URLS);
53 override_test.urls.push_back(GURL("dev.chromium.org"));
54 SessionStartupPref::SetStartupPref(pref_service_.get(), override_test);
55
56 result = SessionStartupPref::GetStartupPref(pref_service_.get());
57 EXPECT_EQ(3u, result.urls.size());
58 }
59