diff --git a/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php b/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php index bc6387f0a2683e2b7f0c4d5d83115953355d21e7..0c23c1c21a10d8aac8e779b041b5a52c00f10303 100644 --- a/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php +++ b/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php @@ -34,12 +34,15 @@ class ModulesTest extends \PHPUnit_Framework_TestCase { $this->objectManager = $this->getMockForAbstractClass('Magento\Framework\ObjectManagerInterface'); $this->status = $this->getMock('Magento\Framework\Module\Status', [], [], '', false); - $objectManagerFactory = $this->getMock('Magento\Setup\Model\ObjectManagerFactory', [], [], '', false); - $objectManagerFactory->expects($this->once())->method('create')->willReturn($this->objectManager); + /** @var + * $objectManagerProvider \PHPUnit_Framework_MockObject_MockObject|\Magento\Setup\Model\ObjectManagerProvider + */ + $objectManagerProvider = $this->getMock('Magento\Setup\Model\ObjectManagerProvider', [], [], '', false); + $objectManagerProvider->expects($this->once())->method('get')->willReturn($this->objectManager); $this->modules = $this->getMock('Magento\Setup\Model\ModuleStatus', [], [], '', false); $this->status = $this->getMock('Magento\Framework\Module\Status', [], [], '', false); $this->objectManager->expects($this->once())->method('create')->will($this->returnValue($this->status)); - $this->controller = new Modules($this->modules, $objectManagerFactory); + $this->controller = new Modules($this->modules, $objectManagerProvider); } /**