Repository navigation
Expand file tree
/
Copy pathTaskAspectTest.php
More file actions
108 lines (91 loc) · 3.25 KB
/
Copy pathTaskAspectTest.php
File metadata and controls
108 lines (91 loc) · 3.25 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
<?php
declare(strict_types=1);
/**
* This file is part of Hyperf.
*
* @link https://www.hyperf.io
* @document https://hyperf.wiki
* @contact [email protected]
* @license https://github.com/hyperf/hyperf/blob/master/LICENSE
*/
namespace HyperfTest\Task;
use Hyperf\Context\ApplicationContext;
use Hyperf\Contract\ContainerInterface;
use Hyperf\Di\Aop\ProceedingJoinPoint;
use Hyperf\Task\Aspect\TaskAspect;
use Hyperf\Task\Task;
use Hyperf\Task\TaskExecutor;
use HyperfTest\Task\Stub\Foo;
use Mockery;
use PHPUnit\Framework\Attributes\CoversNothing;
use PHPUnit\Framework\TestCase;
/**
* @internal
* @coversNothing
*/
#[CoversNothing]
class TaskAspectTest extends TestCase
{
protected $isTaskEnvironment = false;
protected function tearDown(): void
{
Mockery::close();
$this->isTaskEnvironment = false;
}
public function testTaskAspect()
{
$container = $this->getContainer();
$aspect = new TaskAspect($container);
$closure = function ($id, $name) {
return ['id' => $id, 'name' => $name];
};
$point = new ProceedingJoinPoint($closure, Foo::class, 'getIdAndName', [
'keys' => $data = ['id' => uniqid(), 'name' => 'Hyperf'],
'order' => ['id', 'name'],
]);
$point->pipe = function (ProceedingJoinPoint $point) {
return $point->processOriginalMethod();
};
$res = $aspect->process($point);
$this->assertSame($data, $res);
$this->isTaskEnvironment = true;
$res = $aspect->process($point);
$this->assertSame($data, $res);
}
public function testTaskAspectVariadic()
{
$container = $this->getContainer();
$aspect = new TaskAspect($container);
$closure = function ($id, $arguments) {
return ['id' => $id, 'arguments' => $arguments];
};
$point = new ProceedingJoinPoint($closure, Foo::class, 'dump', [
'keys' => $data = ['id' => 1, 'arguments' => [1, 2, 3]],
'order' => ['id', 'arguments'],
]);
$point->pipe = function (ProceedingJoinPoint $point) {
return $point->processOriginalMethod();
};
$res = $aspect->process($point);
$this->assertSame($data, $res);
$this->isTaskEnvironment = true;
$res = $aspect->process($point);
$this->assertSame($data, $res);
}
protected function getContainer()
{
$container = Mockery::mock(ContainerInterface::class);
ApplicationContext::setContainer($container);
$container->shouldReceive('get')->with(Foo::class)->andReturn(new Foo());
$container->shouldReceive('get')->with(TaskExecutor::class)->andReturnUsing(function () use ($container) {
$executor = Mockery::mock(TaskExecutor::class);
$executor->shouldReceive('isTaskEnvironment')->andReturn($this->isTaskEnvironment);
$executor->shouldReceive('execute')->with(Mockery::any(), Mockery::any())->andReturnUsing(function (Task $task, $_) use ($container) {
[$class, $method] = $task->callback;
return $container->get($class)->{$method}(...$task->arguments);
});
return $executor;
});
return $container;
}
}