Fix mock methods with new version of phpunit

fxpio · Apr 1, 2016 · 7cf57ab · 7cf57ab
1 parent 58939d3
commit 7cf57ab
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/Tests/Composer/ScriptHandlerTest.php b/Tests/Composer/ScriptHandlerTest.php
@@ -145,13 +145,17 @@ protected function createEvent($composerType)
             ->method('getType')
             ->will($this->returnValue($composerType));
 
-        $this->operation->expects($this->any())
-            ->method('getTargetPackage')
-            ->will($this->returnValue($this->package));
-
-        $this->operation->expects($this->any())
-            ->method('getPackage')
-            ->will($this->returnValue($this->package));
+        if ($this->operation instanceof UpdateOperation) {
+            $this->operation->expects($this->any())
+                ->method('getTargetPackage')
+                ->will($this->returnValue($this->package));
+        }
+
+        if ($this->operation instanceof InstallOperation) {
+            $this->operation->expects($this->any())
+                ->method('getPackage')
+                ->will($this->returnValue($this->package));
+        }
 
         /* @var PolicyInterface $policy */
         $policy = $this->getMock('Composer\DependencyResolver\PolicyInterface');