| Index: chrome/browser/bookmarks/bookmark_extension_helpers_unittest.cc
|
| diff --git a/chrome/browser/bookmarks/bookmark_extension_helpers_unittest.cc b/chrome/browser/bookmarks/bookmark_extension_helpers_unittest.cc
|
| index 5d0385975d250a1ed2d69a3098316d01e394827a..50850f190f9bc6c3bc147390052be7698c2772d1 100644
|
| --- a/chrome/browser/bookmarks/bookmark_extension_helpers_unittest.cc
|
| +++ b/chrome/browser/bookmarks/bookmark_extension_helpers_unittest.cc
|
| @@ -2,60 +2,64 @@
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| +#include "chrome/browser/bookmarks/bookmark_extension_helpers.h"
|
|
|
| +#include "base/memory/scoped_ptr.h"
|
| #include "base/utf_string_conversions.h"
|
| #include "base/values.h"
|
| -#include "chrome/browser/bookmarks/bookmark_model.h"
|
| #include "chrome/browser/bookmarks/bookmark_extension_api_constants.h"
|
| -#include "chrome/browser/bookmarks/bookmark_extension_helpers.h"
|
| +#include "chrome/browser/bookmarks/bookmark_model.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
|
|
| namespace keys = bookmark_extension_api_constants;
|
|
|
| class ExtensionBookmarksTest : public testing::Test {
|
| public:
|
| - virtual void SetUp() {
|
| + virtual void SetUp() OVERRIDE {
|
| model_.reset(new BookmarkModel(NULL));
|
| model_->AddURL(model_->other_node(), 0, ASCIIToUTF16("Digg"),
|
| GURL("http://www.reddit.com"));
|
| model_->AddURL(model_->other_node(), 0, ASCIIToUTF16("News"),
|
| GURL("http://www.foxnews.com"));
|
| - folder = model_->AddFolder(
|
| + folder_ = model_->AddFolder(
|
| model_->other_node(), 0, ASCIIToUTF16("outer folder"));
|
| - model_->AddFolder(folder, 0, ASCIIToUTF16("inner folder 1"));
|
| - model_->AddFolder(folder, 0, ASCIIToUTF16("inner folder 2"));
|
| - model_->AddURL(folder, 0, ASCIIToUTF16("Digg"), GURL("http://reddit.com"));
|
| - model_->AddURL(folder, 0, ASCIIToUTF16("CNet"), GURL("http://cnet.com"));
|
| + model_->AddFolder(folder_, 0, ASCIIToUTF16("inner folder 1"));
|
| + model_->AddFolder(folder_, 0, ASCIIToUTF16("inner folder 2"));
|
| + model_->AddURL(folder_, 0, ASCIIToUTF16("Digg"), GURL("http://reddit.com"));
|
| + model_->AddURL(folder_, 0, ASCIIToUTF16("CNet"), GURL("http://cnet.com"));
|
| }
|
|
|
| scoped_ptr<BookmarkModel> model_;
|
| - const BookmarkNode* folder;
|
| + const BookmarkNode* folder_;
|
| };
|
| TEST_F(ExtensionBookmarksTest, GetFullTreeFromRoot) {
|
| - DictionaryValue* tree = bookmark_extension_helpers::GetNodeDictionary(
|
| - model_->other_node(),
|
| - true, // Recurse.
|
| - false); // Not only folders.
|
| + scoped_ptr<DictionaryValue> tree(
|
| + bookmark_extension_helpers::GetNodeDictionary(
|
| + model_->other_node(),
|
| + true, // Recurse.
|
| + false)); // Not only folders.
|
| ListValue* children;
|
| tree->GetList(keys::kChildrenKey, &children);
|
| ASSERT_EQ(3U, children->GetSize());
|
| }
|
|
|
| TEST_F(ExtensionBookmarksTest, GetFoldersOnlyFromRoot) {
|
| - DictionaryValue* tree = bookmark_extension_helpers::GetNodeDictionary(
|
| - model_->other_node(),
|
| - true, // Recurse.
|
| - true); // Only folders.
|
| + scoped_ptr<DictionaryValue> tree(
|
| + bookmark_extension_helpers::GetNodeDictionary(
|
| + model_->other_node(),
|
| + true, // Recurse.
|
| + true)); // Only folders.
|
| ListValue* children;
|
| tree->GetList(keys::kChildrenKey, &children);
|
| ASSERT_EQ(1U, children->GetSize());
|
| }
|
|
|
| TEST_F(ExtensionBookmarksTest, GetSubtree) {
|
| - DictionaryValue* tree = bookmark_extension_helpers::GetNodeDictionary(
|
| - folder,
|
| - true, // Recurse.
|
| - false); // Not only folders.
|
| + scoped_ptr<DictionaryValue> tree(
|
| + bookmark_extension_helpers::GetNodeDictionary(
|
| + folder_,
|
| + true, // Recurse.
|
| + false)); // Not only folders.
|
| ListValue* children;
|
| tree->GetList(keys::kChildrenKey, &children);
|
| ASSERT_EQ(4U, children->GetSize());
|
| @@ -67,10 +71,11 @@ TEST_F(ExtensionBookmarksTest, GetSubtree) {
|
| }
|
|
|
| TEST_F(ExtensionBookmarksTest, GetSubtreeFoldersOnly) {
|
| - DictionaryValue* tree = bookmark_extension_helpers::GetNodeDictionary(
|
| - folder,
|
| - true, // Recurse.
|
| - true); // Only folders.
|
| + scoped_ptr<DictionaryValue> tree(
|
| + bookmark_extension_helpers::GetNodeDictionary(
|
| + folder_,
|
| + true, // Recurse.
|
| + true)); // Only folders.
|
| ListValue* children;
|
| tree->GetList(keys::kChildrenKey, &children);
|
| ASSERT_EQ(2U, children->GetSize());
|
|
|