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());
}