Server : Apache System : Linux server2.corals.io 4.18.0-348.2.1.el8_5.x86_64 #1 SMP Mon Nov 15 09:17:08 EST 2021 x86_64 User : corals ( 1002) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system Directory :  /home/corals/old/dev/tests/setup-integration/testsuite/Magento/Setup/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Magento\Setup;

use Magento\Framework\App\DeploymentConfig;
use Magento\Framework\App\ResourceConnection;
use Magento\TestFramework\Deploy\CliCommand;
use Magento\TestFramework\Deploy\DescribeTable;
use Magento\TestFramework\Deploy\ShardingConfig;
use Magento\TestFramework\Helper\Bootstrap;
use Magento\TestFramework\TestCase\SetupTestCase;
use Magento\TestFramework\Deploy\TestModuleManager;

/**
 * The purpose of this test is verifying declarative installation works with different shard.
 */
class ShardingTest extends SetupTestCase
{
    /**
     * @var CliCommand
     */
    private $cliCommand;

    /**
     * @var DescribeTable
     */
    private $describeTable;

    /**
     * @var ResourceConnection
     */
    private $resourceConnection;

    /**
     * @var DeploymentConfig
     */
    private $deploymentConfig;

    /**
     * @var  TestModuleManager
     */
    private $moduleManager;

    /**
     * @var ShardingConfig
     */
    private $shardingConfig;

    protected function setUp(): void
    {
        $objectManager= Bootstrap::getObjectManager();
        $this->cliCommand = $objectManager->get(CliCommand::class);
        $this->resourceConnection = $objectManager->get(ResourceConnection::class);
        $this->deploymentConfig = $objectManager->get(DeploymentConfig::class);
        $this->describeTable = $objectManager->get(DescribeTable::class);
        $this->moduleManager = $objectManager->get(TestModuleManager::class);
        $this->shardingConfig = $objectManager->get(ShardingConfig::class);
    }

    /**
     * @moduleName Magento_TestSetupDeclarationModule2
     * @dataProviderFromFile Magento/TestSetupDeclarationModule2/fixture/shards.php
     */
    public function testInstall()
    {
        $this->shardingConfig->applyConfiguration();
        $this->cliCommand->install(['Magento_TestSetupDeclarationModule2']);
        $this->deploymentConfig->resetData();

        $default = $this->describeTable->describeShard('default');
        $shard1 = $this->describeTable->describeShard('shard_one');
        $shard2 = $this->describeTable->describeShard('shard_two');
        //Check if tables were installed on different shards
        self::assertCount(2, $default);
        self::assertCount(1, $shard1);
        self::assertCount(1, $shard2);
        $this->assertTableCreationStatements($this->getData(), array_replace($default, $shard1, $shard2));
    }

    /**
     * @moduleName Magento_TestSetupDeclarationModule2
     * @dataProviderFromFile Magento/TestSetupDeclarationModule2/fixture/shards.php
     */
    public function testUpgrade()
    {
        $this->shardingConfig->applyConfiguration();
        $this->cliCommand->install(['Magento_TestSetupDeclarationModule2']);
        $this->deploymentConfig->resetData();
        $this->cliCommand->upgrade();

        $default = $this->describeTable->describeShard('default');
        $shard1 = $this->describeTable->describeShard('shard_one');
        $shard2 = $this->describeTable->describeShard('shard_two');
        //Check if tables were installed on different shards
        self::assertCount(2, $default);
        self::assertCount(1, $shard1);
        self::assertCount(1, $shard2);
        $this->assertTableCreationStatements($this->getData(), array_replace($default, $shard1, $shard2));
    }

    /**
     * Assert table creation statements
     *
     * @param array $expectedData
     * @param array $actualData
     */
    private function assertTableCreationStatements(array $expectedData, array $actualData): void
    {
        if (!$this->isUsingAuroraDb()) {
            $this->assertEquals($expectedData, $actualData);
        } else {
            ksort($expectedData);
            ksort($actualData);
            $this->assertSameSize($expectedData, $actualData);
            foreach ($expectedData as $key => $value) {
                $this->assertStringContainsString($actualData[$key], $value);
            }
        }
    }
}