1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
//===-- 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');
}
|
