62 lines · cpp
1//===-- Unittests for isatty ----------------------------------------------===//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-exception6//7//===----------------------------------------------------------------------===//8 9#include "src/fcntl/open.h"10#include "src/unistd/close.h"11#include "src/unistd/isatty.h"12#include "test/UnitTest/ErrnoCheckingTest.h"13#include "test/UnitTest/ErrnoSetterMatcher.h"14#include "test/UnitTest/Test.h"15 16#include <sys/stat.h>17 18using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;19using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds;20using LlvmLibcIsATTYTest = LIBC_NAMESPACE::testing::ErrnoCheckingTest;21 22TEST_F(LlvmLibcIsATTYTest, StdInOutTests) {23 // If stdin is connected to a terminal, assume that all of the standard i/o24 // fds are.25 if (LIBC_NAMESPACE::isatty(0)) {26 EXPECT_THAT(LIBC_NAMESPACE::isatty(0), Succeeds(1)); // stdin27 EXPECT_THAT(LIBC_NAMESPACE::isatty(1), Succeeds(1)); // stdout28 EXPECT_THAT(LIBC_NAMESPACE::isatty(2), Succeeds(1)); // stderr29 } else {30 EXPECT_THAT(LIBC_NAMESPACE::isatty(0), Fails(ENOTTY, 0)); // stdin31 EXPECT_THAT(LIBC_NAMESPACE::isatty(1), Fails(ENOTTY, 0)); // stdout32 EXPECT_THAT(LIBC_NAMESPACE::isatty(2), Fails(ENOTTY, 0)); // stderr33 }34}35 36TEST_F(LlvmLibcIsATTYTest, BadFdTest) {37 EXPECT_THAT(LIBC_NAMESPACE::isatty(-1), Fails(EBADF, 0)); // invalid fd38}39 40TEST_F(LlvmLibcIsATTYTest, DevTTYTest) {41 constexpr const char *FILENAME = "/dev/tty";42 auto TTY_FILE = libc_make_test_file_path(FILENAME);43 int fd = LIBC_NAMESPACE::open(TTY_FILE, O_RDONLY);44 if (fd > 0) {45 ASSERT_ERRNO_SUCCESS();46 EXPECT_THAT(LIBC_NAMESPACE::isatty(fd), Succeeds(1));47 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));48 } else {49 ASSERT_ERRNO_FAILURE();50 }51}52 53TEST_F(LlvmLibcIsATTYTest, FileTest) {54 constexpr const char *FILENAME = "isatty.test";55 auto TEST_FILE = libc_make_test_file_path(FILENAME);56 int fd = LIBC_NAMESPACE::open(TEST_FILE, O_WRONLY | O_CREAT, S_IRWXU);57 ASSERT_ERRNO_SUCCESS();58 ASSERT_GT(fd, 0);59 EXPECT_THAT(LIBC_NAMESPACE::isatty(fd), Fails(ENOTTY, 0));60 ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0));61}62