public function testGetOrder() { $orderId = 100000041; $this->shipment->setOrderId($orderId); $entityName = 'shipment'; $order = $this->getMock('Magento\\Sales\\Model\\Order', ['load', 'setHistoryEntityName', '__wakeUp'], [], '', false); $this->shipment->setOrderId($orderId); $order->expects($this->atLeastOnce())->method('setHistoryEntityName')->with($entityName)->will($this->returnSelf()); $this->orderRepository->expects($this->atLeastOnce())->method('get')->will($this->returnValue($order)); $this->assertEquals($order, $this->shipment->getOrder()); }