Branch data     Line data    Source code

       1                 :            : // Copyright (c) 2011-2022 The Bitcoin Core developers
       2                 :            : // Distributed under the MIT software license, see the accompanying
       3                 :            : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4                 :            : 
       5                 :            : /**
       6                 :            :  * See https://www.boost.org/doc/libs/1_78_0/libs/test/doc/html/boost_test/adv_scenarios/single_header_customizations/multiple_translation_units.html
       7                 :            :  */
       8                 :            : #define BOOST_TEST_MODULE Bitcoin Core Test Suite
       9                 :            : 
      10                 :            : #include <boost/test/included/unit_test.hpp>
      11                 :            : 
      12                 :            : #include <test/util/setup_common.h>
      13                 :            : 
      14                 :            : #include <functional>
      15                 :            : #include <iostream>
      16                 :            : 
      17                 :            : /** Redirect debug log to unit_test.log files */
      18                 :          0 : const std::function<void(const std::string&)> G_TEST_LOG_FUN = [](const std::string& s) {
      19                 :          0 :     static const bool should_log{std::any_of(
      20                 :          0 :         &boost::unit_test::framework::master_test_suite().argv[1],
      21                 :          0 :         &boost::unit_test::framework::master_test_suite().argv[boost::unit_test::framework::master_test_suite().argc],
      22                 :          0 :         [](const char* arg) {
      23                 :          0 :             return std::string{"DEBUG_LOG_OUT"} == arg;
      24                 :          0 :         })};
      25                 :          0 :     if (!should_log) return;
      26                 :          0 :     std::cout << s;
      27                 :          0 : };
      28                 :            : 
      29                 :            : /**
      30                 :            :  * Retrieve the command line arguments from boost.
      31                 :            :  * Allows usage like:
      32                 :            :  * `test_bitcoin --run_test="net_tests/cnode_listen_port" -- -checkaddrman=1 -printtoconsole=1`
      33                 :            :  * which would return `["-checkaddrman=1", "-printtoconsole=1"]`.
      34                 :            :  */
      35                 :          0 : const std::function<std::vector<const char*>()> G_TEST_COMMAND_LINE_ARGUMENTS = []() {
      36                 :          0 :     std::vector<const char*> args;
      37                 :          0 :     for (int i = 1; i < boost::unit_test::framework::master_test_suite().argc; ++i) {
      38                 :          0 :         args.push_back(boost::unit_test::framework::master_test_suite().argv[i]);
      39                 :          0 :     }
      40                 :          0 :     return args;
      41                 :          0 : };