FriendsOfSymfony / FOSElasticaBundle

 */

class ElasticaDataCollectorTest extends TestCase

{

    public function testCorrectAmountOfQueries()

    {

        /** @var $requestMock \PHPUnit_Framework_MockObject_MockObject|Request */

        $requestMock = $this->createMock(Request::class);

        /** @var $responseMock \PHPUnit_Framework_MockObject_MockObject|Response */

        $responseMock = $this->createMock(Response::class);

        /** @var $loggerMock \PHPUnit_Framework_MockObject_MockObject|ElasticaLogger */

        $loggerMock = $this->createMock(ElasticaLogger::class);

        $totalQueries = rand();

        $loggerMock->expects($this->once())

            ->method('getNbQueries')

            ->will($this->returnValue($totalQueries));

        $elasticaDataCollector = new ElasticaDataCollector($loggerMock);

        $elasticaDataCollector->collect($requestMock, $responseMock);

        $this->assertSame($totalQueries, $elasticaDataCollector->getQueryCount());

    }

    public function testCorrectQueriesReturned()

    {

        $this->assertSame($totalQueries, $elasticaDataCollector->getQueryCount());

    }

    public function testCorrectQueriesReturned()

    {

        /** @var $requestMock \PHPUnit_Framework_MockObject_MockObject|Request */

        $requestMock = $this->createMock(Request::class);

        /** @var $responseMock \PHPUnit_Framework_MockObject_MockObject|Response */

        $responseMock = $this->createMock(Response::class);

        /** @var $loggerMock \PHPUnit_Framework_MockObject_MockObject|ElasticaLogger */

        $loggerMock = $this->createMock(ElasticaLogger::class);

        $queries = ['testQueries'];

        $loggerMock->expects($this->once())

            ->method('getQueries')

            ->will($this->returnValue($queries));

        $elasticaDataCollector = new ElasticaDataCollector($loggerMock);

        $elasticaDataCollector->collect($requestMock, $responseMock);

        $this->assertSame($queries, $elasticaDataCollector->getQueries());

    }

    public function testCorrectQueriesTime()

    {