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 | |
14 | TEST(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 | |
21 | TEST(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 | |
31 | TEST(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 | |
42 | TEST(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 | |
48 | TEST(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 | |
63 | TEST(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 | |
70 | TEST(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 | |
79 | TEST(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 | |
90 | TEST(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 | |