Import default template args

Ericsson · Mar 21, 2018 · a7ba003 · a7ba003
1 parent 6ef11d8
commit a7ba003
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 1 deletion.
diff --git a/lib/AST/ASTImporter.cpp b/lib/AST/ASTImporter.cpp
@@ -4174,7 +4174,7 @@ Decl *ASTNodeImporter::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
   // is created.
 
   // FIXME: Import default argument.
-  return TemplateTypeParmDecl::Create(Importer.getToContext(),
+  auto ToD = TemplateTypeParmDecl::Create(Importer.getToContext(),
                               Importer.getToContext().getTranslationUnitDecl(),
                                       Importer.Import(D->getLocStart()),
                                       Importer.Import(D->getLocation()),
@@ -4183,6 +4183,13 @@ Decl *ASTNodeImporter::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
                                       Importer.Import(D->getIdentifier()),
                                       D->wasDeclaredWithTypename(),
                                       D->isParameterPack());
+  Importer.Imported(D, ToD);
+  if (D->hasDefaultArgument()) {
+    TypeSourceInfo *ToDefaultArgInfo =
+        Importer.Import(D->getDefaultArgumentInfo());
+    ToD->setDefaultArgument(ToDefaultArgInfo);
+  }
+  return ToD;
 }
 
 Decl *

diff --git a/unittests/AST/ASTImporterTest.cpp b/unittests/AST/ASTImporterTest.cpp
@@ -2392,5 +2392,55 @@ long b = strtol(&a);
   EXPECT_EQ(DeclCounter<UsingShadowDecl>().match(ToTU, usingShadowDecl()), 1u);
 }
 
+TEST_F(Fixture, ShouldImportDefaultTemplateArgs) {
+  Decl *FromTU =
+      getTuDecl("template <class T=int> struct X{};", Lang_CXX, "input0.cc");
+  auto FromD =
+      FirstDeclMatcher<ClassTemplateDecl>().match(FromTU, classTemplateDecl());
+
+  {
+    auto TPL = FromD->getTemplateParameters();
+    ASSERT_EQ((TPL->end() - TPL->begin()), 1);
+    TemplateTypeParmDecl *ParmD = cast<TemplateTypeParmDecl>(*(TPL->begin()));
+    ASSERT_TRUE(ParmD->hasDefaultArgument());
+  }
+
+  {
+    auto ToD = cast<ClassTemplateDecl>(Import(FromD, Lang_CXX));
+    auto TPL = ToD->getTemplateParameters();
+    ASSERT_EQ((TPL->end() - TPL->begin()), 1);
+    TemplateTypeParmDecl *ParmD = cast<TemplateTypeParmDecl>(*(TPL->begin()));
+    EXPECT_TRUE(ParmD->hasDefaultArgument());
+  }
+}
+
+TEST_F(Fixture, ShouldImportDefaultTemplateArgs2) {
+  Decl *FromTU = getTuDecl("template <class T0=int, class T1=char> struct X{};",
+                           Lang_CXX, "input0.cc");
+  auto FromD =
+      FirstDeclMatcher<ClassTemplateDecl>().match(FromTU, classTemplateDecl());
+
+  {
+    auto TPL = FromD->getTemplateParameters();
+    ASSERT_EQ((TPL->end() - TPL->begin()), 2);
+    auto it = TPL->begin();
+    TemplateTypeParmDecl *ParmD0 = cast<TemplateTypeParmDecl>(*it++);
+    ASSERT_TRUE(ParmD0->hasDefaultArgument());
+    TemplateTypeParmDecl *ParmD1 = cast<TemplateTypeParmDecl>(*it);
+    ASSERT_TRUE(ParmD1->hasDefaultArgument());
+  }
+
+  {
+    auto ToD = cast<ClassTemplateDecl>(Import(FromD, Lang_CXX));
+    auto TPL = ToD->getTemplateParameters();
+    ASSERT_EQ((TPL->end() - TPL->begin()), 2);
+    auto it = TPL->begin();
+    TemplateTypeParmDecl *ParmD0 = cast<TemplateTypeParmDecl>(*it++);
+    EXPECT_TRUE(ParmD0->hasDefaultArgument());
+    TemplateTypeParmDecl *ParmD1 = cast<TemplateTypeParmDecl>(*it);
+    EXPECT_TRUE(ParmD1->hasDefaultArgument());
+  }
+}
+
 } // end namespace ast_matchers
 } // end namespace clang