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(nullptr, [] { return new int(47); }));
34   EXPECT_EQ(47, *P);
35 
36   errno = EINTR;
37   EXPECT_EQ(-1, RetryAfterSignal(-1, [] { return -1; }));
38 }
39