1 //
2 // Copyright (C) 2012 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16
17 // based on pam_google_testrunner.cc
18
19 #include <string>
20
21 #include <xz.h>
22
23 #include <base/at_exit.h>
24 #include <base/command_line.h>
25 #include <base/environment.h>
26 #include <brillo/test_helpers.h>
27 #include <gtest/gtest.h>
28
29 #include "update_engine/common/terminator.h"
30 #include "update_engine/common/test_utils.h"
31 #include "update_engine/payload_generator/xz.h"
32
main(int argc,char ** argv)33 int main(int argc, char **argv) {
34 LOG(INFO) << "started";
35 base::AtExitManager exit_manager;
36 // xz-embedded requires to initialize its CRC-32 table once on startup.
37 xz_crc32_init();
38 // The LZMA SDK-based Xz compressor used in the payload generation requires
39 // this one-time initialization.
40 chromeos_update_engine::XzCompressInit();
41 // TODO(garnold) temporarily cause the unittest binary to exit with status
42 // code 2 upon catching a SIGTERM. This will help diagnose why the unittest
43 // binary is perceived as failing by the buildbot. We should revert it to use
44 // the default exit status of 1. Corresponding reverts are necessary in
45 // terminator_unittest.cc.
46 chromeos_update_engine::Terminator::Init(2);
47 // In Android bsdiff is located in update_engine_unittests, add it to PATH.
48 #ifdef __ANDROID__
49 std::unique_ptr<base::Environment> env(base::Environment::Create());
50 std::string path_env;
51 CHECK(env->GetVar("PATH", &path_env));
52 path_env +=
53 ":" + chromeos_update_engine::test_utils::GetBuildArtifactsPath().value();
54 CHECK(env->SetVar("PATH", path_env));
55 #endif
56 LOG(INFO) << "parsing command line arguments";
57 base::CommandLine::Init(argc, argv);
58 LOG(INFO) << "initializing gtest";
59 SetUpTests(&argc, argv, true);
60 LOG(INFO) << "running unit tests";
61 int test_result = RUN_ALL_TESTS();
62 LOG(INFO) << "unittest return value: " << test_result;
63 return test_result;
64 }
65