Your IP : 216.73.216.220
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
declare(strict_types=1);
namespace Magento\Framework\EntityManager\Test\Unit\Db;
use Magento\Framework\App\ResourceConnection;
use Magento\Framework\DB\Adapter\AdapterInterface;
use Magento\Framework\EntityManager\Db\UpdateRow;
use Magento\Framework\EntityManager\EntityMetadataInterface;
use Magento\Framework\EntityManager\MetadataPool;
use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;
class UpdateRowTest extends TestCase
{
/**
* @var UpdateRow
*/
protected $model;
/**
* @var MetadataPool|MockObject
*/
protected $metadataPoolMock;
/**
* @var ResourceConnection|MockObject
*/
protected $resourceConnectionMock;
/**
* @var AdapterInterface|MockObject
*/
protected $connectionMock;
/**
* @var EntityMetadataInterface|MockObject
*/
protected $metadataMock;
protected function setUp(): void
{
$this->metadataPoolMock = $this->getMockBuilder(MetadataPool::class)
->disableOriginalConstructor()
->getMock();
$this->resourceConnectionMock = $this->getMockBuilder(ResourceConnection::class)
->disableOriginalConstructor()
->getMock();
$this->metadataMock = $this->getMockBuilder(EntityMetadataInterface::class)
->getMockForAbstractClass();
$this->connectionMock = $this->getMockBuilder(AdapterInterface::class)
->getMockForAbstractClass();
$this->metadataMock = $this->getMockBuilder(EntityMetadataInterface::class)
->getMockForAbstractClass();
$this->model = (new ObjectManager($this))->getObject(UpdateRow::class, [
'metadataPool' => $this->metadataPoolMock,
'resourceConnection' => $this->resourceConnectionMock,
]);
}
/**
* @dataProvider columnsDataProvider
* @param array $data
* @param array $columns
* @param array $preparedColumns
*/
public function testExecute(array $data, array $columns, array $preparedColumns)
{
$primaryKeyName = 'entity_id';
$this->metadataPoolMock->expects($this->once())
->method('getMetadata')
->with('test')
->willReturn($this->metadataMock);
$this->resourceConnectionMock->expects($this->once())
->method('getConnectionByName')
->willReturn($this->connectionMock);
$this->metadataMock->expects($this->once())
->method('getEntityConnectionName')
->willReturn('test_connection_name');
$this->metadataMock->expects($this->atLeastOnce())
->method('getEntityTable')
->willReturn('test_entity_table');
$this->connectionMock->expects($this->once())
->method('update')
->with('test_entity_table', $preparedColumns, ['test_link_field' . ' = ?' => $data['test_link_field']]);
$this->connectionMock->expects($this->once())->method('getIndexList')
->willReturn([$primaryKeyName => ['COLUMNS_LIST' => ['test_link_field']]]);
$this->connectionMock->expects($this->once())->method('getPrimaryKeyName')
->willReturn($primaryKeyName);
$this->connectionMock->expects($this->once())
->method('describeTable')
->willReturn($columns);
$this->metadataMock->expects($this->exactly(2))
->method('getIdentifierField')
->willReturn('test_identified_field');
if (empty($data['updated_at'])) {
unset($data['updated_at']);
}
$this->assertSame($data, $this->model->execute('test', $data));
}
/**
* @return array
*/
public function columnsDataProvider()
{
$data = [
'test_link_field' => 1,
'identified_field' => 'test_identified_field',
'test_simple' => 'test_value',
];
$columns = [
'test_nullable' => [
'NULLABLE' => true,
'DEFAULT' => false,
'IDENTITY' => false,
'COLUMN_NAME' => 'test_nullable',
],
'test_simple' => [
'NULLABLE' => true,
'DEFAULT' => false,
'IDENTITY' => false,
'COLUMN_NAME' => 'test_simple',
],
];
$preparedColumns = [
'test_identified_field' => null,
'test_nullable' => null,
'test_simple' => 'test_value',
];
return [
'default' => [
'data' => $data,
'columns' => $columns,
'preparedColumns' => $preparedColumns,
],
'empty timestamp field' => [
'data' => array_merge($data, ['updated_at' => '']),
'columns' => array_merge(
$columns,
[
'updated_at' => [
'NULLABLE' => false,
'DEFAULT' => 'CURRENT_TIMESTAMP',
'IDENTITY' => false,
'COLUMN_NAME' => 'updated_at',
],
]
),
'preparedColumns' => $preparedColumns,
],
'filled timestamp field' => [
'data' => array_merge($data, ['updated_at' => '2016-01-01 00:00:00']),
'columns' => array_merge(
$columns,
[
'updated_at' => [
'NULLABLE' => false,
'DEFAULT' => 'CURRENT_TIMESTAMP',
'IDENTITY' => false,
'COLUMN_NAME' => 'updated_at',
],
]
),
'preparedColumns' => array_merge($preparedColumns, ['updated_at' => '2016-01-01 00:00:00']),
],
];
}
}