//===-- Unittests for fegetround and fesetround ---------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "src/fenv/fegetround.h"
#include "src/fenv/fesetround.h"

#include "test/UnitTest/FEnvSafeTest.h"
#include "test/UnitTest/Test.h"

#include "hdr/fenv_macros.h"

using LlvmLibcRoundingModeTest = LIBC_NAMESPACE::testing::FEnvSafeTest;

TEST_F(LlvmLibcRoundingModeTest, SetAndGet) {
  struct ResetDefaultRoundingMode {
    int original = LIBC_NAMESPACE::fegetround();
    ~ResetDefaultRoundingMode() { LIBC_NAMESPACE::fesetround(original); }
  } reset;

  int s = LIBC_NAMESPACE::fesetround(FE_TONEAREST);
  EXPECT_EQ(s, 0);
  int rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_TONEAREST);

  s = LIBC_NAMESPACE::fesetround(FE_UPWARD);
  EXPECT_EQ(s, 0);
  rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_UPWARD);

  s = LIBC_NAMESPACE::fesetround(FE_DOWNWARD);
  EXPECT_EQ(s, 0);
  rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_DOWNWARD);

  s = LIBC_NAMESPACE::fesetround(FE_TOWARDZERO);
  EXPECT_EQ(s, 0);
  rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_TOWARDZERO);
}