From 222e2a7620e6520ffaf4fc4e69d79c18da31542e Mon Sep 17 00:00:00 2001 From: "Zancanaro; Carlo" Date: Mon, 24 Sep 2012 09:58:17 +1000 Subject: Add the clang library to the repo (with some of my changes, too). --- clang/unittests/Frontend/FrontendActionTest.cpp | 74 +++++++++++++++++++++++++ 1 file changed, 74 insertions(+) create mode 100644 clang/unittests/Frontend/FrontendActionTest.cpp (limited to 'clang/unittests/Frontend/FrontendActionTest.cpp') diff --git a/clang/unittests/Frontend/FrontendActionTest.cpp b/clang/unittests/Frontend/FrontendActionTest.cpp new file mode 100644 index 0000000..2d4befc --- /dev/null +++ b/clang/unittests/Frontend/FrontendActionTest.cpp @@ -0,0 +1,74 @@ +//===- unittests/Frontend/FrontendActionTest.cpp - FrontendAction tests ---===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#include "clang/AST/RecursiveASTVisitor.h" +#include "clang/AST/ASTConsumer.h" +#include "clang/Frontend/CompilerInstance.h" +#include "clang/Frontend/CompilerInvocation.h" +#include "clang/Frontend/FrontendAction.h" + +#include "llvm/ADT/Triple.h" +#include "llvm/Support/MemoryBuffer.h" + +#include "gtest/gtest.h" + +using namespace llvm; +using namespace clang; + +namespace { + +class TestASTFrontendAction : public ASTFrontendAction { +public: + std::vector decl_names; + + virtual ASTConsumer *CreateASTConsumer(CompilerInstance &CI, + StringRef InFile) { + return new Visitor(decl_names); + } + +private: + class Visitor : public ASTConsumer, public RecursiveASTVisitor { + public: + Visitor(std::vector &decl_names) : decl_names_(decl_names) {} + + virtual void HandleTranslationUnit(ASTContext &context) { + TraverseDecl(context.getTranslationUnitDecl()); + } + + virtual bool VisitNamedDecl(NamedDecl *Decl) { + decl_names_.push_back(Decl->getQualifiedNameAsString()); + return true; + } + + private: + std::vector &decl_names_; + }; +}; + +TEST(ASTFrontendAction, Sanity) { + CompilerInvocation *invocation = new CompilerInvocation; + invocation->getPreprocessorOpts().addRemappedFile( + "test.cc", MemoryBuffer::getMemBuffer("int main() { float x; }")); + invocation->getFrontendOpts().Inputs.push_back(FrontendInputFile("test.cc", + IK_CXX)); + invocation->getFrontendOpts().ProgramAction = frontend::ParseSyntaxOnly; + invocation->getTargetOpts().Triple = "i386-unknown-linux-gnu"; + CompilerInstance compiler; + compiler.setInvocation(invocation); + compiler.createDiagnostics(0, NULL); + + TestASTFrontendAction test_action; + ASSERT_TRUE(compiler.ExecuteAction(test_action)); + ASSERT_EQ(3U, test_action.decl_names.size()); + EXPECT_EQ("__builtin_va_list", test_action.decl_names[0]); + EXPECT_EQ("main", test_action.decl_names[1]); + EXPECT_EQ("x", test_action.decl_names[2]); +} + +} // anonymous namespace -- cgit v1.2.3