forked from codeigniter4/queue
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathQueueJobModelTest.php
More file actions
135 lines (107 loc) · 4.34 KB
/
Copy pathQueueJobModelTest.php
File metadata and controls
135 lines (107 loc) · 4.34 KB
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
<?php
declare(strict_types=1);
/**
* This file is part of CodeIgniter Queue.
*
* (c) CodeIgniter Foundation <admin@codeigniter.com>
*
* For the full copyright and license information, please view
* the LICENSE file that was distributed with this source code.
*/
namespace Tests\Models;
use CodeIgniter\Queue\Entities\QueueJob;
use CodeIgniter\Queue\Enums\Status;
use CodeIgniter\Queue\Models\QueueJobModel;
use CodeIgniter\Test\ReflectionHelper;
use Tests\Support\Database\Seeds\TestDatabaseQueueSeeder;
use Tests\Support\TestCase;
/**
* @internal
*/
final class QueueJobModelTest extends TestCase
{
use ReflectionHelper;
protected $seed = TestDatabaseQueueSeeder::class;
public function testQueueJobModel(): void
{
$model = model(QueueJobModel::class);
$this->assertInstanceOf(QueueJobModel::class, $model);
}
public function testSkipLocked(): void
{
$model = model(QueueJobModel::class);
$method = $this->getPrivateMethodInvoker($model, 'skipLocked');
$sql = 'SELECT * FROM queue_jobs WHERE queue = "test" AND status = 0 AND available_at < 123456 LIMIT 1';
$result = $method($sql);
if ($model->db->DBDriver === 'SQLite3') {
$this->assertSame($sql, $result);
} elseif ($model->db->DBDriver === 'SQLSRV') {
$this->assertStringContainsString('WITH (ROWLOCK,UPDLOCK,READPAST) WHERE', (string) $result);
} else {
$this->assertStringContainsString('FOR UPDATE SKIP LOCKED', (string) $result);
}
}
public function testSkipLockedFalse(): void
{
config('Queue')->database['skipLocked'] = false;
$model = model(QueueJobModel::class);
$method = $this->getPrivateMethodInvoker($model, 'skipLocked');
$sql = 'SELECT * FROM queue_jobs WHERE queue = "test" AND status = 0 AND available_at < 123456 LIMIT 1';
$result = $method($sql);
$this->assertSame($sql, $result);
}
public function testGetFromQueueReservesPendingJob(): void
{
$model = model(QueueJobModel::class);
$job = $model->getFromQueue('queue1', ['default']);
$this->assertInstanceOf(QueueJob::class, $job);
$this->assertSame('queue1', $job->queue);
$this->seeInDatabase('queue_jobs', [
'id' => $job->id,
'status' => Status::RESERVED->value,
]);
}
public function testGetFromQueueReturnsNullWhenNothingPending(): void
{
$model = model(QueueJobModel::class);
$this->assertNull($model->getFromQueue('queue123', ['default']));
}
public function testGetFromQueueOnSQLite3ClaimsEachPendingJobOnce(): void
{
$model = model(QueueJobModel::class);
if ($model->db->DBDriver !== 'SQLite3') {
$this->markTestSkipped('BEGIN IMMEDIATE claim is SQLite3-only.');
}
$model->insert(new QueueJob([
'queue' => 'queue1',
'payload' => ['job' => 'extra', 'data' => []],
'priority' => 'default',
'status' => Status::PENDING->value,
'attempts' => 0,
'available_at' => 1_697_269_865,
]));
$job1 = $model->getFromQueue('queue1', ['default']);
$job2 = $model->getFromQueue('queue1', ['default']);
$this->assertInstanceOf(QueueJob::class, $job1);
$this->assertInstanceOf(QueueJob::class, $job2);
$this->assertSame(2, $job1->id);
$this->assertSame(3, $job2->id);
$this->assertNull($model->getFromQueue('queue1', ['default']));
$this->seeInDatabase('queue_jobs', ['id' => 2, 'status' => Status::RESERVED->value]);
$this->seeInDatabase('queue_jobs', ['id' => 3, 'status' => Status::RESERVED->value]);
}
public function testSetPriorityEscapesPriorityValues(): void
{
$model = model(QueueJobModel::class);
$method = $this->getPrivateMethodInvoker($model, 'setPriority');
$builder = $model->builder();
$priority = [
'priority_key' => "default' THEN 0 ELSE 1 END --",
'default',
];
$result = $method($builder, $priority);
$sql = (string) $result->getCompiledSelect();
$this->assertStringContainsString($model->db->escape($priority['priority_key']), $sql);
$this->assertStringNotContainsString('priority_key', $sql);
}
}