Add StringRef::{substr, startswith}.

[oota-llvm.git] / unittests / ADT / StringRefTest.cpp

diff --git a/unittests/ADT/StringRefTest.cpp b/unittests/ADT/StringRefTest.cpp
index 8a2c7435d8f13c338386bb04e910e3d21a02eee8..bb46dc010a563a6522f1ef3950784cfc92478727 100644 (file)
--- a/unittests/ADT/StringRefTest.cpp
+++ b/unittests/ADT/StringRefTest.cpp
@@ -56,4 +56,16 @@ TEST(StringRefTest, Operators) {
   EXPECT_EQ('a', StringRef("aab")[1]);
 }
 
+TEST(StringRefTest, Utilities) {
+  StringRef Str("hello");
+  EXPECT_TRUE(Str.substr(3) == "lo");
+  EXPECT_TRUE(Str.substr(100) == "");
+  EXPECT_TRUE(Str.substr(0, 100) == "hello");
+  EXPECT_TRUE(Str.substr(4, 10) == "o");
+
+  EXPECT_TRUE(Str.startswith("he"));
+  EXPECT_FALSE(Str.startswith("helloworld"));
+  EXPECT_FALSE(Str.startswith("hi"));
+}
+
 } // end anonymous namespace