87 lines · cpp
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-exception6//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 span35 // 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