1//===-- Unittests for strspn ----------------------------------------------===//
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 "hdr/signal_macros.h"
10#include "src/string/strspn.h"
11
12#include "test/UnitTest/Test.h"
13
14TEST(LlvmLibcStrSpnTest, EmptyStringShouldReturnZeroLengthSpan) {
15 // The search should not include the null terminator.
16 EXPECT_EQ(LIBC_NAMESPACE::strspn("", ""), size_t{0});
17 EXPECT_EQ(LIBC_NAMESPACE::strspn("_", ""), size_t{0});
18 EXPECT_EQ(LIBC_NAMESPACE::strspn("", "_"), size_t{0});
19}
20
21TEST(LlvmLibcStrSpnTest, ShouldNotSpanAnythingAfterNullTerminator) {
22 const char src[4] = {'a', 'b', '\0', 'c'};
23 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "ab"), size_t{2});
24 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "c"), size_t{0});
25
26 // Same goes for the segment to be searched for.
27 const char segment[4] = {'1', '2', '\0', '3'};
28 EXPECT_EQ(LIBC_NAMESPACE::strspn("123", segment), size_t{2});
29}
30
31TEST(LlvmLibcStrSpnTest, SpanEachIndividualCharacter) {
32 const char *src = "12345";
33 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "1"), size_t{1});
34 // Since '1' is not within the segment, the span
35 // size should remain zero.
36 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "2"), size_t{0});
37 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "3"), size_t{0});
38 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "4"), size_t{0});
39 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "5"), size_t{0});
40}
41
42TEST(LlvmLibcStrSpnTest, UnmatchedCharacterShouldNotBeCountedInSpan) {
43 EXPECT_EQ(LIBC_NAMESPACE::strspn("a", "b"), size_t{0});
44 EXPECT_EQ(LIBC_NAMESPACE::strspn("abcdef", "1"), size_t{0});
45 EXPECT_EQ(LIBC_NAMESPACE::strspn("123", "4"), size_t{0});
46}
47
48TEST(LlvmLibcStrSpnTest, SequentialCharactersShouldSpan) {
49 const char *src = "abcde";
50 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "a"), size_t{1});
51 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "ab"), size_t{2});
52 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "abc"), size_t{3});
53 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "abcd"), size_t{4});
54 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "abcde"), size_t{5});
55 // Same thing for when the roles are reversed.
56 EXPECT_EQ(LIBC_NAMESPACE::strspn("abcde", src), size_t{5});
57 EXPECT_EQ(LIBC_NAMESPACE::strspn("abcd", src), size_t{4});
58 EXPECT_EQ(LIBC_NAMESPACE::strspn("abc", src), size_t{3});
59 EXPECT_EQ(LIBC_NAMESPACE::strspn("ab", src), size_t{2});
60 EXPECT_EQ(LIBC_NAMESPACE::strspn("a", src), size_t{1});
61}
62
63TEST(LlvmLibcStrSpnTest, NonSequentialCharactersShouldNotSpan) {
64 const char *src = "123456789";
65 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "_1_abc_2_def_3_"), size_t{3});
66 // Only spans 4 since '5' is not within the span.
67 EXPECT_EQ(LIBC_NAMESPACE::strspn(src, "67__34abc12"), size_t{4});
68}
69
70TEST(LlvmLibcStrSpnTest, ReverseCharacters) {
71 // Since these are still sequential, this should span.
72 EXPECT_EQ(LIBC_NAMESPACE::strspn("12345", "54321"), size_t{5});
73 // Does not span any since '1' is not within the span.
74 EXPECT_EQ(LIBC_NAMESPACE::strspn("12345", "432"), size_t{0});
75 // Only spans 1 since '2' is not within the span.
76 EXPECT_EQ(LIBC_NAMESPACE::strspn("12345", "51"), size_t{1});
77}
78
79TEST(LlvmLibcStrSpnTest, DuplicatedCharactersToBeSearchedForShouldStillMatch) {
80 // Only a single character, so only spans 1.
81 EXPECT_EQ(LIBC_NAMESPACE::strspn("a", "aa"), size_t{1});
82 // This should count once for each 'a' in the source string.
83 EXPECT_EQ(LIBC_NAMESPACE::strspn("aa", "aa"), size_t{2});
84 EXPECT_EQ(LIBC_NAMESPACE::strspn("aaa", "aa"), size_t{3});
85 EXPECT_EQ(LIBC_NAMESPACE::strspn("aaaa", "aa"), size_t{4});
86}
87
88#if defined(LIBC_ADD_NULL_CHECKS) && !defined(LIBC_HAS_SANITIZER)
89
90TEST(LlvmLibcStrSpnTest, CrashOnNullPtr) {
91 ASSERT_DEATH([]() { LIBC_NAMESPACE::strspn(nullptr, nullptr); },
92 WITH_SIGNAL(-1));
93}
94
95#endif // defined(LIBC_ADD_NULL_CHECKS) && !defined(LIBC_HAS_SANITIZER)
96

source code of libc/test/src/string/strspn_test.cpp