File: fork_and_run.h

package info (click to toggle)
process-cpp 3.0.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 632 kB
  • sloc: cpp: 2,545; ansic: 228; makefile: 17
file content (115 lines) | stat: -rw-r--r-- 5,171 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*
 * Copyright © 2012-2013 Canonical Ltd.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Authored by: Thomas Voß <thomas.voss@canonical.com>
 */
#ifndef CORE_TESTING_FORK_AND_RUN_H_
#define CORE_TESTING_FORK_AND_RUN_H_

#include <core/posix/exit.h>
#include <core/posix/fork.h>
#include <core/posix/visibility.h>

#include <functional>

namespace core
{
namespace testing
{
/**
 * @brief The ForkAndRunResult enum models the different failure modes of fork_and_run.
 */
enum class ForkAndRunResult
{
    empty = 0, ///< Special value indicating no bit being set.
    client_failed = 1 << 0, ///< The client failed.
    service_failed = 1 << 1 ///< The service failed.
};

CORE_POSIX_DLL_PUBLIC ForkAndRunResult operator|(ForkAndRunResult lhs, ForkAndRunResult rhs);
CORE_POSIX_DLL_PUBLIC ForkAndRunResult operator&(ForkAndRunResult lhs, ForkAndRunResult rhs);

/**
 * @brief Forks two processes for both the service and the client.
 *
 * The function does the following:
 *   - Forks a process for the service and runs the respective closure.
 *   - Forks a process for the client and runs the respective closure.
 *   - After the client has finished, the service is signalled with sigterm.
 *
 * @throw std::system_error if an error occurred during process interaction.
 * @throw std::runtime_error for signalling all other error conditions.
 * @param [in] service The service to be executed in a child process.
 * @param [in] client The client to be executed in a child process.
 * @return ForkAndRunResult indicating if either of service or client failed.
 */
CORE_POSIX_DLL_PUBLIC ForkAndRunResult fork_and_run(
        const std::function<core::posix::exit::Status()>& service,
        const std::function<core::posix::exit::Status()>& client);
}
}

/**
 * Test definition macro which runs a TEST in a forked process.
 * Note that you can only use EXPECT_*, not
 * ASSERT_*!
 *
 * Usage:
 * TESTP(test_suite, test_name, {
 *      test code ...
 *      EXPECT_* ...
 * })
 */
#define TESTP(test_suite, test_name, CODE)                                             \
    TEST(test_suite, test_name) {                                                       \
        auto test = [&]() {                                                             \
            CODE                                                                        \
            return HasFailure() ? core::posix::exit::Status::failure                    \
                            : core::posix::exit::Status::success;                       \
        };                                                                              \
        auto child = core::posix::fork(                                                 \
            test,                                                                       \
            core::posix::StandardStream::empty);                                        \
        auto result = child.wait_for(core::posix::wait::Flags::untraced);               \
        EXPECT_EQ(core::posix::wait::Result::Status::exited, result.status);            \
        EXPECT_EQ(core::posix::exit::Status::success, result.detail.if_exited.status);  \
    } \

/**
 * Test definition macro which runs a TEST_F in a forked process.
 * Note that you can only use EXPECT_*, not ASSERT_*!
 *
 * Usage:
 * TESTP_F(FixtureName, TestName, {
 *    ... test code ...
 *    EXPECT_* ...
 *  })
 */
#define TESTP_F(test_fixture, test_name, CODE)                                          \
    TEST_F(test_fixture, test_name) {                                                   \
        auto test = [&]() {                                                             \
            CODE                                                                        \
            return HasFailure() ? core::posix::exit::Status::failure                    \
                            : core::posix::exit::Status::success;                       \
        };                                                                              \
        auto child = core::posix::fork(                                                 \
            test,                                                                       \
            core::posix::StandardStream::empty);                                        \
        auto result = child.wait_for(core::posix::wait::Flags::untraced);               \
        EXPECT_EQ(core::posix::wait::Result::Status::exited, result.status);            \
        EXPECT_EQ(core::posix::exit::Status::success, result.detail.if_exited.status);  \
    } \

#endif // CORE_TESTING_FORK_AND_RUN_H_