1 files changed, 8 insertions, 9 deletions
diff --git a/llvm/unittests/IR/PassManagerTest.cpp b/llvm/unittests/IR/PassManagerTest.cpp
index 7b60e38..cf7b6e4 100644
--- a/llvm/unittests/IR/PassManagerTest.cpp
+++ b/llvm/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);
 }
-
 }