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()9FileTypePoliciesTestOverlay::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()18FileTypePoliciesTestOverlay::~FileTypePoliciesTestOverlay() { 19 SwapConfig(orig_config_); 20 } 21 SwapConfig(std::unique_ptr<DownloadFileTypeConfig> & new_config) const22void FileTypePoliciesTestOverlay::SwapConfig( 23 std::unique_ptr<DownloadFileTypeConfig>& new_config) const { 24 FileTypePolicies::GetInstance()->SwapConfig(new_config); 25 } 26 27 std::unique_ptr<DownloadFileTypeConfig> DuplicateConfig() const28FileTypePoliciesTestOverlay::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