Mocker for OSCrypt

components/signin/core/browser/webdata/token_service_table_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 "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
-#include "components/os_crypt/os_crypt.h"
+#include "components/os_crypt/os_crypt_mocker.h"
 #include "components/signin/core/browser/webdata/token_service_table.h"
 #include "components/webdata/common/web_database.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::Time;
 
 class TokenServiceTableTest : public testing::Test {
  public:
   TokenServiceTableTest() {}
   ~TokenServiceTableTest() override {}
 
  protected:
   void SetUp() override {
-#if defined(OS_MACOSX)
-    OSCrypt::UseMockKeychain(true);
-#endif
+    OSCryptMocker::SetUpWithSingleton();
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     file_ = temp_dir_.path().AppendASCII("TestWebDatabase");
 
     table_.reset(new TokenServiceTable);
     db_.reset(new WebDatabase);
     db_->AddTable(table_.get());
     ASSERT_EQ(sql::INIT_OK, db_->Init(file_));
   }
 
+  void TearDown() override { OSCryptMocker::TearDown(); }
+
   base::FilePath file_;
   base::ScopedTempDir temp_dir_;
   std::unique_ptr<TokenServiceTable> table_;
   std::unique_ptr<WebDatabase> db_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TokenServiceTableTest);
 };
 
 TEST_F(TokenServiceTableTest, TokenServiceGetAllRemoveAll) {
@@ skipping 57 matching lines @@
   service = "testservice";
   service2 = "othertestservice";
 
   EXPECT_TRUE(table_->SetTokenForService(service, "pepperoni"));
   EXPECT_TRUE(table_->SetTokenForService(service2, "steak"));
   EXPECT_TRUE(table_->RemoveTokenForService(service));
   EXPECT_TRUE(table_->GetAllTokens(&out_map));
   EXPECT_EQ(0u, out_map.count(service));
   EXPECT_EQ("steak", out_map.find(service2)->second);
 }