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

namespace Magento\Framework\Indexer\Test\Unit;

use Magento\Framework\DB\Adapter\AdapterInterface;
use Magento\Framework\Indexer\BatchSizeManagement;
use Magento\Framework\Indexer\IndexTableRowSizeEstimatorInterface;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;
use Psr\Log\LoggerInterface;

class BatchSizeManagementTest extends TestCase
{
    /**
     * @var BatchSizeManagement
     */
    private $model;

    /**
     * @var IndexTableRowSizeEstimatorInterface|MockObject
     */
    private $rowSizeEstimatorMock;

    /**
     * @var LoggerInterface|MockObject
     */
    private $loggerMock;

    /**
     * @inheritdoc
     */
    protected function setUp(): void
    {
        $this->rowSizeEstimatorMock = $this->createMock(
            IndexTableRowSizeEstimatorInterface::class
        );
        $this->loggerMock = $this->getMockForAbstractClass(LoggerInterface::class);
        $this->model = new BatchSizeManagement($this->rowSizeEstimatorMock, $this->loggerMock);
    }

    /**
     * @return void
     */
    public function testEnsureBatchSize(): void
    {
        $batchSize = 200;
        $maxHeapTableSize = 16384;
        $tmpTableSize = 16384;
        $size = 20000;
        $innodbPollSize = 100;

        $this->rowSizeEstimatorMock->expects($this->once())->method('estimateRowSize')->willReturn(100);
        $adapterMock = $this->getMockForAbstractClass(AdapterInterface::class);

        $this->loggerMock->expects($this->once())
            ->method('warning')
            ->with(__(
                "Memory size allocated for the temporary table is more than 20% of innodb_buffer_pool_size. " .
                "Please update innodb_buffer_pool_size or decrease batch size value " .
                "(which decreases memory usages for the temporary table). " .
                "Current batch size: %1; Allocated memory size: %2 bytes; InnoDB buffer pool size: %3 bytes.",
                [$batchSize, $size, $innodbPollSize]
            ));

        $adapterMock
            ->method('fetchOne')
            ->willReturnCallback(
                function ($arg1, $arg2) use ($maxHeapTableSize, $tmpTableSize, $innodbPollSize) {
                    if ($arg1 == 'SELECT @@max_heap_table_size;' && empty($arg2)) {
                        return $maxHeapTableSize;
                    } elseif ($arg1 == 'SELECT @@tmp_table_size;' && empty($arg2)) {
                        return $tmpTableSize;
                    } elseif ($arg1 == 'SELECT @@innodb_buffer_pool_size;' && empty($arg2)) {
                        return $innodbPollSize;
                    }
                }
            );

        $adapterMock
            ->method('query')
            ->willReturnCallback(
                function ($arg1, $arg2) use ($size) {
                    if ($arg1 == 'SET SESSION tmp_table_size = ' . $size . ';' && empty($arg2)) {
                        return null;
                    } elseif ($arg1 == 'SET SESSION max_heap_table_size = ' . $size . ';' && empty($arg2)) {
                        return null;
                    }
                }
            );

        $this->model->ensureBatchSize($adapterMock, $batchSize);
    }
}