-
Notifications
You must be signed in to change notification settings - Fork 211
/
getPendingJobs.php
81 lines (71 loc) · 1.93 KB
/
getPendingJobs.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
<?php
use WP_Rocket\Engine\Preload\Database\Queries\Cache;
use WP_Rocket\Engine\Preload\Database\Queries\RocketCache;
use WP_Rocket\Tests\Unit\TestCase;
use Brain\Monkey\Filters;
/**
* @covers \WP_Rocket\Engine\Preload\Database\Queries\RocketCache::get_pending_jobs
*
* @group Database
* @group Preload
*/
class Test_GetPendingJobs extends TestCase {
protected $query;
protected function setUp(): void
{
parent::setUp();
$this->query = $this->createPartialMock(Cache::class, ['get_rows_by_url', 'query']);
}
/**
* @dataProvider configTestData
*/
public function testShouldReturnPending($config, $expected) {
$queue = new SplQueue();
if($config['total'] > 0 && $config['in-progress'] < $config['total'] ) {
$queue->enqueue(
[
'params' =>
[
'count' => true,
'status' => 'in-progress',
'is_locked' => false,
],
'return' => $config['in-progress']
]
);
$queue->enqueue(
[
'params' =>
[
'number' => $config['total']-$config['in-progress'],
'status' => 'pending',
'fields' => [
'id',
'url',
],
'job_id__not_in' => [
'not_in' => '',
],
'orderby' => Filters\applied( 'rocket_preload_order' ) > 0 ? 'id' : 'modified',
'order' => 'asc',
'is_locked' => false
],
'return' => $config['results']
]
);
$this->query->expects(self::exactly(2))->method('query')
->willReturnCallback(function($params) use ($queue) {
$dequeue = $queue->dequeue();
$this->assertEquals($dequeue['params'], $params);
return $dequeue['return'];
});
} else {
$this->query->expects(self::once())->method('query')->with([
'count' => true,
'status' => 'in-progress',
'is_locked' => false,
])->willReturn($config['in-progress']);
}
$this->assertSame($expected, $this->query->get_pending_jobs($config['total']));
}
}