Home | History | Annotate | Download | only in update_engine
      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 
     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