Inside Bitcoin

1// Copyright (c) 2018-present 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#include <util/string.h>
6#include <util/threadnames.h>
7
8#include <mutex>
9#include <set>
10#include <string>
11#include <thread>
12#include <vector>
13
14#include <boost/test/unit_test.hpp>
15
16using util::ToString;
17
18BOOST_AUTO_TEST_SUITE(util_threadnames_tests)
19
20const std::string TEST_THREAD_NAME_BASE = "test_thrd.";
21
22/**
23 * Run a bunch of threads to all call util::ThreadRename.
24 *
25 * @return the set of name each thread has after attempted renaming.
26 */
27std::set<std::string> RenameEnMasse(int num_threads)
28{
29    std::vector<std::thread> threads;
30    std::set<std::string> names;
31    std::mutex lock;
32
33    auto RenameThisThread = [&](int i) {
34        util::ThreadRename(TEST_THREAD_NAME_BASE + ToString(i));
35        std::lock_guard<std::mutex> guard(lock);
36        names.insert(util::ThreadGetInternalName());
37    };
38
39    threads.reserve(num_threads);
40    for (int i = 0; i < num_threads; ++i) {
41        threads.emplace_back(RenameThisThread, i);
42    }
43
44    for (std::thread& thread : threads) thread.join();
45
46    return names;
47}
48
49/**
50 * Rename a bunch of threads with the same basename (expect_multiple=true), ensuring suffixes are
51 * applied properly.
52 */
53BOOST_AUTO_TEST_CASE(util_threadnames_test_rename_threaded)
54{
55    std::set<std::string> names = RenameEnMasse(100);
56
57    BOOST_CHECK_EQUAL(names.size(), 100U);
58
59    // Names "test_thrd.[n]" should exist for n = [0, 99]
60    for (int i = 0; i < 100; ++i) {
61        BOOST_CHECK(names.contains(TEST_THREAD_NAME_BASE + ToString(i)));
62    }
63
64}
65
66BOOST_AUTO_TEST_SUITE_END()
67