components/bookmarks/managed/managed_bookmarks_tracker_unittest.cc

   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/managed/managed_bookmarks_tracker.h"
   6
   7 #include "base/bind.h"
   8 #include "base/files/file_path.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/message_loop/message_loop.h"
  11 #include "base/prefs/pref_registry_simple.h"
  12 #include "base/prefs/testing_pref_service.h"
  13 #include "base/strings/utf_string_conversions.h"
  14 #include "components/bookmarks/browser/bookmark_model.h"
  15 #include "components/bookmarks/browser/bookmark_model_observer.h"
  16 #include "components/bookmarks/browser/bookmark_node.h"
  17 #include "components/bookmarks/common/bookmark_pref_names.h"
  18 #include "components/bookmarks/test/bookmark_test_helpers.h"
  19 #include "components/bookmarks/test/mock_bookmark_model_observer.h"
  20 #include "components/bookmarks/test/test_bookmark_client.h"
  21 #include "grit/components_strings.h"
  22 #include "testing/gmock/include/gmock/gmock.h"
  23 #include "testing/gtest/include/gtest/gtest.h"
  24 #include "ui/base/l10n/l10n_util.h"
  25 #include "url/gurl.h"
  26
  27 using testing::Mock;
  28 using testing::_;
  29
  30 namespace bookmarks {
  31
  32 class ManagedBookmarksTrackerTest : public testing::Test {
  33  public:
  34   ManagedBookmarksTrackerTest() : managed_node_(NULL) {}
  35   ~ManagedBookmarksTrackerTest() override {}
  36
  37   void SetUp() override {
  38     prefs_.registry()->RegisterListPref(bookmarks::prefs::kManagedBookmarks);
  39     prefs_.registry()->RegisterListPref(bookmarks::prefs::kSupervisedBookmarks);
  40     prefs_.registry()->RegisterListPref(
  41         bookmarks::prefs::kBookmarkEditorExpandedNodes);
  42   }
  43
  44   void TearDown() override {
  45     if (model_)
  46       model_->RemoveObserver(&observer_);
  47   }
  48
  49   void CreateModel(bool is_supervised) {
  50     // Simulate the creation of the managed node by the BookmarkClient.
  51     BookmarkPermanentNode* managed_node = new BookmarkPermanentNode(100);
  52     ManagedBookmarksTracker::LoadInitial(
  53         managed_node,
  54         prefs_.GetList(ManagedBookmarksTracker::GetPrefName(is_supervised)),
  55         101);
  56     managed_node->set_visible(!managed_node->empty());
  57     managed_node->SetTitle(l10n_util::GetStringUTF16(
  58         is_supervised ? IDS_BOOKMARK_BAR_SUPERVISED_FOLDER_DEFAULT_NAME
  59                       : IDS_BOOKMARK_BAR_MANAGED_FOLDER_DEFAULT_NAME));
  60
  61     bookmarks::BookmarkPermanentNodeList extra_nodes;
  62     extra_nodes.push_back(managed_node);
  63     client_.SetExtraNodesToLoad(extra_nodes.Pass());
  64
  65     model_.reset(new BookmarkModel(&client_));
  66     model_->AddObserver(&observer_);
  67     EXPECT_CALL(observer_, BookmarkModelLoaded(model_.get(), _));
  68     model_->Load(&prefs_,
  69                  std::string(),
  70                  base::FilePath(),
  71                  base::MessageLoopProxy::current(),
  72                  base::MessageLoopProxy::current());
  73     bookmarks::test::WaitForBookmarkModelToLoad(model_.get());
  74     Mock::VerifyAndClearExpectations(&observer_);
  75
  76     ASSERT_EQ(1u, client_.extra_nodes().size());
  77     managed_node_ = client_.extra_nodes()[0];
  78     ASSERT_EQ(managed_node, managed_node_);
  79
  80     managed_bookmarks_tracker_.reset(new ManagedBookmarksTracker(
  81         model_.get(),
  82         &prefs_,
  83         is_supervised,
  84         base::Bind(&ManagedBookmarksTrackerTest::GetManagementDomain)));
  85     managed_bookmarks_tracker_->Init(managed_node_);
  86   }
  87
  88   const BookmarkNode* managed_node() {
  89     return managed_node_;
  90   }
  91
  92   bool IsManaged(const BookmarkNode* node) {
  93     return node && node->HasAncestor(managed_node_);
  94   }
  95
  96   static base::DictionaryValue* CreateBookmark(const std::string& title,
  97                                                const std::string& url) {
  98     EXPECT_TRUE(GURL(url).is_valid());
  99     base::DictionaryValue* dict = new base::DictionaryValue();
 100     dict->SetString("name", title);
 101     dict->SetString("url", GURL(url).spec());
 102     return dict;
 103   }
 104
 105   static base::DictionaryValue* CreateFolder(const std::string& title,
 106                                              base::ListValue* children) {
 107     base::DictionaryValue* dict = new base::DictionaryValue();
 108     dict->SetString("name", title);
 109     dict->Set("children", children);
 110     return dict;
 111   }
 112
 113   static base::ListValue* CreateTestTree() {
 114     base::ListValue* folder = new base::ListValue();
 115     base::ListValue* empty = new base::ListValue();
 116     folder->Append(CreateFolder("Empty", empty));
 117     folder->Append(CreateBookmark("Youtube", "http://youtube.com/"));
 118
 119     base::ListValue* list = new base::ListValue();
 120     list->Append(CreateBookmark("Google", "http://google.com/"));
 121     list->Append(CreateFolder("Folder", folder));
 122
 123     return list;
 124   }
 125
 126   static std::string GetManagementDomain() {
 127     return std::string();
 128   }
 129
 130   static std::string GetManagedFolderTitle() {
 131     return l10n_util::GetStringUTF8(
 132         IDS_BOOKMARK_BAR_MANAGED_FOLDER_DEFAULT_NAME);
 133   }
 134
 135   static base::DictionaryValue* CreateExpectedTree() {
 136     return CreateFolder(GetManagedFolderTitle(), CreateTestTree());
 137   }
 138
 139   static bool NodeMatchesValue(const BookmarkNode* node,
 140                                const base::DictionaryValue* dict) {
 141     base::string16 title;
 142     if (!dict->GetString("name", &title) || node->GetTitle() != title)
 143       return false;
 144
 145     if (node->is_folder()) {
 146       const base::ListValue* children = NULL;
 147       if (!dict->GetList("children", &children) ||
 148           node->child_count() != static_cast<int>(children->GetSize())) {
 149         return false;
 150       }
 151       for (int i = 0; i < node->child_count(); ++i) {
 152         const base::DictionaryValue* child = NULL;
 153         if (!children->GetDictionary(i, &child) ||
 154             !NodeMatchesValue(node->GetChild(i), child)) {
 155           return false;
 156         }
 157       }
 158     } else if (node->is_url()) {
 159       std::string url;
 160       if (!dict->GetString("url", &url) || node->url() != GURL(url))
 161         return false;
 162     } else {
 163       return false;
 164     }
 165     return true;
 166   }
 167
 168   base::MessageLoop loop_;
 169   TestingPrefServiceSimple prefs_;
 170   bookmarks::TestBookmarkClient client_;
 171   scoped_ptr<BookmarkModel> model_;
 172   bookmarks::MockBookmarkModelObserver observer_;
 173   BookmarkPermanentNode* managed_node_;
 174   scoped_ptr<ManagedBookmarksTracker> managed_bookmarks_tracker_;
 175 };
 176
 177 TEST_F(ManagedBookmarksTrackerTest, Empty) {
 178   CreateModel(false /* is_supervised */);
 179   EXPECT_TRUE(model_->bookmark_bar_node()->empty());
 180   EXPECT_TRUE(model_->other_node()->empty());
 181   EXPECT_TRUE(managed_node()->empty());
 182   EXPECT_FALSE(managed_node()->IsVisible());
 183 }
 184
 185 TEST_F(ManagedBookmarksTrackerTest, LoadInitial) {
 186   // Set a policy before loading the model.
 187   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 188   CreateModel(false /* is_supervised */);
 189   EXPECT_TRUE(model_->bookmark_bar_node()->empty());
 190   EXPECT_TRUE(model_->other_node()->empty());
 191   EXPECT_FALSE(managed_node()->empty());
 192   EXPECT_TRUE(managed_node()->IsVisible());
 193
 194   scoped_ptr<base::DictionaryValue> expected(CreateExpectedTree());
 195   EXPECT_TRUE(NodeMatchesValue(managed_node(), expected.get()));
 196 }
 197
 198 TEST_F(ManagedBookmarksTrackerTest, SupervisedTrackerIgnoresManagedPref) {
 199   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 200   CreateModel(true /* is_supervised */);
 201   EXPECT_TRUE(managed_node()->empty());
 202   EXPECT_FALSE(managed_node()->IsVisible());
 203 }
 204
 205 TEST_F(ManagedBookmarksTrackerTest, SupervisedTrackerHandlesSupervisedPref) {
 206   prefs_.SetManagedPref(bookmarks::prefs::kSupervisedBookmarks,
 207                         CreateTestTree());
 208   CreateModel(true /* is_supervised */);
 209   EXPECT_FALSE(managed_node()->empty());
 210   EXPECT_TRUE(managed_node()->IsVisible());
 211   // Don't bother checking the actual contents, the non-supervised tests cover
 212   // that already.
 213 }
 214
 215 TEST_F(ManagedBookmarksTrackerTest, SwapNodes) {
 216   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 217   CreateModel(false /* is_supervised */);
 218
 219   // Swap the Google bookmark with the Folder.
 220   scoped_ptr<base::ListValue> updated(CreateTestTree());
 221   scoped_ptr<base::Value> removed;
 222   ASSERT_TRUE(updated->Remove(0, &removed));
 223   updated->Append(removed.release());
 224
 225   // These two nodes should just be swapped.
 226   const BookmarkNode* parent = managed_node();
 227   EXPECT_CALL(observer_, BookmarkNodeMoved(model_.get(), parent, 1, parent, 0));
 228   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks,
 229                         updated->DeepCopy());
 230   Mock::VerifyAndClearExpectations(&observer_);
 231
 232   // Verify the final tree.
 233   scoped_ptr<base::DictionaryValue> expected(
 234       CreateFolder(GetManagedFolderTitle(), updated.release()));
 235   EXPECT_TRUE(NodeMatchesValue(managed_node(), expected.get()));
 236 }
 237
 238 TEST_F(ManagedBookmarksTrackerTest, RemoveNode) {
 239   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 240   CreateModel(false /* is_supervised */);
 241
 242   // Remove the Folder.
 243   scoped_ptr<base::ListValue> updated(CreateTestTree());
 244   ASSERT_TRUE(updated->Remove(1, NULL));
 245
 246   const BookmarkNode* parent = managed_node();
 247   EXPECT_CALL(observer_, BookmarkNodeRemoved(model_.get(), parent, 1, _, _));
 248   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks,
 249                         updated->DeepCopy());
 250   Mock::VerifyAndClearExpectations(&observer_);
 251
 252   // Verify the final tree.
 253   scoped_ptr<base::DictionaryValue> expected(
 254       CreateFolder(GetManagedFolderTitle(), updated.release()));
 255   EXPECT_TRUE(NodeMatchesValue(managed_node(), expected.get()));
 256 }
 257
 258 TEST_F(ManagedBookmarksTrackerTest, CreateNewNodes) {
 259   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 260   CreateModel(false /* is_supervised */);
 261
 262   // Put all the nodes inside another folder.
 263   scoped_ptr<base::ListValue> updated(new base::ListValue);
 264   updated->Append(CreateFolder("Container", CreateTestTree()));
 265
 266   EXPECT_CALL(observer_, BookmarkNodeAdded(model_.get(), _, _)).Times(5);
 267   // The remaining nodes have been pushed to positions 1 and 2; they'll both be
 268   // removed when at position 1.
 269   const BookmarkNode* parent = managed_node();
 270   EXPECT_CALL(observer_, BookmarkNodeRemoved(model_.get(), parent, 1, _, _))
 271       .Times(2);
 272   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks,
 273                         updated->DeepCopy());
 274   Mock::VerifyAndClearExpectations(&observer_);
 275
 276   // Verify the final tree.
 277   scoped_ptr<base::DictionaryValue> expected(
 278       CreateFolder(GetManagedFolderTitle(), updated.release()));
 279   EXPECT_TRUE(NodeMatchesValue(managed_node(), expected.get()));
 280 }
 281
 282 TEST_F(ManagedBookmarksTrackerTest, RemoveAll) {
 283   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 284   CreateModel(false /* is_supervised */);
 285   EXPECT_TRUE(managed_node()->IsVisible());
 286
 287   // Remove the policy.
 288   const BookmarkNode* parent = managed_node();
 289   EXPECT_CALL(observer_, BookmarkNodeRemoved(model_.get(), parent, 0, _, _))
 290       .Times(2);
 291   prefs_.RemoveManagedPref(bookmarks::prefs::kManagedBookmarks);
 292   Mock::VerifyAndClearExpectations(&observer_);
 293
 294   EXPECT_TRUE(managed_node()->empty());
 295   EXPECT_FALSE(managed_node()->IsVisible());
 296 }
 297
 298 TEST_F(ManagedBookmarksTrackerTest, IsManaged) {
 299   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 300   CreateModel(false /* is_supervised */);
 301
 302   EXPECT_FALSE(IsManaged(model_->root_node()));
 303   EXPECT_FALSE(IsManaged(model_->bookmark_bar_node()));
 304   EXPECT_FALSE(IsManaged(model_->other_node()));
 305   EXPECT_FALSE(IsManaged(model_->mobile_node()));
 306   EXPECT_TRUE(IsManaged(managed_node()));
 307
 308   const BookmarkNode* parent = managed_node();
 309   ASSERT_EQ(2, parent->child_count());
 310   EXPECT_TRUE(IsManaged(parent->GetChild(0)));
 311   EXPECT_TRUE(IsManaged(parent->GetChild(1)));
 312
 313   parent = parent->GetChild(1);
 314   ASSERT_EQ(2, parent->child_count());
 315   EXPECT_TRUE(IsManaged(parent->GetChild(0)));
 316   EXPECT_TRUE(IsManaged(parent->GetChild(1)));
 317 }
 318
 319 TEST_F(ManagedBookmarksTrackerTest, RemoveAllUserBookmarksDoesntRemoveManaged) {
 320   prefs_.SetManagedPref(bookmarks::prefs::kManagedBookmarks, CreateTestTree());
 321   CreateModel(false /* is_supervised */);
 322   EXPECT_EQ(2, managed_node()->child_count());
 323
 324   EXPECT_CALL(observer_,
 325               BookmarkNodeAdded(model_.get(), model_->bookmark_bar_node(), 0));
 326   EXPECT_CALL(observer_,
 327               BookmarkNodeAdded(model_.get(), model_->bookmark_bar_node(), 1));
 328   model_->AddURL(model_->bookmark_bar_node(),
 329                  0,
 330                  base::ASCIIToUTF16("Test"),
 331                  GURL("http://google.com/"));
 332   model_->AddFolder(
 333       model_->bookmark_bar_node(), 1, base::ASCIIToUTF16("Test Folder"));
 334   EXPECT_EQ(2, model_->bookmark_bar_node()->child_count());
 335   Mock::VerifyAndClearExpectations(&observer_);
 336
 337   EXPECT_CALL(observer_, BookmarkAllUserNodesRemoved(model_.get(), _));
 338   model_->RemoveAllUserBookmarks();
 339   EXPECT_EQ(2, managed_node()->child_count());
 340   EXPECT_EQ(0, model_->bookmark_bar_node()->child_count());
 341   Mock::VerifyAndClearExpectations(&observer_);
 342 }
 343
 344 }  // namespace bookmarks