forked from OSchip/llvm-project
77 lines
2.5 KiB
C++
77 lines
2.5 KiB
C++
//===--- ClangTidyTest.h - clang-tidy ---------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_TOOLS_EXTRA_UNITTESTS_CLANG_TIDY_CLANG_TIDY_TEST_H
|
|
#define LLVM_CLANG_TOOLS_EXTRA_UNITTESTS_CLANG_TIDY_CLANG_TIDY_TEST_H
|
|
|
|
#include "ClangTidy.h"
|
|
#include "ClangTidyDiagnosticConsumer.h"
|
|
#include "clang/ASTMatchers/ASTMatchFinder.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/FrontendActions.h"
|
|
#include "clang/Tooling/Refactoring.h"
|
|
#include "clang/Tooling/Tooling.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace clang {
|
|
namespace tidy {
|
|
|
|
template <typename T> class ClangTidyTest : public ::testing::Test {
|
|
protected:
|
|
ClangTidyTest() : Check(new T), Context(&Errors) {}
|
|
|
|
std::string runCheckOn(StringRef Code) {
|
|
ClangTidyDiagnosticConsumer DiagConsumer(Context);
|
|
Check->setContext(&Context);
|
|
EXPECT_TRUE(
|
|
tooling::runToolOnCode(new TestPPAction(*Check, &Context), Code));
|
|
ast_matchers::MatchFinder Finder;
|
|
Check->registerMatchers(&Finder);
|
|
OwningPtr<tooling::FrontendActionFactory> Factory(
|
|
tooling::newFrontendActionFactory(&Finder));
|
|
EXPECT_TRUE(tooling::runToolOnCode(Factory->create(), Code));
|
|
DiagConsumer.finish();
|
|
tooling::Replacements Fixes;
|
|
for (SmallVector<ClangTidyError, 16>::const_iterator I = Errors.begin(),
|
|
E = Errors.end();
|
|
I != E; ++I)
|
|
Fixes.insert(I->Fix.begin(), I->Fix.end());
|
|
return tooling::applyAllReplacements(Code, Fixes);
|
|
}
|
|
|
|
void expectNoChanges(StringRef Code) { EXPECT_EQ(Code, runCheckOn(Code)); }
|
|
|
|
private:
|
|
class TestPPAction : public PreprocessOnlyAction {
|
|
public:
|
|
TestPPAction(ClangTidyCheck &Check, ClangTidyContext *Context)
|
|
: Check(Check), Context(Context) {}
|
|
|
|
private:
|
|
virtual bool BeginSourceFileAction(CompilerInstance &Compiler,
|
|
llvm::StringRef file_name) {
|
|
Context->setSourceManager(&Compiler.getSourceManager());
|
|
Check.registerPPCallbacks(Compiler);
|
|
return true;
|
|
}
|
|
|
|
ClangTidyCheck &Check;
|
|
ClangTidyContext *Context;
|
|
};
|
|
|
|
OwningPtr<ClangTidyCheck> Check;
|
|
SmallVector<ClangTidyError, 16> Errors;
|
|
ClangTidyContext Context;
|
|
};
|
|
|
|
} // namespace tidy
|
|
} // namespace clang
|
|
|
|
#endif // LLVM_CLANG_TOOLS_EXTRA_UNITTESTS_CLANG_TIDY_CLANG_TIDY_TEST_H
|