125 lines
4.0 KiB
PHP
Executable File
125 lines
4.0 KiB
PHP
Executable File
<?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);
|
|
}
|
|
}
|
|
}
|
|
}
|