-
Notifications
You must be signed in to change notification settings - Fork 3.4k
/
TransactionFixtureStrategy.php
89 lines (79 loc) · 2.73 KB
/
TransactionFixtureStrategy.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
<?php
declare(strict_types=1);
/**
* CakePHP(tm) : Rapid Development Framework (https://cakephp.org)
* Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
*
* Licensed under The MIT License
* For full copyright and license information, please see the LICENSE.txt
* Redistributions of files must retain the above copyright notice.
*
* @copyright Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
* @link https://cakephp.org CakePHP(tm) Project
* @since 5.0.0
* @license https://opensource.org/licenses/mit-license.php MIT License
*/
namespace Cake\TestSuite\Fixture;
use Cake\Database\Connection;
use Cake\Database\Exception\DatabaseException;
/**
* Fixture strategy that wraps fixtures in a transaction that is rolled back
* after each test.
*
* Any test that calls Connection::rollback(true) will break this strategy.
*/
class TransactionFixtureStrategy implements FixtureStrategyInterface
{
/**
* @var \Cake\TestSuite\Fixture\FixtureHelper
*/
protected FixtureHelper $helper;
/**
* @var array<\Cake\Datasource\FixtureInterface>
*/
protected array $fixtures = [];
/**
* Initialize strategy.
*/
public function __construct()
{
$this->helper = new FixtureHelper();
}
/**
* @inheritDoc
*/
public function setupTest(array $fixtureNames): void
{
$this->fixtures = $this->helper->loadFixtures($fixtureNames);
$this->helper->runPerConnection(function ($connection): void {
if ($connection instanceof Connection) {
assert(
$connection->inTransaction() === false,
'Cannot start transaction strategy inside a transaction. This is most likely a bug.'
);
$connection->enableSavePoints();
if (!$connection->isSavePointsEnabled()) {
throw new DatabaseException(
"Could not enable save points for the `{$connection->configName()}` connection. " .
'Your database needs to support savepoints in order to use ' .
'TransactionFixtureStrategy.'
);
}
$connection->begin();
$connection->createSavePoint('__fixtures__');
}
}, $this->fixtures);
$this->helper->insert($this->fixtures);
}
/**
* @inheritDoc
*/
public function teardownTest(): void
{
$this->helper->runPerConnection(function (Connection $connection): void {
if ($connection->inTransaction()) {
$connection->rollback(true);
}
}, $this->fixtures);
}
}