1// Copyright 2014 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 "components/bookmarks/browser/bookmark_expanded_state_tracker.h"
6
7#include "base/files/file_path.h"
8#include "base/message_loop/message_loop.h"
9#include "base/prefs/pref_service.h"
10#include "base/prefs/pref_service_factory.h"
11#include "base/prefs/testing_pref_store.h"
12#include "base/run_loop.h"
13#include "base/strings/utf_string_conversions.h"
14#include "components/bookmarks/browser/bookmark_model.h"
15#include "components/bookmarks/common/bookmark_pref_names.h"
16#include "components/bookmarks/test/bookmark_test_helpers.h"
17#include "components/bookmarks/test/test_bookmark_client.h"
18#include "components/pref_registry/pref_registry_syncable.h"
19#include "testing/gtest/include/gtest/gtest.h"
20
21namespace bookmarks {
22
23scoped_ptr<PrefService> PrefServiceForTesting() {
24  scoped_refptr<user_prefs::PrefRegistrySyncable> registry(
25      new user_prefs::PrefRegistrySyncable());
26  registry->RegisterListPref(prefs::kBookmarkEditorExpandedNodes,
27                             new base::ListValue,
28                             user_prefs::PrefRegistrySyncable::UNSYNCABLE_PREF);
29  registry->RegisterListPref(prefs::kManagedBookmarks,
30                             user_prefs::PrefRegistrySyncable::UNSYNCABLE_PREF);
31  base::PrefServiceFactory factory;
32  factory.set_user_prefs(make_scoped_refptr(new TestingPrefStore()));
33  return factory.Create(registry.get());
34}
35
36class BookmarkExpandedStateTrackerTest : public testing::Test {
37 public:
38  BookmarkExpandedStateTrackerTest();
39  virtual ~BookmarkExpandedStateTrackerTest();
40
41 protected:
42  // testing::Test:
43  virtual void SetUp() OVERRIDE;
44  virtual void TearDown() OVERRIDE;
45
46  base::MessageLoop message_loop_;
47  TestBookmarkClient client_;
48  scoped_ptr<PrefService> prefs_;
49  scoped_ptr<BookmarkModel> model_;
50
51  DISALLOW_COPY_AND_ASSIGN(BookmarkExpandedStateTrackerTest);
52};
53
54BookmarkExpandedStateTrackerTest::BookmarkExpandedStateTrackerTest() {}
55
56BookmarkExpandedStateTrackerTest::~BookmarkExpandedStateTrackerTest() {}
57
58void BookmarkExpandedStateTrackerTest::SetUp() {
59  prefs_ = PrefServiceForTesting();
60  model_.reset(new BookmarkModel(&client_));
61  model_->Load(prefs_.get(),
62               std::string(),
63               base::FilePath(),
64               base::MessageLoopProxy::current(),
65               base::MessageLoopProxy::current());
66  test::WaitForBookmarkModelToLoad(model_.get());
67}
68
69void BookmarkExpandedStateTrackerTest::TearDown() {
70  model_.reset();
71}
72
73// Various assertions for SetExpandedNodes.
74TEST_F(BookmarkExpandedStateTrackerTest, SetExpandedNodes) {
75  BookmarkExpandedStateTracker* tracker = model_->expanded_state_tracker();
76
77  // Should start out initially empty.
78  EXPECT_TRUE(tracker->GetExpandedNodes().empty());
79
80  BookmarkExpandedStateTracker::Nodes nodes;
81  nodes.insert(model_->bookmark_bar_node());
82  tracker->SetExpandedNodes(nodes);
83  EXPECT_EQ(nodes, tracker->GetExpandedNodes());
84
85  // Add a folder and mark it expanded.
86  const BookmarkNode* n1 = model_->AddFolder(
87      model_->bookmark_bar_node(), 0, base::ASCIIToUTF16("x"));
88  nodes.insert(n1);
89  tracker->SetExpandedNodes(nodes);
90  EXPECT_EQ(nodes, tracker->GetExpandedNodes());
91
92  // Remove the folder, which should remove it from the list of expanded nodes.
93  model_->Remove(model_->bookmark_bar_node(), 0);
94  nodes.erase(n1);
95  n1 = NULL;
96  EXPECT_EQ(nodes, tracker->GetExpandedNodes());
97}
98
99TEST_F(BookmarkExpandedStateTrackerTest, RemoveAllUserBookmarks) {
100  BookmarkExpandedStateTracker* tracker = model_->expanded_state_tracker();
101
102  // Add a folder and mark it expanded.
103  const BookmarkNode* n1 = model_->AddFolder(
104      model_->bookmark_bar_node(), 0, base::ASCIIToUTF16("x"));
105  BookmarkExpandedStateTracker::Nodes nodes;
106  nodes.insert(n1);
107  tracker->SetExpandedNodes(nodes);
108  // Verify that the node is present.
109  EXPECT_EQ(nodes, tracker->GetExpandedNodes());
110  // Call remove all.
111  model_->RemoveAllUserBookmarks();
112  // Verify node is not present.
113  EXPECT_TRUE(tracker->GetExpandedNodes().empty());
114}
115
116}  // namespace bookmarks
117