1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "components/safe_browsing/core/file_type_policies_test_util.h"
6 
7 namespace safe_browsing {
8 
FileTypePoliciesTestOverlay()9 FileTypePoliciesTestOverlay::FileTypePoliciesTestOverlay()
10     : orig_config_(new DownloadFileTypeConfig()) {
11   // Make a copy of the global config so we can put it back later.
12   // Swap out, copy, swap back in.
13   SwapConfig(orig_config_);
14   std::unique_ptr<DownloadFileTypeConfig> copy_config = DuplicateConfig();
15   SwapConfig(copy_config);
16 }
17 
~FileTypePoliciesTestOverlay()18 FileTypePoliciesTestOverlay::~FileTypePoliciesTestOverlay() {
19   SwapConfig(orig_config_);
20 }
21 
SwapConfig(std::unique_ptr<DownloadFileTypeConfig> & new_config) const22 void FileTypePoliciesTestOverlay::SwapConfig(
23     std::unique_ptr<DownloadFileTypeConfig>& new_config) const {
24   FileTypePolicies::GetInstance()->SwapConfig(new_config);
25 }
26 
27 std::unique_ptr<DownloadFileTypeConfig>
DuplicateConfig() const28 FileTypePoliciesTestOverlay::DuplicateConfig() const {
29   std::unique_ptr<DownloadFileTypeConfig> new_config(
30       new DownloadFileTypeConfig());
31   // Deep copy
32   new_config->CopyFrom(*orig_config_);
33   return new_config;
34 }
35 
36 }  // namespace safe_browsing
37