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