rlz/test/rlz_unittest_main.cc

   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 // Main entry point for all unit tests.
   6
   7 #include "base/at_exit.h"
   8 #include "base/command_line.h"
   9 #include "rlz/lib/rlz_lib.h"
  10 #include "testing/gmock/include/gmock/gmock.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 #if defined(OS_POSIX)
  14 #include "base/files/file_path.h"
  15 #include "base/files/scoped_temp_dir.h"
  16 #include "rlz/lib/rlz_value_store.h"
  17 #endif
  18
  19 int main(int argc, char **argv) {
  20   base::AtExitManager at_exit;
  21   base::CommandLine::Init(argc, argv);
  22
  23   testing::InitGoogleMock(&argc, argv);
  24   testing::InitGoogleTest(&argc, argv);
  25
  26   int ret = RUN_ALL_TESTS();
  27   if (ret == 0) {
  28     // Now re-run all the tests using a supplementary brand code.  This brand
  29     // code will remain in effect for the lifetime of the branding object.
  30 #if defined(OS_POSIX)
  31     // Set a temporary directory for RLZ here, because SupplementaryBranding
  32     // creates and owns RlzValueStore object for its lifetime.
  33     base::ScopedTempDir temp_dir;
  34     if (temp_dir.CreateUniqueTempDir())
  35       rlz_lib::testing::SetRlzStoreDirectory(temp_dir.path());
  36 #endif
  37     rlz_lib::SupplementaryBranding branding("TEST");
  38     ret = RUN_ALL_TESTS();
  39   }
  40
  41   return ret;
  42 }