[ADT] Follow up to fix bug in "Add makeVisitor to STLExtras.h"

Address mistakenly comparing the pointer values of two C-style strings rather than comparing their contents in the unit tests for makeVisitor, added in 6d6f35e
llvm · Jul 1, 2021 · 83887df · 83887df
1 parent c360553
commit 83887df
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/llvm/unittests/ADT/STLExtrasTest.cpp b/llvm/unittests/ADT/STLExtrasTest.cpp
@@ -777,9 +777,9 @@ TEST(STLExtrasTest, MakeVisitorOneCallable) {
 
 TEST(STLExtrasTest, MakeVisitorTwoCallables) {
   auto Visitor =
-      makeVisitor([](int) { return "int"; }, [](std::string) { return "str"; });
-  EXPECT_EQ(Visitor(42), "int");
-  EXPECT_EQ(Visitor("foo"), "str");
+      makeVisitor([](int) { return 0; }, [](std::string) { return 1; });
+  EXPECT_EQ(Visitor(42), 0);
+  EXPECT_EQ(Visitor("foo"), 1);
 }
 
 TEST(STLExtrasTest, MakeVisitorCallableMultipleOperands) {
@@ -793,20 +793,20 @@ TEST(STLExtrasTest, MakeVisitorDefaultCase) {
   {
     auto Visitor = makeVisitor([](int I) { return I + 100; },
                                [](float F) { return F * 2; },
-                               [](auto) { return "unhandled type"; });
+                               [](auto) { return -1; });
     EXPECT_EQ(Visitor(24), 124);
     EXPECT_EQ(Visitor(2.f), 4.f);
-    EXPECT_EQ(Visitor(2.), "unhandled type");
-    EXPECT_EQ(Visitor(Visitor), "unhandled type");
+    EXPECT_EQ(Visitor(2.), -1);
+    EXPECT_EQ(Visitor(Visitor), -1);
   }
   {
-    auto Visitor = makeVisitor([](auto) { return "unhandled type"; },
+    auto Visitor = makeVisitor([](auto) { return -1; },
                                [](int I) { return I + 100; },
                                [](float F) { return F * 2; });
     EXPECT_EQ(Visitor(24), 124);
     EXPECT_EQ(Visitor(2.f), 4.f);
-    EXPECT_EQ(Visitor(2.), "unhandled type");
-    EXPECT_EQ(Visitor(Visitor), "unhandled type");
+    EXPECT_EQ(Visitor(2.), -1);
+    EXPECT_EQ(Visitor(Visitor), -1);
   }
 }