89 lines
2.5 KiB
C++
89 lines
2.5 KiB
C++
//===-- Unittests for wmemset ---------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "hdr/types/size_t.h"
|
|
#include "hdr/types/wchar_t.h"
|
|
#include "src/wchar/wmemset.h"
|
|
#include "test/UnitTest/Test.h"
|
|
|
|
TEST(LlvmLibcWMemsetTest, SmallStringBoundCheck) {
|
|
wchar_t str[5];
|
|
for (int i = 0; i < 5; i++)
|
|
str[i] = 'A';
|
|
|
|
wchar_t *output = LIBC_NAMESPACE::wmemset(str + 1, 'B', 3);
|
|
|
|
EXPECT_EQ(output, str + 1);
|
|
|
|
// EXPECT_TRUE being used since there isn't currently support for printing
|
|
// wide chars in the future, it would be preferred to switch these to
|
|
// EXPECT_EQ
|
|
EXPECT_TRUE(str[0] == (wchar_t)'A');
|
|
EXPECT_TRUE(str[1] == (wchar_t)'B');
|
|
EXPECT_TRUE(str[2] == (wchar_t)'B');
|
|
EXPECT_TRUE(str[3] == (wchar_t)'B');
|
|
EXPECT_TRUE(str[4] == (wchar_t)'A');
|
|
}
|
|
|
|
TEST(LlvmLibcWMemsetTest, LargeStringBoundCheck) {
|
|
constexpr int str_size = 1000;
|
|
wchar_t str[str_size];
|
|
for (int i = 0; i < str_size; i++)
|
|
str[i] = 'A';
|
|
|
|
wchar_t *output = LIBC_NAMESPACE::wmemset(str + 1, 'B', str_size - 2);
|
|
|
|
EXPECT_EQ(output, str + 1);
|
|
|
|
EXPECT_TRUE(str[0] == (wchar_t)'A');
|
|
for (int i = 1; i < str_size - 1; i++)
|
|
EXPECT_TRUE(str[i] == (wchar_t)'B');
|
|
|
|
EXPECT_TRUE(str[str_size - 1] == (wchar_t)'A');
|
|
}
|
|
|
|
TEST(LlvmLibcWMemsetTest, WCharSizeSmallString) {
|
|
// ensure we can handle full range of widechars
|
|
wchar_t str[5];
|
|
const wchar_t target = WCHAR_MAX;
|
|
|
|
for (int i = 0; i < 5; i++)
|
|
str[i] = 'A';
|
|
|
|
wchar_t *output = LIBC_NAMESPACE::wmemset(str + 1, target, 3);
|
|
|
|
EXPECT_EQ(output, str + 1);
|
|
|
|
EXPECT_TRUE(str[0] == (wchar_t)'A');
|
|
EXPECT_TRUE(str[1] == target);
|
|
EXPECT_TRUE(str[2] == target);
|
|
EXPECT_TRUE(str[3] == target);
|
|
EXPECT_TRUE(str[4] == (wchar_t)'A');
|
|
}
|
|
|
|
TEST(LlvmLibcWMemsetTest, WCharSizeLargeString) {
|
|
// ensure we can handle full range of widechars
|
|
constexpr int str_size = 1000;
|
|
wchar_t str[str_size];
|
|
|
|
const wchar_t target = WCHAR_MAX;
|
|
|
|
for (int i = 0; i < str_size; i++)
|
|
str[i] = 'A';
|
|
|
|
wchar_t *output = LIBC_NAMESPACE::wmemset(str + 1, target, str_size - 2);
|
|
|
|
EXPECT_EQ(output, str + 1);
|
|
|
|
EXPECT_TRUE(str[0] == (wchar_t)'A');
|
|
for (int i = 1; i < str_size - 1; i++)
|
|
EXPECT_TRUE(str[i] == target);
|
|
|
|
EXPECT_TRUE(str[str_size - 1] == (wchar_t)'A');
|
|
}
|