chrome/browser/extensions/scoped_database_manager_for_test.cc

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/extensions/scoped_database_manager_for_test.h"

#include "chrome/browser/extensions/blocklist.h"
#include "components/safe_browsing/core/browser/db/database_manager.h"

using safe_browsing::SafeBrowsingDatabaseManager;

namespace extensions {

ScopedDatabaseManagerForTest::ScopedDatabaseManagerForTest(
    scoped_refptr<SafeBrowsingDatabaseManager> database_manager)
    : original_(Blocklist::GetDatabaseManager()) {
  Blocklist::SetDatabaseManager(database_manager);
}

ScopedDatabaseManagerForTest::~ScopedDatabaseManagerForTest() {
  Blocklist::SetDatabaseManager(original_);
}

}  // namespace extensions