Enable import/export of passwords into/from Password Manager

chrome/browser/ui/webui/options/password_manager_handler_unittest.cc

+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "base/metrics/histogram.h"
+#include "base/metrics/statistics_recorder.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/password_manager/mock_password_store_service.h"
+#include "chrome/browser/password_manager/password_store_factory.h"
+#include "chrome/browser/ui/passwords/password_manager_presenter.h"
+#include "chrome/browser/ui/webui/options/password_manager_handler.h"
+#include "chrome/test/base/testing_profile.h"
+#include "components/password_manager/core/browser/mock_password_store.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class MockPasswordManagerPresenter : public PasswordManagerPresenter {
+ public:
+  MockPasswordManagerPresenter(PasswordUIView* handler)
+      : PasswordManagerPresenter(handler), handler_(handler) {}
+  ~MockPasswordManagerPresenter() override {}
+
+  MOCK_METHOD1(AddPasswordsToStore,
+               bool(const std::vector<autofill::PasswordForm>&));
+  MOCK_METHOD0(RequestToExportPassword, bool());
+
+ private:
+  PasswordUIView* handler_;
+  DISALLOW_COPY_AND_ASSIGN(MockPasswordManagerPresenter);
+};
+
+class DummyPasswordManagerHandler : public PasswordUIView {
+ public:
+  explicit DummyPasswordManagerHandler(Profile* profile)
+      : profile_(profile), password_manager_presenter_(this) {
+    password_manager_presenter_.Initialize();
+  }
+  ~DummyPasswordManagerHandler() override {}
+  Profile* GetProfile() override;
+
+  void ShowPassword(size_t,
+                    const std::string&,
+                    const std::string&,
+                    const base::string16&) override {}
+  void SetPasswordList(const ScopedVector<autofill::PasswordForm>&,
+                       bool) override {}
+  void SetPasswordExceptionList(
+      const ScopedVector<autofill::PasswordForm>&) override {}
+
+#if !defined(OS_ANDROID)
+  gfx::NativeWindow GetNativeWindow() const override;
+#endif
+ private:
+  Profile* profile_;
+  PasswordManagerPresenter password_manager_presenter_;
+  DISALLOW_COPY_AND_ASSIGN(DummyPasswordManagerHandler);
+};
+
+#if !defined(OS_ANDROID)
+gfx::NativeWindow DummyPasswordManagerHandler::GetNativeWindow() const {
+  return NULL;
+}
+#endif
+Profile* DummyPasswordManagerHandler::GetProfile() {
+  return profile_;
+}
+
+class PasswordManagerHandlerTest : public testing::Test {
+ protected:
+  PasswordManagerHandlerTest() {}
+  ~PasswordManagerHandlerTest() override {
+    delete handler_;
+    delete dummy_handler_;
+  }
+
+  void SetUp() override {
+    UMA_HISTOGRAM_COUNTS(
+        "PasswordManager.ImportedPasswordsPerUserInCSV.Success", 0);
+    PasswordStoreFactory::GetInstance()->SetTestingFactory(
+        &profile_, MockPasswordStoreService::Build);
+    dummy_handler_ = new DummyPasswordManagerHandler(&profile_);
+    handler_ = new options::PasswordManagerHandler(
+        new MockPasswordManagerPresenter(dummy_handler_));
+  }
+
+  MockPasswordManagerPresenter* GetPresenter() {
+    return static_cast<MockPasswordManagerPresenter*>(
+        handler_->password_manager_presenter_.get());
+  }
+
+  void ExportPassword() { handler_->HandlePasswordExport(NULL); }
+
+  void ImportPassword() {
+    autofill::PasswordForm form(MakeForm());
+    std::vector<autofill::PasswordForm>* passwordForms =
+        new std::vector<autofill::PasswordForm>;
+    passwordForms->push_back(form);
+    passwordForms->push_back(form);
+    handler_->ImportPasswordFileRead(
+        password_manager::PasswordImporter::SUCCESS, *passwordForms);
+  }
+
+  autofill::PasswordForm MakeForm() {
+    autofill::PasswordForm form;
+    GURL origin("http://test.com");
+    form.origin = origin;
+    form.username_element = base::ASCIIToUTF16("Email");
+    form.username_value = base::ASCIIToUTF16("test@test.com");
+    form.password_element = base::ASCIIToUTF16("Password");
+    form.password_value = base::ASCIIToUTF16("abcdef");
+    return form;
+  }
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+  TestingProfile profile_;
+  options::PasswordManagerHandler* handler_;
+  DummyPasswordManagerHandler* dummy_handler_;
+
+  DISALLOW_COPY_AND_ASSIGN(PasswordManagerHandlerTest);
+};
+
+MATCHER_P(FormMatches, form, "") {
+  return arg.size() == 2 && form.signon_realm == arg[0].signon_realm &&
+         form.origin == arg[0].origin && form.action == arg[0].action &&
+         form.username_element == arg[0].username_element &&
+         form.password_element == arg[0].password_element &&
+         form.new_password_element == arg[0].new_password_element &&
+         form.password_autocomplete_set == arg[0].password_autocomplete_set &&
+         form.submit_element == arg[0].submit_element;
+}
+
+TEST_F(PasswordManagerHandlerTest, PasswordImport) {
+  autofill::PasswordForm form(MakeForm());
+  base::HistogramBase* exported_passwords =
+      base::StatisticsRecorder::FindHistogram(
+          "PasswordManager.ImportedPasswordsPerUserInCSV.Success");
+  scoped_ptr<base::HistogramSamples> samples(
+      exported_passwords->SnapshotSamples());
+  EXPECT_CALL(*GetPresenter(), AddPasswordsToStore(FormMatches(form)));
+  ImportPassword();
+  EXPECT_EQ(samples->TotalCount(), 1);
+}
+
+TEST_F(PasswordManagerHandlerTest, PasswordExport) {
+  EXPECT_CALL(*GetPresenter(), RequestToExportPassword());
+  ExportPassword();
+}