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