[BUGFIX] Fixed broken test.

Test was not running in EE 1.13
netz98 · May 3, 2013 · 045efc1 · 045efc1
1 parent 22a7081
commit 045efc1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/app/code/community/N98/LayoutHelper/Test/Block/Sales/Order/Info.php b/src/app/code/community/N98/LayoutHelper/Test/Block/Sales/Order/Info.php
@@ -46,6 +46,7 @@ class N98_LayoutHelper_Test_Block_Sales_Order_Info extends EcomDev_PHPUnit_Test_
     public function setUp()
     {
         parent::setUp();
+
         $infoBlock = $this->getBlockMock('payment/info');
         $paymentHelperMock = $this->getHelperMock('payment/data', array('getInfoBlock'));
 
@@ -55,7 +56,10 @@ public function setUp()
 
         $this->replaceByMock('helper', 'payment', $paymentHelperMock);
 
-        $orderMock = $this->getModelMock('sales/order', array('hasShipments', 'hasCreditmemos'));
+        $payment = $this->getModelMock('sales/order_payment');
+
+        $orderMock = $this->getModelMock('sales/order', array('hasShipments', 'hasCreditmemos', 'getPayment'));
+        $orderMock->expects($this->any())->method('getPayment')->will($this->returnValue($payment));
 
         Mage::register('current_order', $orderMock);
     }