rlz/test/rlz_test_helpers.h

   1 // Copyright (c) 2012 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 // Helper functions used by the tests.
   6
   7 #ifndef RLZ_TEST_RLZ_TEST_HELPERS_H
   8 #define RLZ_TEST_RLZ_TEST_HELPERS_H
   9
  10 #include "base/compiler_specific.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 #if defined(OS_POSIX)
  14 #include "base/files/scoped_temp_dir.h"
  15 #endif
  16
  17 #if defined(OS_WIN)
  18 #include "base/test/test_reg_util_win.h"
  19 #endif
  20
  21 class RlzLibTestNoMachineState : public ::testing::Test {
  22  protected:
  23   virtual void SetUp() OVERRIDE;
  24   virtual void TearDown() OVERRIDE;
  25
  26 #if defined(OS_POSIX)
  27   base::ScopedTempDir temp_dir_;
  28 #endif
  29
  30 #if defined(OS_WIN)
  31   registry_util::RegistryOverrideManager override_manager_;
  32 #endif
  33 };
  34
  35 class RlzLibTestBase : public RlzLibTestNoMachineState {
  36  protected:
  37   virtual void SetUp() OVERRIDE;
  38 };
  39
  40 #endif  // RLZ_TEST_RLZ_TEST_HELPERS_H