Enable import/export of passwords into/from Password Manager

components/password_manager/core/browser/import/password_importer_unittest.cc

+// Copyright 2014 The Chromium Authors. All rights reserved.

[vabr (Chromium), 2016/03/10 10:53:32]

2016

[xunlu, 2016/03/16 07:23:59]

Done.

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/password_manager/core/browser/import/password_importer.h"
+
+#include "base/bind.h"
+#include "base/bind_helpers.h"
+#include "base/files/file_util.h"
+#include "base/memory/ref_counted.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/test/test_simple_task_runner.h"
+#include "base/thread_task_runner_handle.h"
+#include "components/autofill/core/common/password_form.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace password_manager {
+
+namespace {
+
+const char kTestOriginURL[] = "http://accounts.google.com/a/LoginAuth";
+const char kTestSignonRealm[] = "http://accounts.google.com/";
+const char kTestUsername[] = "test@gmail.com";
+const char kTestPassword[] = "test1";
+
+// Creates a test input file in a temporary directory with the specified

[vabr (Chromium), 2016/03/10 10:53:32]

Consider using a ScopedTempDir to keep your testing file(s) instead. That would
spare you defining the clean-up yourself.

[xunlu, 2016/03/16 07:23:59]

Done.

+// contents and extension; and cleans it up when the object is destroyed.
+class ScopedInputFile {
+ public:
+  ScopedInputFile() {}
+  ~ScopedInputFile() { base::DeleteFile(input_file_path_.DirName(), true); }
+
+  void Create(const base::FilePath::CharType* extension,
+              base::StringPiece contents) {
+    base::FilePath temporary_dir;
+    ASSERT_TRUE(base::CreateNewTempDirectory(base::FilePath::StringType(),
+                                             &temporary_dir));
+    input_file_path_ =
+        temporary_dir.AppendASCII("passwords").AddExtension(extension);
+    ASSERT_TRUE(
+        base::WriteFile(input_file_path_, contents.data(), contents.size()));
+  }
+
+  base::FilePath path() { return input_file_path_; }
+
+ private:
+  base::FilePath input_file_path_;
+
+  DISALLOW_COPY_AND_ASSIGN(ScopedInputFile);
+};
+
+}  // namespace
+
+class PasswordImporterTest : public testing::Test {
+ public:
+  PasswordImporterTest()
+      : callback_called_(false), result_(PasswordImporter::SUCCESS) {}
+
+ protected:
+  void StartImportAndWaitForCompletion(const base::FilePath& input_file) {
+    PasswordImporter::Import(input_file, message_loop_.task_runner(),
+                             base::Bind(&PasswordImporterTest::OnImportFinished,
+                                        base::Unretained(this)));
+
+    base::RunLoop run_loop;
+    run_loop.RunUntilIdle();
+
+    ASSERT_TRUE(callback_called_);
+  }
+
+  void OnImportFinished(PasswordImporter::Result result,
+                        const std::vector<autofill::PasswordForm>& passwords) {
+    callback_called_ = true;
+    result_ = result;
+    imported_passwords_ = passwords;
+  }
+
+  const PasswordImporter::Result& result() { return result_; }
+  const std::vector<autofill::PasswordForm>& imported_passwords() {
+    return imported_passwords_;
+  }
+
+ private:
+  base::MessageLoop message_loop_;
+
+  bool callback_called_;
+  PasswordImporter::Result result_;
+  std::vector<autofill::PasswordForm> imported_passwords_;
+
+  DISALLOW_COPY_AND_ASSIGN(PasswordImporterTest);
+};
+
+TEST_F(PasswordImporterTest, CSVImport) {
+  const char kTestCSVInput[] =
+      "Url,Username,Password\n"
+      "http://accounts.google.com/a/LoginAuth,test@gmail.com,test1\n";
+
+  ScopedInputFile input_file;
+  ASSERT_NO_FATAL_FAILURE(
+      input_file.Create(FILE_PATH_LITERAL(".csv"), kTestCSVInput));
+
+  ASSERT_NO_FATAL_FAILURE(StartImportAndWaitForCompletion(input_file.path()));
+
+  EXPECT_EQ(PasswordImporter::SUCCESS, result());
+  ASSERT_EQ(1u, imported_passwords().size());
+  EXPECT_EQ(GURL(kTestOriginURL), imported_passwords()[0].origin);
+  EXPECT_EQ(kTestSignonRealm, imported_passwords()[0].signon_realm);
+  EXPECT_EQ(base::ASCIIToUTF16(kTestUsername),
+            imported_passwords()[0].username_value);
+  EXPECT_EQ(base::ASCIIToUTF16(kTestPassword),
+            imported_passwords()[0].password_value);
+}
+
+TEST_F(PasswordImporterTest, ImportIOErrorDueToUnreadableFile) {
+  base::FilePath non_existent_input_file(FILE_PATH_LITERAL("nonexistent.csv"));
+  ASSERT_NO_FATAL_FAILURE(
+      StartImportAndWaitForCompletion(non_existent_input_file));
+
+  EXPECT_EQ(PasswordImporter::IO_ERROR, result());
+  ASSERT_EQ(0u, imported_passwords().size());
+}
+
+}  // namespace password_manager