public function testGetAddressShippingMethod() { $addressMock = $this->getMock('Magento\\Quote\\Model\\Quote\\Address', ['getShippingMethod', '__wakeup'], [], '', false); $addressMock->expects($this->once())->method('getShippingMethod')->will($this->returnValue('expected shipping method')); $this->assertEquals('expected shipping method', $this->model->getAddressShippingMethod($addressMock)); }