1// Copyright (c) 2012 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/ui/bookmarks/recently_used_folders_combo_model.h"
6
7#include "base/memory/scoped_ptr.h"
8#include "base/message_loop/message_loop.h"
9#include "base/strings/utf_string_conversions.h"
10#include "chrome/browser/bookmarks/bookmark_model_factory.h"
11#include "chrome/test/base/testing_profile.h"
12#include "components/bookmarks/browser/bookmark_model.h"
13#include "components/bookmarks/test/bookmark_test_helpers.h"
14#include "content/public/test/test_browser_thread.h"
15#include "testing/gtest/include/gtest/gtest.h"
16#include "ui/base/models/combobox_model_observer.h"
17
18using content::BrowserThread;
19
20// Implementation of ComboboxModelObserver that records when
21// OnComboboxModelChanged() is invoked.
22class TestComboboxModelObserver : public ui::ComboboxModelObserver {
23 public:
24  TestComboboxModelObserver() : changed_(false) {}
25  virtual ~TestComboboxModelObserver() {}
26
27  // Returns whether the model changed and clears changed state.
28  bool GetAndClearChanged() {
29    const bool changed = changed_;
30    changed_ = false;
31    return changed;
32  }
33
34  // ComboboxModelObserver:
35  virtual void OnComboboxModelChanged(ui::ComboboxModel* model) OVERRIDE {
36    changed_ = true;
37  }
38
39 private:
40  bool changed_;
41
42  DISALLOW_COPY_AND_ASSIGN(TestComboboxModelObserver);
43};
44
45class RecentlyUsedFoldersComboModelTest : public testing::Test {
46 public:
47  RecentlyUsedFoldersComboModelTest();
48
49  virtual void SetUp() OVERRIDE;
50  virtual void TearDown() OVERRIDE;
51
52 protected:
53  BookmarkModel* GetModel();
54
55 private:
56  base::MessageLoopForUI message_loop_;
57  content::TestBrowserThread ui_thread_;
58  content::TestBrowserThread file_thread_;
59  scoped_ptr<TestingProfile> profile_;
60
61  DISALLOW_COPY_AND_ASSIGN(RecentlyUsedFoldersComboModelTest);
62};
63
64RecentlyUsedFoldersComboModelTest::RecentlyUsedFoldersComboModelTest()
65    : ui_thread_(BrowserThread::UI, &message_loop_),
66      file_thread_(BrowserThread::FILE, &message_loop_) {
67}
68
69void RecentlyUsedFoldersComboModelTest::SetUp() {
70  profile_.reset(new TestingProfile());
71  profile_->CreateBookmarkModel(true);
72  test::WaitForBookmarkModelToLoad(GetModel());
73}
74
75void RecentlyUsedFoldersComboModelTest::TearDown() {
76  // Flush the message loop to make application verifiers happy.
77  message_loop_.RunUntilIdle();
78}
79
80BookmarkModel* RecentlyUsedFoldersComboModelTest::GetModel() {
81  return BookmarkModelFactory::GetForProfile(profile_.get());
82}
83
84// Verifies there are no duplicate nodes in the model.
85TEST_F(RecentlyUsedFoldersComboModelTest, NoDups) {
86  const BookmarkNode* new_node = GetModel()->AddURL(
87      GetModel()->bookmark_bar_node(), 0, base::ASCIIToUTF16("a"),
88      GURL("http://a"));
89  RecentlyUsedFoldersComboModel model(GetModel(), new_node);
90  std::set<base::string16> items;
91  for (int i = 0; i < model.GetItemCount(); ++i) {
92    if (!model.IsItemSeparatorAt(i))
93      EXPECT_EQ(0u, items.count(model.GetItemAt(i)));
94  }
95}
96
97// Verifies that observers are notified on changes.
98TEST_F(RecentlyUsedFoldersComboModelTest, NotifyObserver) {
99  const BookmarkNode* folder = GetModel()->AddFolder(
100      GetModel()->bookmark_bar_node(), 0, base::ASCIIToUTF16("a"));
101  const BookmarkNode* sub_folder = GetModel()->AddFolder(
102      folder, 0, base::ASCIIToUTF16("b"));
103  const BookmarkNode* new_node = GetModel()->AddURL(
104      sub_folder, 0, base::ASCIIToUTF16("a"), GURL("http://a"));
105  RecentlyUsedFoldersComboModel model(GetModel(), new_node);
106  TestComboboxModelObserver observer;
107  model.AddObserver(&observer);
108
109  const int initial_count = model.GetItemCount();
110  // Remove a folder, it should remove an item from the model too.
111  GetModel()->Remove(folder, folder->GetIndexOf(sub_folder));
112  EXPECT_TRUE(observer.GetAndClearChanged());
113  const int updated_count = model.GetItemCount();
114  EXPECT_LT(updated_count, initial_count);
115
116  // Remove all, which should remove a folder too.
117  GetModel()->RemoveAllUserBookmarks();
118  EXPECT_TRUE(observer.GetAndClearChanged());
119  EXPECT_LT(model.GetItemCount(), updated_count);
120
121  model.RemoveObserver(&observer);
122}
123