This source file includes following definitions.
- SetUp
- GetModel
- TearDown
- TEST_F
- TEST_F
#include "chrome/browser/bookmarks/bookmark_expanded_state_tracker.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/bookmarks/bookmark_model.h"
#include "chrome/browser/bookmarks/bookmark_model_factory.h"
#include "chrome/browser/bookmarks/bookmark_test_helpers.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "testing/gtest/include/gtest/gtest.h"
class BookmarkExpandedStateTrackerTest : public testing::Test {
public:
BookmarkExpandedStateTrackerTest();
virtual void SetUp() OVERRIDE;
virtual void TearDown() OVERRIDE;
protected:
BookmarkModel* GetModel();
private:
scoped_ptr<TestingProfile> profile_;
content::TestBrowserThreadBundle thread_bundle_;
DISALLOW_COPY_AND_ASSIGN(BookmarkExpandedStateTrackerTest);
};
BookmarkExpandedStateTrackerTest::BookmarkExpandedStateTrackerTest() {}
void BookmarkExpandedStateTrackerTest::SetUp() {
profile_.reset(new TestingProfile);
profile_->CreateBookmarkModel(true);
test::WaitForBookmarkModelToLoad(GetModel());
}
BookmarkModel* BookmarkExpandedStateTrackerTest::GetModel() {
return BookmarkModelFactory::GetForProfile(profile_.get());
}
void BookmarkExpandedStateTrackerTest::TearDown() {
profile_.reset(NULL);
}
TEST_F(BookmarkExpandedStateTrackerTest, SetExpandedNodes) {
BookmarkModel* model = GetModel();
BookmarkExpandedStateTracker* tracker = model->expanded_state_tracker();
EXPECT_TRUE(tracker->GetExpandedNodes().empty());
BookmarkExpandedStateTracker::Nodes nodes;
nodes.insert(model->bookmark_bar_node());
tracker->SetExpandedNodes(nodes);
EXPECT_EQ(nodes, tracker->GetExpandedNodes());
const BookmarkNode* n1 = model->AddFolder(model->bookmark_bar_node(), 0,
base::ASCIIToUTF16("x"));
nodes.insert(n1);
tracker->SetExpandedNodes(nodes);
EXPECT_EQ(nodes, tracker->GetExpandedNodes());
model->Remove(model->bookmark_bar_node(), 0);
nodes.erase(n1);
n1 = NULL;
EXPECT_EQ(nodes, tracker->GetExpandedNodes());
}
TEST_F(BookmarkExpandedStateTrackerTest, RemoveAll) {
BookmarkModel* model = GetModel();
BookmarkExpandedStateTracker* tracker = model->expanded_state_tracker();
const BookmarkNode* n1 =
model->AddFolder(model->bookmark_bar_node(), 0, base::ASCIIToUTF16("x"));
BookmarkExpandedStateTracker::Nodes nodes;
nodes.insert(n1);
tracker->SetExpandedNodes(nodes);
EXPECT_EQ(nodes, tracker->GetExpandedNodes());
model->RemoveAll();
EXPECT_TRUE(tracker->GetExpandedNodes().empty());
}