/** * Test display method */ public function testDisplayAndRegisterOutput() { $outputOne = $this->getMock('Magento\\Framework\\Profiler\\Driver\\Standard\\OutputInterface'); $outputOne->expects($this->once())->method('display')->with($this->_stat); $outputTwo = $this->getMock('Magento\\Framework\\Profiler\\Driver\\Standard\\OutputInterface'); $outputTwo->expects($this->once())->method('display')->with($this->_stat); $this->_driver->registerOutput($outputOne); $this->_driver->registerOutput($outputTwo); \Magento\Framework\Profiler::enable(); $this->_driver->display(); \Magento\Framework\Profiler::disable(); $this->_driver->display(); }