[PM] Split the analysis manager into a function-specific interface and

[oota-llvm.git] / unittests / IR / PassManagerTest.cpp

diff --git a/unittests/IR/PassManagerTest.cpp b/unittests/IR/PassManagerTest.cpp
index 7b60e3899a0ef4ebfa8d69b7387bf7429acc9fc9..cf7b6e47488f0eeb7e7229762eeff1530e74a0d5 100644 (file)
--- a/unittests/IR/PassManagerTest.cpp
+++ b/unittests/IR/PassManagerTest.cpp
@@ -61,9 +61,9 @@ struct TestModulePass {
 };
 
 struct TestFunctionPass {
-  TestFunctionPass(AnalysisManager &AM, int &RunCount, int &AnalyzedInstrCount)
-      : AM(AM), RunCount(RunCount), AnalyzedInstrCount(AnalyzedInstrCount) {
-  }
+  TestFunctionPass(FunctionAnalysisManager &AM, int &RunCount,
+                   int &AnalyzedInstrCount)
+      : AM(AM), RunCount(RunCount), AnalyzedInstrCount(AnalyzedInstrCount) {}
 
   bool run(Function *F) {
     ++RunCount;
@@ -74,7 +74,7 @@ struct TestFunctionPass {
     return true;
   }
 
-  AnalysisManager &AM;
+  FunctionAnalysisManager &AM;
   int &RunCount;
   int &AnalyzedInstrCount;
 };
@@ -106,10 +106,10 @@ public:
 };
 
 TEST_F(PassManagerTest, Basic) {
-  AnalysisManager AM(M.get());
-  AM.registerAnalysisPass(TestAnalysisPass());
+  FunctionAnalysisManager AM;
+  AM.registerPass(TestAnalysisPass());
 
-  ModulePassManager MPM(M.get(), &AM);
+  ModulePassManager MPM;
   FunctionPassManager FPM(&AM);
 
   // Count the runs over a module.
@@ -122,10 +122,9 @@ TEST_F(PassManagerTest, Basic) {
   FPM.addPass(TestFunctionPass(AM, FunctionPassRunCount, AnalyzedInstrCount));
   MPM.addPass(FPM);
 
-  MPM.run();
+  MPM.run(M.get());
   EXPECT_EQ(1, ModulePassRunCount);
   EXPECT_EQ(3, FunctionPassRunCount);
   EXPECT_EQ(5, AnalyzedInstrCount);
 }
-
 }