1 | //===- ErrnoTest.cpp - Error handling unit tests --------------------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "llvm/Support/Errno.h" |
10 | #include "gtest/gtest.h" |
11 | |
12 | using namespace llvm::sys; |
13 | |
14 | TEST(ErrnoTest, RetryAfterSignal) { |
15 | EXPECT_EQ(1, RetryAfterSignal(-1, [] { return 1; })); |
16 | |
17 | EXPECT_EQ(-1, RetryAfterSignal(-1, [] { |
18 | errno = EAGAIN; |
19 | return -1; |
20 | })); |
21 | EXPECT_EQ(EAGAIN, errno); |
22 | |
23 | unsigned calls = 0; |
24 | EXPECT_EQ(1, RetryAfterSignal(-1, [&calls] { |
25 | errno = EINTR; |
26 | ++calls; |
27 | return calls == 1 ? -1 : 1; |
28 | })); |
29 | EXPECT_EQ(2u, calls); |
30 | |
31 | EXPECT_EQ(1, RetryAfterSignal(-1, [](int x) { return x; }, 1)); |
32 | |
33 | std::unique_ptr<int> P(RetryAfterSignal(Fail: nullptr, F: [] { return new int(47); })); |
34 | EXPECT_EQ(47, *P); |
35 | |
36 | errno = EINTR; |
37 | EXPECT_EQ(-1, RetryAfterSignal(-1, [] { return -1; })); |
38 | } |
39 | |