libc/test/src/string/strlcat_test.cpp

   1 //===-- Unittests for strlcat ---------------------------------------------===//
   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/string/strlcat.h"
  10 #include "test/UnitTest/Test.h"
  11
  12 TEST(LlvmLibcStrlcatTest, TooBig) {
  13   const char *str = "cd";
  14   char buf[4]{"ab"};
  15   EXPECT_EQ(LIBC_NAMESPACE::strlcat(buf, str, 3), size_t(4));
  16   EXPECT_STREQ(buf, "ab");
  17   EXPECT_EQ(LIBC_NAMESPACE::strlcat(buf, str, 4), size_t(4));
  18   EXPECT_STREQ(buf, "abc");
  19 }
  20
  21 TEST(LlvmLibcStrlcatTest, Smaller) {
  22   const char *str = "cd";
  23   char buf[7]{"ab"};
  24
  25   EXPECT_EQ(LIBC_NAMESPACE::strlcat(buf, str, 7), size_t(4));
  26   EXPECT_STREQ(buf, "abcd");
  27 }
  28
  29 TEST(LlvmLibcStrlcatTest, SmallerNoOverwriteAfter0) {
  30   const char *str = "cd";
  31   char buf[8]{"ab\0\0efg"};
  32
  33   EXPECT_EQ(LIBC_NAMESPACE::strlcat(buf, str, 8), size_t(4));
  34   EXPECT_STREQ(buf, "abcd");
  35   EXPECT_STREQ(buf + 5, "fg");
  36 }
  37
  38 TEST(LlvmLibcStrlcatTest, No0) {
  39   const char *str = "cd";
  40   char buf[7]{"ab"};
  41   EXPECT_EQ(LIBC_NAMESPACE::strlcat(buf, str, 1), size_t(3));
  42   EXPECT_STREQ(buf, "ab");
  43   EXPECT_EQ(LIBC_NAMESPACE::strlcat(buf, str, 2), size_t(4));
  44   EXPECT_STREQ(buf, "ab");
  45 }