1 files changed, 75 insertions, 0 deletions
diff --git a/clang-tools-extra/unittests/clang-tidy/ClangTidyTest.h b/clang-tools-extra/unittests/clang-tidy/ClangTidyTest.h
new file mode 100644
index 00000000000..bd230285c43
--- /dev/null
+++ b/clang-tools-extra/unittests/clang-tidy/ClangTidyTest.h
@@ -0,0 +1,75 @@
+//===--- 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));
+    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