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 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 } 43