llvm/unittests/Support/ErrnoTest.cpp

   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 }