public function testLoadOrderId() { $this->loader->unsetData('shipment_id'); $orderMock = $this->getMockBuilder('Magento\\Sales\\Model\\Order')->disableOriginalConstructor()->setMethods(['getForcedShipmentWithInvoice', 'getId', 'canShip'])->getMock(); $this->orderRepository->expects($this->once())->method('get')->will($this->returnValue($orderMock)); $orderMock->expects($this->once())->method('getId')->will($this->returnValue($this->loader->getOrderId())); $orderMock->expects($this->any())->method('getForcedShipmentWithInvoice')->will($this->returnValue(false)); $orderMock->expects($this->once())->method('canShip')->will($this->returnValue(true)); $shipmentModelMock = $this->getMockBuilder('Magento\\Sales\\Model\\Order\\Shipment')->disableOriginalConstructor()->setMethods([])->getMock(); $this->shipmentFactory->expects($this->once())->method('create')->with($orderMock, $this->loader->getShipment()['items'])->willReturn($shipmentModelMock); $trackMock = $this->getMockBuilder('Magento\\Sales\\Model\\Order\\Shipment\\Track')->disableOriginalConstructor()->setMethods([])->getMock(); $this->trackFactoryMock->expects($this->any())->method('create')->will($this->returnValue($trackMock)); $trackMock->expects($this->any())->method('addData')->will($this->returnValueMap([[$this->loader->getTracking()[0], $trackMock], [$this->loader->getTracking()[1], $trackMock]])); $shipmentModelMock->expects($this->any())->method('addTrack')->with($this->equalTo($trackMock))->will($this->returnSelf()); $this->registryMock->expects($this->once())->method('register')->with('current_shipment', $shipmentModelMock); $this->assertEquals($shipmentModelMock, $this->loader->load()); }