#7760: M2.1.2 : Shipment Tracking REST API should throw an error if o…

…rder doesn't exist.
magento · Jan 4, 2018 · a732ff6 · a732ff6
1 parent 1b3304a
commit a732ff6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/app/code/Magento/Sales/Test/Unit/Model/ResourceModel/Order/Shipment/TrackTest.php b/app/code/Magento/Sales/Test/Unit/Model/ResourceModel/Order/Shipment/TrackTest.php
@@ -89,6 +89,8 @@ protected function setUp()
      */
     public function testSave()
     {
+        $shipmentMock = $this->createMock(\Magento\Sales\Model\Order\Shipment::class);
+        $orderMock = $this->createMock(\Magento\Sales\Model\Order::class);
         $this->entitySnapshotMock->expects($this->once())
             ->method('isModified')
             ->with($this->trackModelMock)
@@ -98,6 +100,8 @@ public function testSave()
             ->with($this->equalTo($this->trackModelMock))
             ->will($this->returnValue([]));
         $this->trackModelMock->expects($this->any())->method('getData')->willReturn([]);
+        $this->trackModelMock->expects($this->atLeastOnce())->method('getShipment')->willReturn($shipmentMock);
+        $shipmentMock->expects($this->atLeastOnce())->method('getOrder')->willReturn($orderMock);
         $this->trackResource->save($this->trackModelMock);
         $this->assertTrue(true);
     }