commit b15c9d7dd22d3ffd7046397172902f11f71e4fe7
author Stephen Kelly <steveire@gmail.com> Thu Apr 15 23:27:49 2021 +0100
committer Copybara-Service <copybara-worker@google.com> Fri Apr 16 12:16:33 2021 -0700
tree b9257ce2ad427b159cfd838094bbc63bfcd68536
parent 4f2ea5884799ab1972bfc73101628850f9382e15

NFC: Add a simple test for introspection call formatting

GitOrigin-RevId: f62ad15cd7df0ca7681e0dbb894ee1c1d2465c51

unittests/Introspection/IntrospectionTest.cpp

diff --git a/unittests/Introspection/IntrospectionTest.cpp b/unittests/Introspection/IntrospectionTest.cpp
index be58945..ad21748 100644
--- a/unittests/Introspection/IntrospectionTest.cpp
+++ b/unittests/Introspection/IntrospectionTest.cpp
@@ -45,6 +45,43 @@
 
 #define STRING_LOCATION_PAIR(INSTANCE, LOC) Pair(#LOC, INSTANCE->LOC)
 
+/**
+  A test formatter for a hypothetical language which needs
+  neither casts nor '->'.
+*/
+class LocationCallFormatterSimple {
+public:
+  static void print(const LocationCall &Call, llvm::raw_ostream &OS) {
+    if (Call.isCast()) {
+      if (const LocationCall *On = Call.on())
+        print(*On, OS);
+      return;
+    }
+    if (const LocationCall *On = Call.on()) {
+      print(*On, OS);
+      OS << '.';
+    }
+    OS << Call.name();
+    if (Call.args().empty()) {
+      OS << "()";
+      return;
+    }
+    OS << '(' << Call.args().front();
+    for (const std::string &Arg : Call.args().drop_front()) {
+      OS << ", " << Arg;
+    }
+    OS << ')';
+  }
+
+  static std::string format(const LocationCall &Call) {
+    std::string Result;
+    llvm::raw_string_ostream OS(Result);
+    print(Call, OS);
+    OS.flush();
+    return Result;
+  }
+};
+
 TEST(Introspection, SourceLocations_CallContainer) {
   SourceLocationMap slm;
   SharedLocationCall Prefix;
@@ -70,6 +107,24 @@
             "getTypeLoc().getSourceRange()");
 }
 
+TEST(Introspection, SourceLocations_Formatter) {
+  SharedLocationCall Prefix;
+  auto chainedCall = llvm::makeIntrusiveRefCnt<LocationCall>(
+      llvm::makeIntrusiveRefCnt<LocationCall>(
+          llvm::makeIntrusiveRefCnt<LocationCall>(
+              llvm::makeIntrusiveRefCnt<LocationCall>(
+                  Prefix, "getTypeSourceInfo", LocationCall::ReturnsPointer),
+              "getTypeLoc"),
+          "getAs<clang::TypeSpecTypeLoc>", LocationCall::IsCast),
+      "getNameLoc");
+
+  EXPECT_EQ("getTypeSourceInfo()->getTypeLoc().getAs<clang::TypeSpecTypeLoc>()."
+            "getNameLoc()",
+            LocationCallFormatterCpp::format(*chainedCall));
+  EXPECT_EQ("getTypeSourceInfo().getTypeLoc().getNameLoc()",
+            LocationCallFormatterSimple::format(*chainedCall));
+}
+
 TEST(Introspection, SourceLocations_Stmt) {
   if (!NodeIntrospection::hasIntrospectionSupport())
     return;