blob: d5f0279413fa13306d5f883e484a124509b25ac6 (
plain) (
blame)
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
|
#include "interrupt_signals.h"
#include <util/datetime/base.h>
#include <library/cpp/testing/unittest/registar.h>
#include <atomic>
#ifdef _win_
#include <windows.h>
#endif
Y_UNIT_TEST_SUITE(TTestInterruptSignals) {
static std::atomic<size_t> HandledSigNum = 0;
static void Handler(int signum) {
HandledSigNum.store(signum);
}
Y_UNIT_TEST(Test1) {
SetInterruptSignalsHandler(Handler);
#ifdef _win_
// TODO: unfortunately GenerateConsoleCtrlEvent fails under Wine
/*
for (auto [winEvent, posixSigNum] : {
std::make_pair(CTRL_C_EVENT, SIGINT),
std::make_pair(CTRL_BREAK_EVENT, SIGTERM)
})
{
if (!GenerateConsoleCtrlEvent(winEvent, 0)) {
UNIT_FAIL("GenerateConsoleCtrlEvent failed: " << LastSystemErrorText());
}
Sleep(TDuration::MilliSeconds(100));
UNIT_ASSERT_VALUES_EQUAL(HandledSigNum.load(), posixSigNum);
}
*/
for (int signum : {SIGINT, SIGTERM}) {
#else
for (int signum : {SIGINT, SIGTERM, SIGHUP}) {
#endif
std::raise(signum);
Sleep(TDuration::MilliSeconds(100)); // give it time to handle an async signal
UNIT_ASSERT_VALUES_EQUAL(HandledSigNum.load(), signum);
}
}
} // Y_UNIT_TEST_SUITE(TTestInterruptSignals)
|
