1//===-- Unittests for wcscspn ---------------------------------------------===//
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/wcscspn.h"
12#include "test/UnitTest/Test.h"
13
14TEST(LlvmLibcWCSCSpnTest, EmptyStringShouldReturnZeroLengthSpan) {
15 // The search should not include the null terminator.
16 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"", L""), size_t(0));
17 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"_", L""), size_t(1));
18 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"", L"_"), size_t(0));
19}
20
21TEST(LlvmLibcWCSCSpnTest, ShouldNotSpanAnythingAfterNullTerminator) {
22 const wchar_t src[4] = {L'a', L'b', L'\0', L'c'};
23 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"de"), size_t(2));
24 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"c"), size_t(2));
25
26 // Same goes for the segment to be searched for.
27 const wchar_t segment[4] = {L'1', L'2', L'\0', L'3'};
28 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"3", segment), size_t(1));
29}
30
31TEST(LlvmLibcWCSCSpnTest, SpanEachIndividualCharacter) {
32 const wchar_t *src = L"12345";
33 // These are all in the segment.
34 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"1"), size_t(0));
35 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"2"), size_t(1));
36 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"3"), size_t(2));
37 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"4"), size_t(3));
38 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"5"), size_t(4));
39}
40
41TEST(LlvmLibcWCSCSpnTest, UnmatchedCharacterShouldReturnLength) {
42 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"a", L"b"), size_t(1));
43 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"abcdef", L"1"), size_t(6));
44 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"123", L"4"), size_t(3));
45}
46
47TEST(LlvmLibcWCSCSpnTest, NonSequentialCharactersShouldNotSpan) {
48 const wchar_t *src = L"abc456789";
49 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"_1_abc_2_def_3_"), size_t(0));
50 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(src, L"67__34xyz12"), size_t(3));
51}
52
53TEST(LlvmLibcWCSCSpnTest, ReverseCharacters) {
54 // These are all in the string.
55 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"12345", L"54321"), size_t(0));
56 // 1 is not in the span.
57 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"12345", L"432"), size_t(1));
58 // 1 is in the span.
59 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"12345", L"51"), size_t(0));
60}
61
62TEST(LlvmLibcWCSCSpnTest, DuplicatedCharactersToBeSearchedForShouldStillMatch) {
63 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"a", L"aa"), size_t(0));
64 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"aa", L"aa"), size_t(0));
65 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"aaa", L"bb"), size_t(3));
66 EXPECT_EQ(LIBC_NAMESPACE::wcscspn(L"aaaa", L"bb"), size_t(4));
67}
68

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