| //===-- Unittests for vfprintf --------------------------------------------===// |
| // |
| // 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 |
| // |
| //===----------------------------------------------------------------------===// |
| |
| // These tests are copies of the non-v variants of the printf functions. This is |
| // because these functions are identical in every way except for how the varargs |
| // are passed. |
| |
| #ifndef LIBC_COPT_STDIO_USE_SYSTEM_FILE |
| #include "src/stdio/fclose.h" |
| #include "src/stdio/ferror.h" |
| #include "src/stdio/fopen.h" |
| #include "src/stdio/fread.h" |
| #endif // LIBC_COPT_STDIO_USE_SYSTEM_FILE |
| |
| #include "src/stdio/vfprintf.h" |
| |
| #include "test/UnitTest/Test.h" |
| |
| #include <stdio.h> |
| |
| namespace printf_test { |
| #ifndef LIBC_COPT_STDIO_USE_SYSTEM_FILE |
| using LIBC_NAMESPACE::fclose; |
| using LIBC_NAMESPACE::ferror; |
| using LIBC_NAMESPACE::fopen; |
| using LIBC_NAMESPACE::fread; |
| #else // defined(LIBC_COPT_STDIO_USE_SYSTEM_FILE) |
| using ::fclose; |
| using ::ferror; |
| using ::fopen; |
| using ::fread; |
| #endif // LIBC_COPT_STDIO_USE_SYSTEM_FILE |
| } // namespace printf_test |
| |
| int call_vfprintf(::FILE *__restrict stream, const char *__restrict format, |
| ...) { |
| va_list vlist; |
| va_start(vlist, format); |
| int ret = LIBC_NAMESPACE::vfprintf(stream, format, vlist); |
| va_end(vlist); |
| return ret; |
| } |
| |
| TEST(LlvmLibcVFPrintfTest, WriteToFile) { |
| const char *FILENAME = "vfprintf_output.test"; |
| auto FILE_PATH = libc_make_test_file_path(FILENAME); |
| |
| ::FILE *file = printf_test::fopen(FILE_PATH, "w"); |
| ASSERT_FALSE(file == nullptr); |
| |
| int written; |
| |
| constexpr char simple[] = "A simple string with no conversions.\n"; |
| written = call_vfprintf(file, simple); |
| EXPECT_EQ(written, 37); |
| |
| constexpr char numbers[] = "1234567890\n"; |
| written = call_vfprintf(file, "%s", numbers); |
| EXPECT_EQ(written, 11); |
| |
| constexpr char format_more[] = "%s and more\n"; |
| constexpr char short_numbers[] = "1234"; |
| written = call_vfprintf(file, format_more, short_numbers); |
| EXPECT_EQ(written, 14); |
| |
| ASSERT_EQ(0, printf_test::fclose(file)); |
| |
| file = printf_test::fopen(FILE_PATH, "r"); |
| ASSERT_FALSE(file == nullptr); |
| |
| char data[50]; |
| ASSERT_EQ(printf_test::fread(data, 1, sizeof(simple) - 1, file), |
| sizeof(simple) - 1); |
| data[sizeof(simple) - 1] = '\0'; |
| ASSERT_STREQ(data, simple); |
| ASSERT_EQ(printf_test::fread(data, 1, sizeof(numbers) - 1, file), |
| sizeof(numbers) - 1); |
| data[sizeof(numbers) - 1] = '\0'; |
| ASSERT_STREQ(data, numbers); |
| ASSERT_EQ(printf_test::fread( |
| data, 1, sizeof(format_more) + sizeof(short_numbers) - 4, file), |
| sizeof(format_more) + sizeof(short_numbers) - 4); |
| data[sizeof(format_more) + sizeof(short_numbers) - 4] = '\0'; |
| ASSERT_STREQ(data, "1234 and more\n"); |
| |
| ASSERT_EQ(printf_test::ferror(file), 0); |
| |
| written = call_vfprintf(file, "Writing to a read only file should fail."); |
| EXPECT_LT(written, 0); |
| |
| ASSERT_EQ(printf_test::fclose(file), 0); |
| } |