Fix Firefox bookmarks import.

chrome/utility/importer/firefox_importer_unittest.cc

Index: chrome/utility/importer/firefox_importer_unittest.cc
diff --git a/chrome/utility/importer/firefox_importer_unittest.cc b/chrome/utility/importer/firefox_importer_unittest.cc
index a7e0a40d33c9b9d1ad18626307a9042032bfcc04..1c2c0e2918321b497442ed482c4a87f106598da0 100644
--- a/chrome/utility/importer/firefox_importer_unittest.cc
+++ b/chrome/utility/importer/firefox_importer_unittest.cc
@@ -8,8 +8,14 @@
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/importer/imported_bookmark_entry.h"
+#include "chrome/common/importer/importer_data_types.h"
+#include "chrome/common/importer/importer_url_row.h"
+#include "chrome/common/importer/mock_importer_bridge.h"
+#include "chrome/utility/importer/firefox_importer.h"
 #include "chrome/utility/importer/firefox_importer_unittest_utils.h"
 #include "chrome/utility/importer/nss_decryptor.h"
+#include "components/favicon_base/favicon_usage_data.h"
 #include "sql/connection.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ -39,11 +45,13 @@ TEST(FirefoxImporterTest, MAYBE_NSS(Firefox3NSS3Decryptor)) {
   ASSERT_TRUE(decryptor_proxy.Setup(nss_path));
 
   ASSERT_TRUE(decryptor_proxy.DecryptorInit(nss_path, db_path));
-  EXPECT_EQ(base::ASCIIToUTF16("hello"),
+  EXPECT_EQ(
+      base::ASCIIToUTF16("hello"),
       decryptor_proxy.Decrypt("MDIEEPgAAAAAAAAAAAAAAAAAAAEwFAYIKoZIhvcNAwcECKa"
                               "jtRg4qFSHBAhv9luFkXgDJA=="));
   // Test UTF-16 encoding.
-  EXPECT_EQ(base::WideToUTF16(L"\x4E2D"),
+  EXPECT_EQ(
+      base::WideToUTF16(L"\x4E2D"),
       decryptor_proxy.Decrypt("MDIEEPgAAAAAAAAAAAAAAAAAAAEwFAYIKoZIhvcNAwcECLW"
                               "qqiccfQHWBAie74hxnULxlw=="));
 
@@ -115,3 +123,76 @@ TEST(FirefoxImporterTest, MAYBE_NSS(FirefoxNSSDecryptorDeduceAuthScheme)) {
   EXPECT_EQ(autofill::PasswordForm::SCHEME_BASIC, forms[0].scheme);
   EXPECT_EQ(autofill::PasswordForm::SCHEME_HTML, forms[1].scheme);
 }
+
+TEST(FirefoxImporterTest, ImportBookmarks) {
+  using ::testing::_;
+  base::FilePath places_path;
+  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &places_path));
+  places_path =
+      places_path.AppendASCII("import").AppendASCII("firefox").AppendASCII(
+          "48.0.2");
+  scoped_refptr<FirefoxImporter> importer = new FirefoxImporter;
+  importer::SourceProfile profile;
+  profile.source_path = places_path;
+  scoped_refptr<MockImporterBridge> bridge = new MockImporterBridge;
+  std::vector<ImportedBookmarkEntry> bookmarks;
+  favicon_base::FaviconUsageDataList favicons;
+  EXPECT_CALL(*bridge, NotifyStarted());
+  EXPECT_CALL(*bridge, NotifyItemStarted(importer::FAVORITES));
+  EXPECT_CALL(*bridge, AddBookmarks(_, _))
+      .WillOnce(::testing::SaveArg<0>(&bookmarks));
+  EXPECT_CALL(*bridge, SetFavicons(_))
+      .WillOnce(::testing::SaveArg<0>(&favicons));
+  EXPECT_CALL(*bridge, NotifyItemEnded(importer::FAVORITES));
+  EXPECT_CALL(*bridge, NotifyEnded());
+  importer->StartImport(profile, importer::FAVORITES, bridge.get());
+  ASSERT_EQ(6u, bookmarks.size());
+  EXPECT_EQ("https://www.mozilla.org/en-US/firefox/central/",
+            bookmarks[0].url.spec());
+  EXPECT_EQ("https://www.mozilla.org/en-US/firefox/help/",
+            bookmarks[1].url.spec());
+  EXPECT_EQ("https://www.mozilla.org/en-US/firefox/customize/",
+            bookmarks[2].url.spec());
+  EXPECT_EQ("https://www.mozilla.org/en-US/contribute/",
+            bookmarks[3].url.spec());
+  EXPECT_EQ("https://www.mozilla.org/en-US/about/", bookmarks[4].url.spec());
+  EXPECT_EQ("https://www.google.com/", bookmarks[5].url.spec());
+  ASSERT_EQ(5u, favicons.size());
+  EXPECT_EQ("http://www.mozilla.org/2005/made-up-favicon/0-1473403921346",
+            favicons[0].favicon_url.spec());
+  EXPECT_EQ("http://www.mozilla.org/2005/made-up-favicon/1-1473403921347",
+            favicons[1].favicon_url.spec());
+  EXPECT_EQ("http://www.mozilla.org/2005/made-up-favicon/2-1473403921348",
+            favicons[2].favicon_url.spec());
+  EXPECT_EQ("http://www.mozilla.org/2005/made-up-favicon/3-1473403921349",
+            favicons[3].favicon_url.spec());
+  EXPECT_EQ("http://www.mozilla.org/2005/made-up-favicon/4-1473403921349",
+            favicons[4].favicon_url.spec());
+}
+
+TEST(FirefoxImporterTest, ImportHistorySchema) {
+  using ::testing::_;
+  base::FilePath places_path;
+  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &places_path));
+  places_path =
+      places_path.AppendASCII("import").AppendASCII("firefox").AppendASCII(
+          "48.0.2");
+  scoped_refptr<FirefoxImporter> ff_importer = new FirefoxImporter;
+  importer::SourceProfile profile;
+  profile.source_path = places_path;
+  scoped_refptr<MockImporterBridge> bridge = new MockImporterBridge;
+  std::vector<ImporterURLRow> history;
+  EXPECT_CALL(*bridge, NotifyStarted());
+  EXPECT_CALL(*bridge, NotifyItemStarted(importer::HISTORY));
+  EXPECT_CALL(*bridge, SetHistoryItems(_, _))
+      .WillOnce(::testing::SaveArg<0>(&history));
+  EXPECT_CALL(*bridge, NotifyItemEnded(importer::HISTORY));
+  EXPECT_CALL(*bridge, NotifyEnded());
+  ff_importer->StartImport(profile, importer::HISTORY, bridge.get());
+  ASSERT_EQ(3u, history.size());
+  EXPECT_EQ("https://www.mozilla.org/en-US/firefox/48.0.2/firstrun/learnmore/",
+            history[0].url.spec());
+  EXPECT_EQ("https://www.mozilla.org/en-US/firefox/48.0.2/firstrun/",
+            history[1].url.spec());
+  EXPECT_EQ("http://google.com/", history[2].url.spec());
+}