Fix pasto in StringRef::count(char)

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@79356 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/llvm/ADT/StringRef.h b/include/llvm/ADT/StringRef.h
index 5433056039065f339e5c90b4f8193fad58140ac1..acfc335b8059e917b78a0e0279de6e7ba6c5f8e9 100644 (file)
--- a/include/llvm/ADT/StringRef.h
+++ b/include/llvm/ADT/StringRef.h
@@ -177,7 +177,7 @@ namespace llvm {
       size_t Count = 0;
       for (size_t i = 0, e = Length; i != e; ++i)
         if (Data[i] == C)
-          return i;
+          ++Count;
       return Count;
     }
 

diff --git a/unittests/ADT/StringRefTest.cpp b/unittests/ADT/StringRefTest.cpp
index ef89e9f3f9d990a12303f270117effd7add5ff9c..320633aca8a1dee635502d769b0049ec0fbe7feb 100644 (file)
--- a/unittests/ADT/StringRefTest.cpp
+++ b/unittests/ADT/StringRefTest.cpp
@@ -96,6 +96,7 @@ TEST(StringRefTest, Utilities) {
   EXPECT_TRUE(Str.find("zz") == StringRef::npos);
 
   EXPECT_TRUE(Str.count('l') == 2);
+  EXPECT_TRUE(Str.count('o') == 1);
   EXPECT_TRUE(Str.count('z') == 0);
   EXPECT_TRUE(Str.count("helloworld") == 0);
   EXPECT_TRUE(Str.count("hello") == 1);