82 lines
2.3 KiB
C++
82 lines
2.3 KiB
C++
#include "../Test.h"
|
|
#include <Compiler/Resolver.h>
|
|
|
|
namespace {
|
|
|
|
using namespace clice;
|
|
|
|
std::vector<const char*> compileArgs = {
|
|
"clang++",
|
|
"-std=c++20",
|
|
"main.cpp",
|
|
"-resource-dir",
|
|
"/home/ykiko/C++/clice2/build/lib/clang/20",
|
|
};
|
|
|
|
struct Visitor : public clang::RecursiveASTVisitor<Visitor> {
|
|
clang::QualType result;
|
|
clang::QualType expect;
|
|
std::unique_ptr<ParsedAST> parsedAST;
|
|
|
|
Visitor(llvm::StringRef content) : parsedAST(ParsedAST::build("main.cpp", content, compileArgs)) {}
|
|
|
|
bool VisitTypeAliasDecl(clang::TypeAliasDecl* decl) {
|
|
if(decl->getName() == "result") {
|
|
TemplateResolver resolver(parsedAST->sema);
|
|
result = resolver.resolve(decl->getUnderlyingType());
|
|
}
|
|
|
|
if(decl->getName() == "expect") {
|
|
expect = decl->getUnderlyingType();
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void test() {
|
|
auto decl = parsedAST->context.getTranslationUnitDecl();
|
|
TraverseDecl(decl);
|
|
EXPECT_EQ(result.getCanonicalType(), expect.getCanonicalType());
|
|
}
|
|
};
|
|
|
|
void match(clang::QualType type, std::string name, std::initializer_list<std::string> args) {
|
|
auto TST = type->getAs<clang::TemplateSpecializationType>();
|
|
ASSERT_TRUE(TST);
|
|
ASSERT_EQ(TST->getTemplateName().getAsTemplateDecl()->getName(), name);
|
|
|
|
auto template_args = TST->template_arguments();
|
|
ASSERT_EQ(template_args.size(), args.size());
|
|
|
|
auto iter = args.begin();
|
|
for(auto arg: template_args) {
|
|
auto T = llvm::dyn_cast<clang::TemplateTypeParmType>(arg.getAsType());
|
|
ASSERT_TRUE(T);
|
|
ASSERT_EQ(T->getDecl()->getName(), *iter);
|
|
++iter;
|
|
}
|
|
}
|
|
|
|
TEST(clice, TemplateResolver) {
|
|
// FIXME: more flexible
|
|
auto path = test_dir() + "/TemplateResolver";
|
|
std::error_code error;
|
|
fs::directory_iterator iter(path, error);
|
|
fs::directory_iterator end;
|
|
while(!error && iter != end) {
|
|
auto file = iter->path();
|
|
llvm::outs() << "test: " << file << " " << error.message() << "\n";
|
|
auto buffer = llvm::MemoryBuffer::getFile(file);
|
|
if(!buffer) {
|
|
llvm::outs() << "failed to open file: " << buffer.getError().message() << file << "\n";
|
|
}
|
|
auto content = buffer.get()->getBuffer();
|
|
Visitor visitor(content);
|
|
visitor.test();
|
|
iter.increment(error);
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|