1 | <?php |
||
11 | trait MockByCallsTrait |
||
12 | { |
||
13 | /** |
||
14 | * @param string $class |
||
15 | * @param Call[] $calls |
||
16 | * |
||
17 | * @return MockObject |
||
18 | */ |
||
19 | 8 | private function getMockByCalls(string $class, array $calls = []): MockObject |
|
|
|||
20 | { |
||
21 | /** @var MockByCallsTrait|TestCase $this */ |
||
22 | 8 | $reflectionClass = new \ReflectionClass($class); |
|
23 | |||
24 | 8 | $mockBuilder = $this->getMockBuilder($class) |
|
25 | 8 | ->disableOriginalConstructor() |
|
26 | 8 | ->disableOriginalClone(); |
|
27 | |||
28 | /* @var MockObject $mock */ |
||
29 | 8 | if ($reflectionClass->isAbstract() || $reflectionClass->isInterface()) { |
|
30 | 7 | $mock = $mockBuilder->getMockForAbstractClass(); |
|
31 | } else { |
||
32 | 1 | $mock = $mockBuilder->getMock(); |
|
33 | } |
||
34 | |||
35 | 8 | $callCount = count($calls); |
|
36 | |||
37 | 8 | if (0 === $callCount) { |
|
38 | 2 | $mock->expects(self::never())->method(self::anything()); |
|
39 | |||
40 | 2 | return $mock; |
|
41 | } |
||
42 | |||
43 | 6 | foreach ($calls as $i => $call) { |
|
44 | 6 | $mock->expects(self::at($i)) |
|
45 | 6 | ->method($call->getMethod()) |
|
46 | 6 | ->willReturnCallback($this->getMockCallback($class, $i, $call, $mock)); |
|
47 | } |
||
48 | |||
49 | 6 | $callIndex = 0; |
|
50 | |||
51 | 6 | $mock->expects(self::any())->method(self::anything())->willReturnCallback( |
|
52 | 6 | function () use ($class, $callCount, &$callIndex) { |
|
53 | 6 | if ($callIndex === $callCount) { |
|
54 | 1 | self::fail(sprintf('Additional call at index %d on class "%s"!', $callIndex, $class)); |
|
55 | } |
||
56 | |||
57 | 6 | ++$callIndex; |
|
58 | 6 | } |
|
59 | ); |
||
60 | |||
61 | 6 | return $mock; |
|
62 | } |
||
63 | |||
64 | /** |
||
65 | * @param string $class |
||
66 | * @param int $i |
||
67 | * @param Call $call |
||
68 | * @param MockObject $mock |
||
69 | * |
||
70 | * @return \Closure |
||
71 | */ |
||
72 | private function getMockCallback( |
||
73 | string $class, |
||
74 | int $i, |
||
75 | Call $call, |
||
76 | MockObject $mock |
||
77 | ): \Closure { |
||
78 | 6 | return function () use ($class, $i, $call, $mock) { |
|
79 | 6 | if ($call->hasWith()) { |
|
80 | 6 | $this->compareArguments($class, $call->getMethod(), $i, $call->getWith(), func_get_args()); |
|
81 | } |
||
82 | |||
83 | 5 | if (null !== $exception = $call->getException()) { |
|
84 | 1 | throw $exception; |
|
85 | } |
||
86 | |||
87 | 4 | if ($call->hasReturnSelf()) { |
|
88 | 1 | return $mock; |
|
89 | } |
||
90 | |||
91 | 3 | if ($call->hasReturn()) { |
|
92 | 2 | return $call->getReturn(); |
|
93 | } |
||
94 | 6 | }; |
|
95 | } |
||
96 | |||
97 | /** |
||
98 | * @param string $class |
||
99 | * @param string $method |
||
100 | * @param int $i |
||
101 | * @param array $expectedArguments |
||
102 | * @param array $arguments |
||
103 | */ |
||
104 | 6 | private function compareArguments( |
|
144 | } |
||
145 |