clang/unittests/Tooling/RecursiveASTVisitorTests/TemplateArgumentLocTraverser.cpp

   1 //===- unittest/Tooling/RecursiveASTVisitorTests/TemplateArgumentLocTraverser.cpp -===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "TestVisitor.h"
  10
  11 using namespace clang;
  12
  13 namespace {
  14
  15 class TemplateArgumentLocTraverser
  16   : public ExpectedLocationVisitor<TemplateArgumentLocTraverser> {
  17 public:
  18   bool TraverseTemplateArgumentLoc(const TemplateArgumentLoc &ArgLoc) {
  19     std::string ArgStr;
  20     llvm::raw_string_ostream Stream(ArgStr);
  21     const TemplateArgument &Arg = ArgLoc.getArgument();
  22
  23     Arg.print(Context->getPrintingPolicy(), Stream, /*IncludeType*/ true);
  24     Match(Stream.str(), ArgLoc.getLocation());
  25     return ExpectedLocationVisitor<TemplateArgumentLocTraverser>::
  26       TraverseTemplateArgumentLoc(ArgLoc);
  27   }
  28 };
  29
  30 TEST(RecursiveASTVisitor, VisitsClassTemplateTemplateParmDefaultArgument) {
  31   TemplateArgumentLocTraverser Visitor;
  32   Visitor.ExpectMatch("X", 2, 40);
  33   EXPECT_TRUE(Visitor.runOver(
  34     "template<typename T> class X;\n"
  35     "template<template <typename> class T = X> class Y;\n"
  36     "template<template <typename> class T> class Y {};\n"));
  37 }
  38
  39 } // end anonymous namespace