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-exception |
6 | // |
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 | |
18 | using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails; |
19 | using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; |
20 | using LlvmLibcIsATTYTest = LIBC_NAMESPACE::testing::ErrnoCheckingTest; |
21 | |
22 | TEST_F(LlvmLibcIsATTYTest, StdInOutTests) { |
23 | // If stdin is connected to a terminal, assume that all of the standard i/o |
24 | // fds are. |
25 | if (LIBC_NAMESPACE::isatty(0)) { |
26 | EXPECT_THAT(LIBC_NAMESPACE::isatty(0), Succeeds(1)); // stdin |
27 | EXPECT_THAT(LIBC_NAMESPACE::isatty(1), Succeeds(1)); // stdout |
28 | EXPECT_THAT(LIBC_NAMESPACE::isatty(2), Succeeds(1)); // stderr |
29 | } else { |
30 | EXPECT_THAT(LIBC_NAMESPACE::isatty(0), Fails(ENOTTY, 0)); // stdin |
31 | EXPECT_THAT(LIBC_NAMESPACE::isatty(1), Fails(ENOTTY, 0)); // stdout |
32 | EXPECT_THAT(LIBC_NAMESPACE::isatty(2), Fails(ENOTTY, 0)); // stderr |
33 | } |
34 | } |
35 | |
36 | TEST_F(LlvmLibcIsATTYTest, BadFdTest) { |
37 | EXPECT_THAT(LIBC_NAMESPACE::isatty(-1), Fails(EBADF, 0)); // invalid fd |
38 | } |
39 | |
40 | TEST_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 | |
53 | TEST_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 | |