Yuri Kuznetsov
137 lines
4.4 KiB
PHP
137 lines
4.4 KiB
PHP
<?php
|
||
/************************************************************************
|
||
* This file is part of EspoCRM.
|
||
*
|
||
* EspoCRM – Open Source CRM application.
|
||
* Copyright (C) 2014-2025 EspoCRM, Inc.
|
||
* Website: https://www.espocrm.com
|
||
*
|
||
* This program is free software: you can redistribute it and/or modify
|
||
* it under the terms of the GNU Affero General Public License as published by
|
||
* the Free Software Foundation, either version 3 of the License, or
|
||
* (at your option) any later version.
|
||
*
|
||
* This program is distributed in the hope that it will be useful,
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
* GNU Affero General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU Affero General Public License
|
||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
*
|
||
* The interactive user interfaces in modified source and object code versions
|
||
* of this program must display Appropriate Legal Notices, as required under
|
||
* Section 5 of the GNU Affero General Public License version 3.
|
||
*
|
||
* In accordance with Section 7(b) of the GNU Affero General Public License version 3,
|
||
* these Appropriate Legal Notices must retain the display of the "EspoCRM" word.
|
||
************************************************************************/
|
||
|
||
namespace tests\unit\Espo\ORM;
|
||
|
||
use Espo\ORM\EntityFactory;
|
||
use Espo\ORM\Locker\BaseLocker;
|
||
use Espo\ORM\Metadata;
|
||
use Espo\ORM\QueryComposer\MysqlQueryComposer;
|
||
use Espo\ORM\TransactionManager;
|
||
|
||
use PDO;
|
||
use PHPUnit\Framework\TestCase;
|
||
|
||
class BaseLockerTest extends TestCase
|
||
{
|
||
private $transactionManager;
|
||
private $locker;
|
||
private $pdo;
|
||
|
||
protected function setUp() : void
|
||
{
|
||
$this->pdo = $this->createMock(PDO::class);
|
||
|
||
$entityFactory = $this->createMock(EntityFactory::class);
|
||
|
||
$metadata = $this->getMockBuilder(Metadata::class)->disableOriginalConstructor()->getMock();
|
||
|
||
$this->transactionManager = $this->getMockBuilder(TransactionManager::class)
|
||
->disableOriginalConstructor()->getMock();
|
||
|
||
$composer = new MysqlQueryComposer($this->pdo, $entityFactory, $metadata);
|
||
|
||
$this->locker = new BaseLocker($this->pdo, $composer, $this->transactionManager);
|
||
}
|
||
|
||
public function testLockCommit()
|
||
{
|
||
$this->transactionManager
|
||
->expects($this->exactly(2))
|
||
->method('start');
|
||
|
||
$invokedCount = $this->exactly(2);
|
||
|
||
$this->pdo
|
||
->expects($invokedCount)
|
||
->method('exec')
|
||
->willReturnCallback(function ($sql) use ($invokedCount) {
|
||
if ($invokedCount->numberOfInvocations() === 1) {
|
||
$this->assertEquals('LOCK TABLES `account` WRITE', $sql);
|
||
}
|
||
|
||
if ($invokedCount->numberOfInvocations() === 2) {
|
||
$this->assertEquals('LOCK TABLES `contact` READ', $sql);
|
||
}
|
||
|
||
return 1;
|
||
});
|
||
|
||
$this->transactionManager
|
||
->expects($this->once())
|
||
->method('commit');
|
||
|
||
$this->locker->lockExclusive('Account');
|
||
$this->locker->lockShare('Contact');
|
||
|
||
$this->assertTrue($this->locker->isLocked());
|
||
|
||
$this->locker->commit();
|
||
|
||
$this->assertFalse($this->locker->isLocked());
|
||
}
|
||
|
||
public function testLockRollback()
|
||
{
|
||
$this->transactionManager
|
||
->expects($this->exactly(2))
|
||
->method('start');
|
||
|
||
$invokedCount = $this->exactly(2);
|
||
|
||
$this->pdo
|
||
->expects($invokedCount)
|
||
->method('exec')
|
||
->willReturnCallback(function ($sql) use ($invokedCount) {
|
||
if ($invokedCount->numberOfInvocations() === 1) {
|
||
$this->assertEquals('LOCK TABLES `account` WRITE', $sql);
|
||
}
|
||
|
||
if ($invokedCount->numberOfInvocations() === 2) {
|
||
$this->assertEquals('LOCK TABLES `contact` READ', $sql);
|
||
}
|
||
|
||
return 1;
|
||
});
|
||
|
||
$this->transactionManager
|
||
->expects($this->once())
|
||
->method('rollback');
|
||
|
||
$this->locker->lockExclusive('Account');
|
||
$this->locker->lockShare('Contact');
|
||
|
||
$this->assertTrue($this->locker->isLocked());
|
||
|
||
$this->locker->rollback();
|
||
|
||
$this->assertFalse($this->locker->isLocked());
|
||
}
|
||
}
|