<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\Elasticsearch\Test\Unit\SearchAdapter\Aggregation;

use Magento\Elasticsearch\SearchAdapter\Aggregation\DataProviderFactory;
use Magento\Elasticsearch\SearchAdapter\Dynamic\DataProvider;
use Magento\Elasticsearch\SearchAdapter\QueryContainer;
use Magento\Framework\ObjectManagerInterface;
use Magento\Framework\Search\Dynamic\DataProviderInterface;
use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;

class DataProviderFactoryTest extends TestCase
{
    /**
     * @var ObjectManagerInterface|MockObject
     */
    private $objectManager;

    /**
     * @var DataProviderFactory
     */
    private $factory;

    protected function setUp(): void
    {
        $objectManager = new ObjectManager($this);
        $this->objectManager = $this->getMockBuilder(ObjectManagerInterface::class)
            ->onlyMethods(['create'])
            ->getMockForAbstractClass();
        $this->factory = $objectManager->getObject(
            DataProviderFactory::class,
            ['objectManager' => $this->objectManager]
        );
    }

    public function testCreateDataProviderWithoutQuery()
    {
        $this->objectManager->expects($this->never())->method('create');
        /** @var DataProviderInterface $dataProvider */
        $dataProvider = $this->getMockBuilder(DataProviderInterface::class)
            ->getMockForAbstractClass();
        $this->assertSame($dataProvider, $this->factory->create($dataProvider));
    }

    public function testCreateDataProviderWithEmptyQuery()
    {
        $this->objectManager->expects($this->never())->method('create');
        /** @var DataProviderInterface $dataProvider */
        $dataProvider = $this->getMockBuilder(DataProviderInterface::class)
            ->getMockForAbstractClass();
        $this->assertSame($dataProvider, $this->factory->create($dataProvider, null));
    }

    public function testCreateDataProviderWithQuery()
    {
        $this->objectManager->expects($this->never())->method('create');
        /** @var DataProviderInterface $dataProvider */
        $dataProvider = $this->getMockBuilder(DataProviderInterface::class)
            ->getMockForAbstractClass();
        /** @var MockObject $queryContainerMock */
        $queryContainerMock = $this->getMockBuilder(QueryContainer::class)
            ->onlyMethods(['getQuery'])
            ->disableOriginalConstructor()
            ->getMock();
        $queryContainerMock->expects($this->never())->method('getQuery');
        $this->assertSame($dataProvider, $this->factory->create($dataProvider, $queryContainerMock));
    }

    public function testCreateQueryAwareDataProvider()
    {
        /** @var DataProviderInterface $dataProvider */
        $dataProvider = $this->getMockBuilder(DataProvider::class)
            ->disableOriginalConstructor()
            ->getMock();
        /** @var MockObject $queryContainer */
        $queryContainer = $this->getMockBuilder(QueryContainer::class)
            ->onlyMethods(['getQuery'])
            ->disableOriginalConstructor()
            ->getMock();
        $queryContainer->expects($this->never())->method('getQuery');
        /** @var DataProviderInterface $recreatedDataProvider */
        $recreatedDataProvider = $this->getMockBuilder(DataProvider::class)
            ->disableOriginalConstructor()
            ->getMock();
        $this->objectManager->expects($this->once())
            ->method('create')
            ->with($this->isType('string'), ['queryContainer' => $queryContainer, 'aggregationFieldName' => null])
            ->willReturn($recreatedDataProvider);
        $result = $this->factory->create($dataProvider, $queryContainer);
        $this->assertNotSame($dataProvider, $result);
        $this->assertSame($recreatedDataProvider, $result);
    }

    public function testCreateContainerAwareDataProviderWithoutQuery()
    {
        $this->expectException(\LogicException::class);
        $this->expectExceptionMessage('must be configured with a search query, but the query is empty');

        $this->objectManager->expects($this->never())->method('create');
        /** @var DataProviderInterface $dataProvider */
        $dataProvider = $this->getMockBuilder(DataProvider::class)
            ->disableOriginalConstructor()
            ->getMock();
        $this->factory->create($dataProvider, null);
    }
}