1 //===-- Unittests for fprintf ---------------------------------------------===//
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 #ifndef LIBC_COPT_STDIO_USE_SYSTEM_FILE
10 #include "src/stdio/fclose.h"
11 #include "src/stdio/ferror.h"
12 #include "src/stdio/fopen.h"
13 #include "src/stdio/fread.h"
14 #endif // LIBC_COPT_STDIO_USE_SYSTEM_FILE
15
16 #include "src/stdio/fprintf.h"
17
18 #include "test/UnitTest/Test.h"
19
20 namespace printf_test {
21 #ifndef LIBC_COPT_STDIO_USE_SYSTEM_FILE
22 using LIBC_NAMESPACE::fclose;
23 using LIBC_NAMESPACE::ferror;
24 using LIBC_NAMESPACE::fopen;
25 using LIBC_NAMESPACE::fread;
26 #else // defined(LIBC_COPT_STDIO_USE_SYSTEM_FILE)
27 using ::fclose;
28 using ::ferror;
29 using ::fopen;
30 using ::fread;
31 #endif // LIBC_COPT_STDIO_USE_SYSTEM_FILE
32 } // namespace printf_test
33
TEST(LlvmLibcFPrintfTest,WriteToFile)34 TEST(LlvmLibcFPrintfTest, WriteToFile) {
35 const char *FILENAME = "fprintf_output.test";
36 auto FILE_PATH = libc_make_test_file_path(FILENAME);
37
38 ::FILE *file = printf_test::fopen(FILE_PATH, "w");
39 ASSERT_FALSE(file == nullptr);
40
41 int written;
42
43 constexpr char simple[] = "A simple string with no conversions.\n";
44 written = LIBC_NAMESPACE::fprintf(file, simple);
45 EXPECT_EQ(written, 37);
46
47 constexpr char numbers[] = "1234567890\n";
48 written = LIBC_NAMESPACE::fprintf(file, "%s", numbers);
49 EXPECT_EQ(written, 11);
50
51 constexpr char format_more[] = "%s and more\n";
52 constexpr char short_numbers[] = "1234";
53 written = LIBC_NAMESPACE::fprintf(file, format_more, short_numbers);
54 EXPECT_EQ(written, 14);
55
56 ASSERT_EQ(0, printf_test::fclose(file));
57
58 file = printf_test::fopen(FILE_PATH, "r");
59 ASSERT_FALSE(file == nullptr);
60
61 char data[50];
62 ASSERT_EQ(printf_test::fread(data, 1, sizeof(simple) - 1, file),
63 sizeof(simple) - 1);
64 data[sizeof(simple) - 1] = '\0';
65 ASSERT_STREQ(data, simple);
66 ASSERT_EQ(printf_test::fread(data, 1, sizeof(numbers) - 1, file),
67 sizeof(numbers) - 1);
68 data[sizeof(numbers) - 1] = '\0';
69 ASSERT_STREQ(data, numbers);
70 ASSERT_EQ(printf_test::fread(
71 data, 1, sizeof(format_more) + sizeof(short_numbers) - 4, file),
72 sizeof(format_more) + sizeof(short_numbers) - 4);
73 data[sizeof(format_more) + sizeof(short_numbers) - 4] = '\0';
74 ASSERT_STREQ(data, "1234 and more\n");
75
76 ASSERT_EQ(printf_test::ferror(file), 0);
77
78 written =
79 LIBC_NAMESPACE::fprintf(file, "Writing to a read only file should fail.");
80 EXPECT_LT(written, 0);
81
82 ASSERT_EQ(printf_test::fclose(file), 0);
83 }
84