summaryrefslogtreecommitdiffstats
path: root/libc/test/src/__support/CPP/stringview_test.cpp
blob: a62cde9bfb818ab31e06dcef051c492f73edf8d0 (plain) (blame)
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
//===-- Unittests for StringView ------------------------------------------===//
//
// 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 "src/__support/CPP/StringView.h"
#include "utils/UnitTest/Test.h"

TEST(LlvmLibcStringViewTest, InitializeCheck) {
  __llvm_libc::cpp::StringView v;
  ASSERT_EQ(v.size(), size_t(0));
  ASSERT_TRUE(v.data() == nullptr);

  v = __llvm_libc::cpp::StringView("");
  ASSERT_EQ(v.size(), size_t(0));
  ASSERT_TRUE(v.data() == nullptr);

  v = __llvm_libc::cpp::StringView(nullptr);
  ASSERT_EQ(v.size(), size_t(0));
  ASSERT_TRUE(v.data() == nullptr);

  v = __llvm_libc::cpp::StringView(nullptr, 10);
  ASSERT_EQ(v.size(), size_t(0));
  ASSERT_TRUE(v.data() == nullptr);

  v = __llvm_libc::cpp::StringView("abc", 0);
  ASSERT_EQ(v.size(), size_t(0));
  ASSERT_TRUE(v.data() == nullptr);

  v = __llvm_libc::cpp::StringView("123456789");
  ASSERT_EQ(v.size(), size_t(9));
}

TEST(LlvmLibcStringViewTest, Equals) {
  __llvm_libc::cpp::StringView v("abc");
  ASSERT_TRUE(v.equals(__llvm_libc::cpp::StringView("abc")));
  ASSERT_FALSE(v.equals(__llvm_libc::cpp::StringView()));
  ASSERT_FALSE(v.equals(__llvm_libc::cpp::StringView("")));
  ASSERT_FALSE(v.equals(__llvm_libc::cpp::StringView("123")));
  ASSERT_FALSE(v.equals(__llvm_libc::cpp::StringView("abd")));
  ASSERT_FALSE(v.equals(__llvm_libc::cpp::StringView("aaa")));
  ASSERT_FALSE(v.equals(__llvm_libc::cpp::StringView("abcde")));
}

TEST(LlvmLibcStringViewTest, RemovePrefix) {
  __llvm_libc::cpp::StringView v("123456789");

  auto p = v.remove_prefix(0);
  ASSERT_EQ(p.size(), size_t(9));
  ASSERT_TRUE(p.equals(__llvm_libc::cpp::StringView("123456789")));

  p = v.remove_prefix(4);
  ASSERT_EQ(p.size(), size_t(5));
  ASSERT_TRUE(p.equals(__llvm_libc::cpp::StringView("56789")));

  p = v.remove_prefix(9);
  ASSERT_EQ(p.size(), size_t(0));
  ASSERT_TRUE(p.data() == nullptr);

  p = v.remove_prefix(10);
  ASSERT_EQ(p.size(), size_t(0));
  ASSERT_TRUE(p.data() == nullptr);
}

TEST(LlvmLibcStringViewTest, RemoveSuffix) {
  __llvm_libc::cpp::StringView v("123456789");

  auto p = v.remove_suffix(0);
  ASSERT_EQ(p.size(), size_t(9));
  ASSERT_TRUE(p.equals(__llvm_libc::cpp::StringView("123456789")));

  p = v.remove_suffix(4);
  ASSERT_EQ(p.size(), size_t(5));
  ASSERT_TRUE(p.equals(__llvm_libc::cpp::StringView("12345")));

  p = v.remove_suffix(9);
  ASSERT_EQ(p.size(), size_t(0));
  ASSERT_TRUE(p.data() == nullptr);

  p = v.remove_suffix(10);
  ASSERT_EQ(p.size(), size_t(0));
  ASSERT_TRUE(p.data() == nullptr);
}

TEST(LlvmLibcStringViewTest, TrimSingleChar) {
  __llvm_libc::cpp::StringView v("     123456789   ");
  auto t = v.trim(' ');
  ASSERT_EQ(t.size(), size_t(9));
  ASSERT_TRUE(t.equals(__llvm_libc::cpp::StringView("123456789")));

  v = __llvm_libc::cpp::StringView("====12345==");
  t = v.trim(' ');
  ASSERT_EQ(v.size(), size_t(11));
  ASSERT_TRUE(t.equals(__llvm_libc::cpp::StringView("====12345==")));

  t = v.trim('=');
  ASSERT_EQ(t.size(), size_t(5));
  ASSERT_TRUE(t.equals(__llvm_libc::cpp::StringView("12345")));

  v = __llvm_libc::cpp::StringView("12345===");
  t = v.trim('=');
  ASSERT_EQ(t.size(), size_t(5));
  ASSERT_TRUE(t.equals(__llvm_libc::cpp::StringView("12345")));

  v = __llvm_libc::cpp::StringView("===========12345");
  t = v.trim('=');
  ASSERT_EQ(t.size(), size_t(5));
  ASSERT_TRUE(t.equals(__llvm_libc::cpp::StringView("12345")));

  v = __llvm_libc::cpp::StringView("============");
  t = v.trim('=');
  ASSERT_EQ(t.size(), size_t(0));
  ASSERT_TRUE(t.data() == nullptr);

  v = __llvm_libc::cpp::StringView();
  t = v.trim(' ');
  ASSERT_EQ(t.size(), size_t(0));
  ASSERT_TRUE(t.data() == nullptr);

  v = __llvm_libc::cpp::StringView("");
  t = v.trim(' ');
  ASSERT_EQ(t.size(), size_t(0));
  ASSERT_TRUE(t.data() == nullptr);
}