diff --git a/Tests/DataCollector/ElasticaDataCollectorTest.php b/Tests/DataCollector/ElasticaDataCollectorTest.php index b83a81c..7ec9800 100644 --- a/Tests/DataCollector/ElasticaDataCollectorTest.php +++ b/Tests/DataCollector/ElasticaDataCollectorTest.php @@ -60,4 +60,29 @@ class ElasticaDataCollectorTest extends \PHPUnit_Framework_TestCase $this->assertEquals($queries, $elasticaDataCollector->getQueries()); } + public function testCorrectQueriesTime() + { + $requestMock = $this->getMockBuilder('Symfony\Component\HttpFoundation\Request') + ->disableOriginalConstructor() + ->getMock(); + + $responseMock = $this->getMockBuilder('Symfony\Component\HttpFoundation\Response') + ->disableOriginalConstructor() + ->getMock(); + + $loggerMock = $this->getMockBuilder('FOQ\ElasticaBundle\Logger\ElasticaLogger') + ->disableOriginalConstructor() + ->getMock(); + + $queries = array(array('executionMS' => 10), array('executionMS' => 20)); + + $loggerMock->expects($this->once()) + ->method('getQueries') + ->will($this->returnValue($queries)); + + $elasticaDataCollector = new ElasticaDataCollector($loggerMock); + $elasticaDataCollector->collect($requestMock, $responseMock); + $this->assertEquals(30, $elasticaDataCollector->getTime()); + } + }