1 // Copyright (c) 2011 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 "testing/gtest/include/gtest/gtest.h"
6
7 #include "base/file_util.h"
8 #include "base/message_loop.h"
9 #include "base/path_service.h"
10 #include "base/string16.h"
11 #include "base/string_util.h"
12 #include "base/time.h"
13 #include "base/utf_string_conversions.h"
14 #include "base/i18n/time_formatting.h"
15 #include "base/memory/scoped_temp_dir.h"
16 #include "chrome/browser/bookmarks/bookmark_html_writer.h"
17 #include "chrome/browser/bookmarks/bookmark_model.h"
18 #include "chrome/browser/importer/firefox2_importer.h"
19 #include "chrome/test/testing_browser_process_test.h"
20 #include "chrome/test/testing_profile.h"
21 #include "content/browser/browser_thread.h"
22 #include "grit/generated_resources.h"
23 #include "third_party/skia/include/core/SkBitmap.h"
24 #include "ui/base/l10n/l10n_util.h"
25 #include "ui/gfx/codec/png_codec.h"
26
27 namespace {
28
29 static const int kIconWidth = 16;
30 static const int kIconHeight = 16;
31
MakeTestSkBitmap(int w,int h,SkBitmap * bmp)32 void MakeTestSkBitmap(int w, int h, SkBitmap* bmp) {
33 bmp->setConfig(SkBitmap::kARGB_8888_Config, w, h);
34 bmp->allocPixels();
35
36 uint32_t* src_data = bmp->getAddr32(0, 0);
37 for (int i = 0; i < w * h; i++) {
38 src_data[i] = SkPreMultiplyARGB(i % 255, i % 250, i % 245, i % 240);
39 }
40 }
41
42 } // namespace
43
44 class BookmarkHTMLWriterTest : public TestingBrowserProcessTest {
45 protected:
SetUp()46 virtual void SetUp() {
47 TestingBrowserProcessTest::SetUp();
48 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
49 path_ = temp_dir_.path().AppendASCII("bookmarks.html");
50 }
51
52 // Converts a BookmarkEntry to a string suitable for assertion testing.
BookmarkEntryToString(const ProfileWriter::BookmarkEntry & entry)53 string16 BookmarkEntryToString(
54 const ProfileWriter::BookmarkEntry& entry) {
55 string16 result;
56 result.append(ASCIIToUTF16("on_toolbar="));
57 if (entry.in_toolbar)
58 result.append(ASCIIToUTF16("false"));
59 else
60 result.append(ASCIIToUTF16("true"));
61
62 result.append(ASCIIToUTF16(" url=") + UTF8ToUTF16(entry.url.spec()));
63
64 result.append(ASCIIToUTF16(" path="));
65 for (size_t i = 0; i < entry.path.size(); ++i) {
66 if (i != 0)
67 result.append(ASCIIToUTF16("/"));
68 result.append(entry.path[i]);
69 }
70
71 result.append(ASCIIToUTF16(" title="));
72 result.append(entry.title);
73
74 result.append(ASCIIToUTF16(" time="));
75 result.append(base::TimeFormatFriendlyDateAndTime(entry.creation_time));
76 return result;
77 }
78
79 // Creates a set of bookmark values to a string for assertion testing.
BookmarkValuesToString(bool on_toolbar,const GURL & url,const string16 & title,base::Time creation_time,const string16 & f1,const string16 & f2,const string16 & f3)80 string16 BookmarkValuesToString(bool on_toolbar,
81 const GURL& url,
82 const string16& title,
83 base::Time creation_time,
84 const string16& f1,
85 const string16& f2,
86 const string16& f3) {
87 ProfileWriter::BookmarkEntry entry;
88 entry.in_toolbar = on_toolbar;
89 entry.url = url;
90 // The first path element should always be 'x', as that is what we passed
91 // to the importer.
92 entry.path.push_back(ASCIIToUTF16("x"));
93 if (!f1.empty()) {
94 entry.path.push_back(f1);
95 if (!f2.empty()) {
96 entry.path.push_back(f2);
97 if (!f3.empty())
98 entry.path.push_back(f3);
99 }
100 }
101 entry.title = title;
102 entry.creation_time = creation_time;
103 return BookmarkEntryToString(entry);
104 }
105
AssertBookmarkEntryEquals(const ProfileWriter::BookmarkEntry & entry,bool on_toolbar,const GURL & url,const string16 & title,base::Time creation_time,const string16 & f1,const string16 & f2,const string16 & f3)106 void AssertBookmarkEntryEquals(const ProfileWriter::BookmarkEntry& entry,
107 bool on_toolbar,
108 const GURL& url,
109 const string16& title,
110 base::Time creation_time,
111 const string16& f1,
112 const string16& f2,
113 const string16& f3) {
114 EXPECT_EQ(BookmarkValuesToString(on_toolbar, url, title, creation_time,
115 f1, f2, f3),
116 BookmarkEntryToString(entry));
117 }
118
119 ScopedTempDir temp_dir_;
120 FilePath path_;
121 };
122
123 // Class that will notify message loop when file is written.
124 class BookmarksObserver : public BookmarksExportObserver {
125 public:
BookmarksObserver(MessageLoop * loop)126 explicit BookmarksObserver(MessageLoop* loop) : loop_(loop) {
127 DCHECK(loop);
128 }
129
OnExportFinished()130 virtual void OnExportFinished() {
131 loop_->Quit();
132 }
133
134 private:
135 MessageLoop* loop_;
136 DISALLOW_COPY_AND_ASSIGN(BookmarksObserver);
137 };
138
139 // Tests bookmark_html_writer by populating a BookmarkModel, writing it out by
140 // way of bookmark_html_writer, then using the importer to read it back in.
TEST_F(BookmarkHTMLWriterTest,Test)141 TEST_F(BookmarkHTMLWriterTest, Test) {
142 MessageLoop message_loop;
143 BrowserThread fake_ui_thread(BrowserThread::UI, &message_loop);
144 BrowserThread fake_file_thread(BrowserThread::FILE, &message_loop);
145
146 TestingProfile profile;
147 profile.CreateHistoryService(true, false);
148 profile.BlockUntilHistoryProcessesPendingRequests();
149 profile.CreateFaviconService();
150 profile.CreateBookmarkModel(true);
151 profile.BlockUntilBookmarkModelLoaded();
152 BookmarkModel* model = profile.GetBookmarkModel();
153
154 // Create test PNG representing favicon for url1.
155 SkBitmap bitmap;
156 MakeTestSkBitmap(kIconWidth, kIconHeight, &bitmap);
157 std::vector<unsigned char> icon_data;
158 gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, false, &icon_data);
159
160 // Populate the BookmarkModel. This creates the following bookmark structure:
161 // Bookmarks bar
162 // F1
163 // url1
164 // F2
165 // url2
166 // url3
167 // url4
168 // Other
169 // url1
170 // url2
171 // F3
172 // F4
173 // url1
174 string16 f1_title = ASCIIToUTF16("F\"&;<1\"");
175 string16 f2_title = ASCIIToUTF16("F2");
176 string16 f3_title = ASCIIToUTF16("F 3");
177 string16 f4_title = ASCIIToUTF16("F4");
178 string16 url1_title = ASCIIToUTF16("url 1");
179 string16 url2_title = ASCIIToUTF16("url&2");
180 string16 url3_title = ASCIIToUTF16("url\"3");
181 string16 url4_title = ASCIIToUTF16("url\"&;");
182 GURL url1("http://url1");
183 GURL url1_favicon("http://url1/icon.ico");
184 GURL url2("http://url2");
185 GURL url3("http://url3");
186 GURL url4("javascript:alert(\"Hello!\");");
187 base::Time t1(base::Time::Now());
188 base::Time t2(t1 + base::TimeDelta::FromHours(1));
189 base::Time t3(t1 + base::TimeDelta::FromHours(1));
190 base::Time t4(t1 + base::TimeDelta::FromHours(1));
191 const BookmarkNode* f1 = model->AddFolder(
192 model->GetBookmarkBarNode(), 0, f1_title);
193 model->AddURLWithCreationTime(f1, 0, url1_title, url1, t1);
194 profile.GetHistoryService(Profile::EXPLICIT_ACCESS)->AddPage(url1,
195 history::SOURCE_BROWSED);
196 profile.GetFaviconService(Profile::EXPLICIT_ACCESS)->SetFavicon(url1,
197 url1_favicon, icon_data, history::FAVICON);
198 message_loop.RunAllPending();
199 const BookmarkNode* f2 = model->AddFolder(f1, 1, f2_title);
200 model->AddURLWithCreationTime(f2, 0, url2_title, url2, t2);
201 model->AddURLWithCreationTime(model->GetBookmarkBarNode(),
202 1, url3_title, url3, t3);
203
204 model->AddURLWithCreationTime(model->other_node(), 0, url1_title, url1, t1);
205 model->AddURLWithCreationTime(model->other_node(), 1, url2_title, url2, t2);
206 const BookmarkNode* f3 = model->AddFolder(model->other_node(), 2, f3_title);
207 const BookmarkNode* f4 = model->AddFolder(f3, 0, f4_title);
208 model->AddURLWithCreationTime(f4, 0, url1_title, url1, t1);
209 model->AddURLWithCreationTime(model->GetBookmarkBarNode(), 2, url4_title,
210 url4, t4);
211
212 // Write to a temp file.
213 BookmarksObserver observer(&message_loop);
214 bookmark_html_writer::WriteBookmarks(&profile, path_, &observer);
215 message_loop.Run();
216
217 // Clear favicon so that it would be read from file.
218 std::vector<unsigned char> empty_data;
219 profile.GetFaviconService(Profile::EXPLICIT_ACCESS)->SetFavicon(url1,
220 url1_favicon, empty_data, history::FAVICON);
221 message_loop.RunAllPending();
222
223 // Read the bookmarks back in.
224 std::vector<ProfileWriter::BookmarkEntry> parsed_bookmarks;
225 std::vector<history::ImportedFaviconUsage> favicons;
226 Firefox2Importer::ImportBookmarksFile(path_,
227 std::set<GURL>(),
228 false,
229 ASCIIToUTF16("x"),
230 NULL,
231 &parsed_bookmarks,
232 NULL,
233 &favicons);
234
235 // Check loaded favicon (url1 is represents by 3 separate bookmarks).
236 EXPECT_EQ(3U, favicons.size());
237 for (size_t i = 0; i < favicons.size(); i++) {
238 if (url1_favicon == favicons[i].favicon_url) {
239 EXPECT_EQ(1U, favicons[i].urls.size());
240 std::set<GURL>::const_iterator iter = favicons[i].urls.find(url1);
241 ASSERT_TRUE(iter != favicons[i].urls.end());
242 ASSERT_TRUE(*iter == url1);
243 ASSERT_TRUE(favicons[i].png_data == icon_data);
244 }
245 }
246
247 // Verify we got back what we wrote.
248 ASSERT_EQ(7U, parsed_bookmarks.size());
249 // Windows and ChromeOS builds use Sentence case.
250 string16 bookmark_folder_name =
251 l10n_util::GetStringUTF16(IDS_BOOMARK_BAR_FOLDER_NAME);
252 AssertBookmarkEntryEquals(parsed_bookmarks[0], false, url1, url1_title, t1,
253 bookmark_folder_name, f1_title, string16());
254 AssertBookmarkEntryEquals(parsed_bookmarks[1], false, url2, url2_title, t2,
255 bookmark_folder_name, f1_title, f2_title);
256 AssertBookmarkEntryEquals(parsed_bookmarks[2], false, url3, url3_title, t3,
257 bookmark_folder_name, string16(), string16());
258 AssertBookmarkEntryEquals(parsed_bookmarks[3], false, url4, url4_title, t4,
259 bookmark_folder_name, string16(), string16());
260 AssertBookmarkEntryEquals(parsed_bookmarks[4], false, url1, url1_title, t1,
261 string16(), string16(), string16());
262 AssertBookmarkEntryEquals(parsed_bookmarks[5], false, url2, url2_title, t2,
263 string16(), string16(), string16());
264 AssertBookmarkEntryEquals(parsed_bookmarks[6], false, url1, url1_title, t1,
265 f3_title, f4_title, string16());
266 }
267