libc/test/src/stdlib/atexit_test.cpp

   1 //===-- Unittests for atexit ----------------------------------------------===//
   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 "src/__support/CPP/array.h"
  10 #include "src/__support/CPP/utility.h"
  11 #include "src/stdlib/atexit.h"
  12 #include "src/stdlib/exit.h"
  13 #include "test/UnitTest/Test.h"
  14
  15 static int a;
  16 TEST(LlvmLibcAtExit, Basic) {
  17   // In case tests ever run multiple times.
  18   a = 0;
  19
  20   auto test = [] {
  21     int status = __llvm_libc::atexit(+[] {
  22       if (a != 1)
  23         __builtin_trap();
  24     });
  25     status |= __llvm_libc::atexit(+[] { a++; });
  26     if (status)
  27       __builtin_trap();
  28
  29     __llvm_libc::exit(0);
  30   };
  31   EXPECT_EXITS(test, 0);
  32 }
  33
  34 TEST(LlvmLibcAtExit, AtExitCallsSysExit) {
  35   auto test = [] {
  36     __llvm_libc::atexit(+[] { _Exit(1); });
  37     __llvm_libc::exit(0);
  38   };
  39   EXPECT_EXITS(test, 1);
  40 }
  41
  42 static int size;
  43 static __llvm_libc::cpp::array<int, 256> arr;
  44
  45 template <int... Ts>
  46 void register_atexit_handlers(__llvm_libc::cpp::integer_sequence<int, Ts...>) {
  47   (__llvm_libc::atexit(+[] { arr[size++] = Ts; }), ...);
  48 }
  49
  50 template <int count> constexpr auto getTest() {
  51   return [] {
  52     __llvm_libc::atexit(+[] {
  53       if (size != count)
  54         __builtin_trap();
  55       for (int i = 0; i < count; i++)
  56         if (arr[i] != count - 1 - i)
  57           __builtin_trap();
  58     });
  59     register_atexit_handlers(
  60         __llvm_libc::cpp::make_integer_sequence<int, count>{});
  61     __llvm_libc::exit(0);
  62   };
  63 }
  64
  65 TEST(LlvmLibcAtExit, ReverseOrder) {
  66   // In case tests ever run multiple times.
  67   size = 0;
  68
  69   auto test = getTest<32>();
  70   EXPECT_EXITS(test, 0);
  71 }
  72
  73 TEST(LlvmLibcAtExit, Many) {
  74   // In case tests ever run multiple times.
  75   size = 0;
  76
  77   auto test = getTest<256>();
  78   EXPECT_EXITS(test, 0);
  79 }
  80
  81 TEST(LlvmLibcAtExit, HandlerCallsAtExit) {
  82   auto test = [] {
  83     __llvm_libc::atexit(
  84         +[] { __llvm_libc::atexit(+[] { __llvm_libc::exit(1); }); });
  85     __llvm_libc::exit(0);
  86   };
  87   EXPECT_EXITS(test, 1);
  88 }