libcryfs/test/cpp-utils/process/SignalHandlerTest.cpp

133 lines
2.8 KiB
C++
Raw Normal View History

#include <gtest/gtest.h>
#include <cpp-utils/process/SignalHandler.h>
using namespace cpputils;
namespace {
std::atomic<int> triggered;
void trigger(int signal) {
triggered = signal;
}
void raise_signal(int signal) {
int error = ::raise(signal);
if (error != 0) {
throw std::runtime_error("Error raising signal");
}
}
TEST(SignalHandlerTest, givenNoSignalHandler_whenRaisingSigint_thenDies) {
EXPECT_DEATH(
raise_signal(SIGINT),
""
);
}
TEST(SignalHandlerTest, givenNoSignalHandler_whenRaisingSigterm_thenDies) {
EXPECT_DEATH(
raise_signal(SIGTERM),
""
);
}
TEST(SignalHandlerTest, givenSigIntHandler_whenRaisingSigInt_thenCatches) {
triggered = 0;
SignalHandlerRAII<&trigger> handler(SIGINT);
raise_signal(SIGINT);
EXPECT_EQ(SIGINT, triggered);
}
TEST(SignalHandlerTest, givenSigIntHandler_whenRaisingSigTerm_thenDies) {
SignalHandlerRAII<&trigger> handler(SIGINT);
EXPECT_DEATH(
raise_signal(SIGTERM),
""
);
}
TEST(SignalHandlerTest, givenSigTermHandler_whenRaisingSigTerm_thenCatches) {
triggered = 0;
SignalHandlerRAII<&trigger> handler(SIGTERM);
raise_signal(SIGTERM);
EXPECT_EQ(SIGTERM, triggered);
}
TEST(SignalHandlerTest, givenSigTermHandler_whenRaisingSigInt_thenDies) {
SignalHandlerRAII<&trigger> handler(SIGTERM);
EXPECT_DEATH(
raise_signal(SIGINT),
""
);
}
TEST(SignalHandlerTest, givenSigIntAndSigTermHandlers_whenRaising_thenCatchesCorrectSignal) {
triggered = 0;
SignalHandlerRAII<&trigger> handler1(SIGINT);
SignalHandlerRAII<&trigger> handler2(SIGTERM);
raise_signal(SIGINT);
EXPECT_EQ(SIGINT, triggered);
raise_signal(SIGTERM);
EXPECT_EQ(SIGTERM, triggered);
raise_signal(SIGINT);
EXPECT_EQ(SIGINT, triggered);
}
std::atomic<int> triggered_count_1;
std::atomic<int> triggered_count_2;
void trigger1(int) {
++triggered_count_1;
}
void trigger2(int) {
++triggered_count_2;
}
TEST(SignalHandlerTest, givenMultipleSigIntHandlers_whenRaising_thenCatchesCorrectSignal) {
triggered_count_1 = 0;
triggered_count_2 = 0;
{
SignalHandlerRAII<&trigger1> handler1(SIGINT);
{
SignalHandlerRAII<&trigger2> handler2(SIGINT);
raise_signal(SIGINT);
EXPECT_EQ(0, triggered_count_1);
EXPECT_EQ(1, triggered_count_2);
raise_signal(SIGINT);
EXPECT_EQ(0, triggered_count_1);
EXPECT_EQ(2, triggered_count_2);
}
raise_signal(SIGINT);
EXPECT_EQ(1, triggered_count_1);
EXPECT_EQ(2, triggered_count_2);
raise_signal(SIGINT);
EXPECT_EQ(2, triggered_count_1);
EXPECT_EQ(2, triggered_count_2);
}
EXPECT_DEATH(
raise_signal(SIGINT),
""
);
}
}