1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61

base / test / scoped_path_override.cc [blame]

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

#include "base/test/scoped_path_override.h"

#include <ostream>

#include "base/check.h"
#include "base/path_service.h"

namespace base {

ScopedPathOverride::ScopedPathOverride(int key) : key_(key) {
  SaveOriginal();
  bool result = temp_dir_.CreateUniqueTempDir();
  CHECK(result);
  result = PathService::Override(key, temp_dir_.GetPath());
  CHECK(result);
}

ScopedPathOverride::ScopedPathOverride(int key, const base::FilePath& dir)
    : key_(key) {
  SaveOriginal();
  bool result = PathService::Override(key, dir);
  CHECK(result);
}

ScopedPathOverride::ScopedPathOverride(int key,
                                       const FilePath& path,
                                       bool is_absolute,
                                       bool create)
    : key_(key) {
  SaveOriginal();
  bool result =
      PathService::OverrideAndCreateIfNeeded(key, path, is_absolute, create);
  CHECK(result);
}

void ScopedPathOverride::SaveOriginal() {
  if (PathService::IsOverriddenForTesting(key_)) {
    original_override_ = PathService::CheckedGet(key_);
  }
}

ScopedPathOverride::~ScopedPathOverride() {
  bool result = PathService::RemoveOverrideForTests(key_);
  CHECK(result) << "The override seems to have been removed already!";
  if (original_override_) {
    // PathService::Override, by default, does some (blocking) checks to ensure
    // that the path is absolute and exists. As the original override must have
    // already gone through these checks, we can skip these checks here.
    // This is needed for some tests which use ScopedPathOverride in scopes that
    // disallow blocking.
    result = PathService::OverrideAndCreateIfNeeded(
        key_, *original_override_, /*is_absolute=*/true, /*create=*/false);
    CHECK(result);
  }
}

}  // namespace base