Mocker for OSCrypt

components/password_manager/core/browser/login_database_unittest.cc

 // Copyright 2014 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 "components/password_manager/core/browser/login_database.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <memory>
+#include <utility>
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/scoped_vector.h"
 #include "base/path_service.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/histogram_tester.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
 #include "components/autofill/core/common/password_form.h"
+#include "components/os_crypt/os_crypt_mocker.h"
 #include "components/password_manager/core/browser/psl_matching_helper.h"
 #include "sql/connection.h"
 #include "sql/statement.h"
 #include "sql/test/test_helpers.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/origin.h"
 
-#if defined(OS_MACOSX)
-#include "components/os_crypt/os_crypt.h"
-#endif
-
 using autofill::PasswordForm;
 using base::ASCIIToUTF16;
 using ::testing::Eq;
 
 namespace password_manager {
 namespace {
 PasswordStoreChangeList AddChangeForForm(const PasswordForm& form) {
   return PasswordStoreChangeList(
       1, PasswordStoreChange(PasswordStoreChange::ADD, form));
 }
@@ skipping 31 matching lines @@
   static const bool is_specialized = false;
 };
 
 template<class T> T GetFirstColumn(const sql::Statement& s) {
   static_assert(must_be_specialized<T>::is_specialized,
                 "Implement a specialization.");
 }
 
 template<> int64_t GetFirstColumn(const sql::Statement& s) {
   return s.ColumnInt64(0);
-};
+}
 
 template<> std::string GetFirstColumn(const sql::Statement& s) {
   return s.ColumnString(0);
-};
+}
 
 bool AddZeroClickableLogin(LoginDatabase* db,
                            const std::string& unique_string) {
   // Example password form.
   PasswordForm form;
   form.origin = GURL("https://example.com/");
   form.username_element = ASCIIToUTF16(unique_string);
   form.username_value = ASCIIToUTF16(unique_string);
   form.password_element = ASCIIToUTF16(unique_string);
   form.submit_element = ASCIIToUTF16("signIn");
@@ skipping 12 matching lines @@
 
 // Serialization routines for vectors implemented in login_database.cc.
 base::Pickle SerializeVector(const std::vector<base::string16>& vec);
 std::vector<base::string16> DeserializeVector(const base::Pickle& pickle);
 
 class LoginDatabaseTest : public testing::Test {
  protected:
   void SetUp() override {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     file_ = temp_dir_.path().AppendASCII("TestMetadataStoreMacDatabase");
-#if defined(OS_MACOSX)
-    OSCrypt::UseMockKeychain(true);
-#endif  // defined(OS_MACOSX)
+    OSCryptMocker::SetUpWithSingleton();
 
     db_.reset(new LoginDatabase(file_));
     ASSERT_TRUE(db_->Init());
   }
 
+  void TearDown() override { OSCryptMocker::TearDown(); }
+
   LoginDatabase& db() { return *db_; }
 
   void TestNonHTMLFormPSLMatching(const PasswordForm::Scheme& scheme) {
     ScopedVector<autofill::PasswordForm> result;
 
     base::Time now = base::Time::Now();
 
     // Simple non-html auth form.
     PasswordForm non_html_auth;
     non_html_auth.origin = GURL("http://example.com");
@@ skipping 1355 matching lines @@
 // Test the migration from GetParam() version to kCurrentVersionNumber.
 class LoginDatabaseMigrationTest : public testing::TestWithParam<int> {
  protected:
   void SetUp() override {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     database_dump_location_ = database_dump_location_.AppendASCII("components")
                                   .AppendASCII("test")
                                   .AppendASCII("data")
                                   .AppendASCII("password_manager");
     database_path_ = temp_dir_.path().AppendASCII("test.db");
-#if defined(OS_MACOSX)
-    OSCrypt::UseMockKeychain(true);
-#endif  // defined(OS_MACOSX)
+    OSCryptMocker::SetUpWithSingleton();
   }
 
+  void TearDown() override { OSCryptMocker::TearDown(); }
+
   // Creates the databse from |sql_file|.
   void CreateDatabase(base::StringPiece sql_file) {
     base::FilePath database_dump;
     ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &database_dump));
     database_dump =
         database_dump.Append(database_dump_location_).AppendASCII(sql_file);
     ASSERT_TRUE(
         sql::test::CreateDatabaseFromSQL(database_path_, database_dump));
   }
 
@@ skipping 157 matching lines @@
                         LoginDatabaseMigrationTest,
                         testing::Range(1, kCurrentVersionNumber + 1));
 INSTANTIATE_TEST_CASE_P(MigrationToVCurrent,
                         LoginDatabaseMigrationTestV9,
                         testing::Values(9));
 INSTANTIATE_TEST_CASE_P(MigrationToVCurrent,
                         LoginDatabaseMigrationTestBroken,
                         testing::Range(1, 4));
 
 }  // namespace password_manager