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

source code of libc/test/src/wchar/wcsspn_test.cpp