Your IP : 216.73.216.220
<?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);
}
}
}
}