OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/api/bookmarks/bookmark_api_helpers.h" | 5 #include "chrome/browser/extensions/api/bookmarks/bookmark_api_helpers.h" |
6 | 6 |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "base/strings/utf_string_conversions.h" | 8 #include "base/strings/utf_string_conversions.h" |
9 #include "base/values.h" | 9 #include "base/values.h" |
10 #include "chrome/browser/bookmarks/bookmark_model_factory.h" | 10 #include "chrome/browser/bookmarks/bookmark_model_factory.h" |
11 #include "chrome/browser/bookmarks/chrome_bookmark_client.h" | 11 #include "chrome/browser/bookmarks/chrome_bookmark_client.h" |
12 #include "chrome/browser/extensions/api/bookmarks/bookmark_api_constants.h" | 12 #include "chrome/browser/extensions/api/bookmarks/bookmark_api_constants.h" |
13 #include "chrome/common/extensions/api/bookmarks.h" | 13 #include "chrome/common/extensions/api/bookmarks.h" |
14 #include "chrome/test/base/testing_profile.h" | 14 #include "chrome/test/base/testing_profile.h" |
15 #include "components/bookmarks/browser/bookmark_model.h" | 15 #include "components/bookmarks/browser/bookmark_model.h" |
16 #include "components/bookmarks/test/bookmark_test_helpers.h" | 16 #include "components/bookmarks/test/bookmark_test_helpers.h" |
17 #include "content/public/test/test_browser_thread_bundle.h" | 17 #include "content/public/test/test_browser_thread_bundle.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
19 | 19 |
20 namespace extensions { | 20 namespace extensions { |
21 | 21 |
22 namespace keys = bookmark_api_constants; | 22 namespace keys = bookmark_api_constants; |
23 using api::bookmarks::BookmarkTreeNode; | 23 using api::bookmarks::BookmarkTreeNode; |
24 | 24 |
25 namespace bookmark_api_helpers { | 25 namespace bookmark_api_helpers { |
26 | 26 |
27 class ExtensionBookmarksTest : public testing::Test { | 27 class ExtensionBookmarksTest : public testing::Test { |
28 public: | 28 public: |
29 ExtensionBookmarksTest() : client_(NULL), model_(NULL), folder_(NULL) {} | 29 ExtensionBookmarksTest() |
| 30 : client_(NULL), model_(NULL), node_(NULL), folder_(NULL) {} |
30 | 31 |
31 virtual void SetUp() OVERRIDE { | 32 virtual void SetUp() OVERRIDE { |
32 profile_.CreateBookmarkModel(false); | 33 profile_.CreateBookmarkModel(false); |
33 client_ = | 34 client_ = |
34 BookmarkModelFactory::GetChromeBookmarkClientForProfile(&profile_); | 35 BookmarkModelFactory::GetChromeBookmarkClientForProfile(&profile_); |
35 model_ = client_->model(); | 36 model_ = client_->model(); |
36 test::WaitForBookmarkModelToLoad(model_); | 37 test::WaitForBookmarkModelToLoad(model_); |
37 | 38 |
38 model_->AddURL(model_->other_node(), 0, base::ASCIIToUTF16("Digg"), | 39 node_ = model_->AddURL(model_->other_node(), 0, base::ASCIIToUTF16("Digg"), |
39 GURL("http://www.reddit.com")); | 40 GURL("http://www.reddit.com")); |
40 model_->AddURL(model_->other_node(), 0, base::ASCIIToUTF16("News"), | 41 model_->AddURL(model_->other_node(), 0, base::ASCIIToUTF16("News"), |
41 GURL("http://www.foxnews.com")); | 42 GURL("http://www.foxnews.com")); |
42 folder_ = model_->AddFolder( | 43 folder_ = model_->AddFolder( |
43 model_->other_node(), 0, base::ASCIIToUTF16("outer folder")); | 44 model_->other_node(), 0, base::ASCIIToUTF16("outer folder")); |
44 model_->AddFolder(folder_, 0, base::ASCIIToUTF16("inner folder 1")); | 45 model_->AddFolder(folder_, 0, base::ASCIIToUTF16("inner folder 1")); |
45 model_->AddFolder(folder_, 0, base::ASCIIToUTF16("inner folder 2")); | 46 model_->AddFolder(folder_, 0, base::ASCIIToUTF16("inner folder 2")); |
46 model_->AddURL( | 47 model_->AddURL( |
47 folder_, 0, base::ASCIIToUTF16("Digg"), GURL("http://reddit.com")); | 48 folder_, 0, base::ASCIIToUTF16("Digg"), GURL("http://reddit.com")); |
48 model_->AddURL( | 49 model_->AddURL( |
49 folder_, 0, base::ASCIIToUTF16("CNet"), GURL("http://cnet.com")); | 50 folder_, 0, base::ASCIIToUTF16("CNet"), GURL("http://cnet.com")); |
50 } | 51 } |
51 | 52 |
52 content::TestBrowserThreadBundle thread_bundle_; | 53 content::TestBrowserThreadBundle thread_bundle_; |
53 TestingProfile profile_; | 54 TestingProfile profile_; |
54 ChromeBookmarkClient* client_; | 55 ChromeBookmarkClient* client_; |
55 BookmarkModel* model_; | 56 BookmarkModel* model_; |
| 57 const BookmarkNode* node_; |
56 const BookmarkNode* folder_; | 58 const BookmarkNode* folder_; |
57 }; | 59 }; |
58 | 60 |
59 TEST_F(ExtensionBookmarksTest, GetFullTreeFromRoot) { | 61 TEST_F(ExtensionBookmarksTest, GetFullTreeFromRoot) { |
60 scoped_ptr<BookmarkTreeNode> tree( | 62 scoped_ptr<BookmarkTreeNode> tree( |
61 GetBookmarkTreeNode(model_->other_node(), | 63 GetBookmarkTreeNode(client_, |
| 64 model_->other_node(), |
62 true, // Recurse. | 65 true, // Recurse. |
63 false)); // Not only folders. | 66 false)); // Not only folders. |
64 ASSERT_EQ(3U, tree->children->size()); | 67 ASSERT_EQ(3U, tree->children->size()); |
65 } | 68 } |
66 | 69 |
67 TEST_F(ExtensionBookmarksTest, GetFoldersOnlyFromRoot) { | 70 TEST_F(ExtensionBookmarksTest, GetFoldersOnlyFromRoot) { |
68 scoped_ptr<BookmarkTreeNode> tree( | 71 scoped_ptr<BookmarkTreeNode> tree( |
69 GetBookmarkTreeNode(model_->other_node(), | 72 GetBookmarkTreeNode(client_, |
| 73 model_->other_node(), |
70 true, // Recurse. | 74 true, // Recurse. |
71 true)); // Only folders. | 75 true)); // Only folders. |
72 ASSERT_EQ(1U, tree->children->size()); | 76 ASSERT_EQ(1U, tree->children->size()); |
73 } | 77 } |
74 | 78 |
75 TEST_F(ExtensionBookmarksTest, GetSubtree) { | 79 TEST_F(ExtensionBookmarksTest, GetSubtree) { |
76 scoped_ptr<BookmarkTreeNode> tree( | 80 scoped_ptr<BookmarkTreeNode> tree( |
77 GetBookmarkTreeNode(folder_, | 81 GetBookmarkTreeNode(client_, |
| 82 folder_, |
78 true, // Recurse. | 83 true, // Recurse. |
79 false)); // Not only folders. | 84 false)); // Not only folders. |
80 ASSERT_EQ(4U, tree->children->size()); | 85 ASSERT_EQ(4U, tree->children->size()); |
81 linked_ptr<BookmarkTreeNode> digg = tree->children->at(1); | 86 linked_ptr<BookmarkTreeNode> digg = tree->children->at(1); |
82 ASSERT_TRUE(digg.get()); | 87 ASSERT_TRUE(digg.get()); |
83 ASSERT_EQ("Digg", digg->title); | 88 ASSERT_EQ("Digg", digg->title); |
84 } | 89 } |
85 | 90 |
86 TEST_F(ExtensionBookmarksTest, GetSubtreeFoldersOnly) { | 91 TEST_F(ExtensionBookmarksTest, GetSubtreeFoldersOnly) { |
87 scoped_ptr<BookmarkTreeNode> tree( | 92 scoped_ptr<BookmarkTreeNode> tree( |
88 GetBookmarkTreeNode(folder_, | 93 GetBookmarkTreeNode(client_, |
| 94 folder_, |
89 true, // Recurse. | 95 true, // Recurse. |
90 true)); // Only folders. | 96 true)); // Only folders. |
91 ASSERT_EQ(2U, tree->children->size()); | 97 ASSERT_EQ(2U, tree->children->size()); |
92 linked_ptr<BookmarkTreeNode> inner_folder = tree->children->at(1); | 98 linked_ptr<BookmarkTreeNode> inner_folder = tree->children->at(1); |
93 ASSERT_TRUE(inner_folder.get()); | 99 ASSERT_TRUE(inner_folder.get()); |
94 ASSERT_EQ("inner folder 1", inner_folder->title); | 100 ASSERT_EQ("inner folder 1", inner_folder->title); |
95 } | 101 } |
96 | 102 |
| 103 TEST_F(ExtensionBookmarksTest, GetModifiableNode) { |
| 104 scoped_ptr<BookmarkTreeNode> tree( |
| 105 GetBookmarkTreeNode(client_, |
| 106 node_, |
| 107 false, // Recurse. |
| 108 false)); // Only folders. |
| 109 EXPECT_EQ("Digg", tree->title); |
| 110 ASSERT_TRUE(tree->url); |
| 111 EXPECT_EQ("http://www.reddit.com/", *tree->url); |
| 112 EXPECT_EQ(BookmarkTreeNode::UNMODIFIABLE_NONE, tree->unmodifiable); |
| 113 } |
| 114 |
| 115 TEST_F(ExtensionBookmarksTest, GetManagedNode) { |
| 116 const BookmarkNode* managed_bookmark = |
| 117 model_->AddURL(client_->managed_node(), |
| 118 0, |
| 119 base::ASCIIToUTF16("Chromium"), |
| 120 GURL("http://www.chromium.org/")); |
| 121 scoped_ptr<BookmarkTreeNode> tree( |
| 122 GetBookmarkTreeNode(client_, |
| 123 managed_bookmark, |
| 124 false, // Recurse. |
| 125 false)); // Only folders. |
| 126 EXPECT_EQ("Chromium", tree->title); |
| 127 EXPECT_EQ("http://www.chromium.org/", *tree->url); |
| 128 EXPECT_EQ(BookmarkTreeNode::UNMODIFIABLE_MANAGED, tree->unmodifiable); |
| 129 } |
| 130 |
97 TEST_F(ExtensionBookmarksTest, RemoveNodeInvalidId) { | 131 TEST_F(ExtensionBookmarksTest, RemoveNodeInvalidId) { |
98 int64 invalid_id = model_->next_node_id(); | 132 int64 invalid_id = model_->next_node_id(); |
99 std::string error; | 133 std::string error; |
100 EXPECT_FALSE(RemoveNode(client_, invalid_id, true, &error)); | 134 EXPECT_FALSE(RemoveNode(client_, invalid_id, true, &error)); |
101 EXPECT_EQ(keys::kNoNodeError, error); | 135 EXPECT_EQ(keys::kNoNodeError, error); |
102 } | 136 } |
103 | 137 |
104 TEST_F(ExtensionBookmarksTest, RemoveNodePermanent) { | 138 TEST_F(ExtensionBookmarksTest, RemoveNodePermanent) { |
105 std::string error; | 139 std::string error; |
106 EXPECT_FALSE(RemoveNode(client_, model_->other_node()->id(), true, &error)); | 140 EXPECT_FALSE(RemoveNode(client_, model_->other_node()->id(), true, &error)); |
(...skipping 19 matching lines...) Expand all Loading... |
126 | 160 |
127 TEST_F(ExtensionBookmarksTest, RemoveNodeRecursive) { | 161 TEST_F(ExtensionBookmarksTest, RemoveNodeRecursive) { |
128 EXPECT_EQ(3, model_->other_node()->child_count()); | 162 EXPECT_EQ(3, model_->other_node()->child_count()); |
129 std::string error; | 163 std::string error; |
130 EXPECT_TRUE(RemoveNode(client_, folder_->id(), true, &error)); | 164 EXPECT_TRUE(RemoveNode(client_, folder_->id(), true, &error)); |
131 EXPECT_EQ(2, model_->other_node()->child_count()); | 165 EXPECT_EQ(2, model_->other_node()->child_count()); |
132 } | 166 } |
133 | 167 |
134 } // namespace bookmark_api_helpers | 168 } // namespace bookmark_api_helpers |
135 } // namespace extensions | 169 } // namespace extensions |
OLD | NEW |