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
|
<?php declare(strict_types = 1);
namespace PHPStan\PhpDocParser\Ast;
use Exception;
use function count;
/**
* Inspired by https://github.com/nikic/PHP-Parser/tree/36a6dcd04e7b0285e8f0868f44bd4927802f7df1
*
* Copyright (c) 2011, Nikita Popov
* All rights reserved.
*/
class NodeVisitorForTesting implements NodeVisitor
{
/** @var list<array{string, Node|Node[]}> */
public array $trace = [];
/** @var list<list<mixed>> */
private array $returns;
private int $returnsPos;
/**
* @param list<list<mixed>> $returns
*/
public function __construct(array $returns = [])
{
$this->returns = $returns;
$this->returnsPos = 0;
}
public function beforeTraverse(array $nodes): ?array
{
return $this->traceEvent('beforeTraverse', $nodes);
}
public function enterNode(Node $node)
{
return $this->traceEvent('enterNode', $node);
}
public function leaveNode(Node $node)
{
return $this->traceEvent('leaveNode', $node);
}
public function afterTraverse(array $nodes): ?array
{
return $this->traceEvent('afterTraverse', $nodes);
}
/**
* @param Node|Node[] $param
* @return mixed
*/
private function traceEvent(string $method, $param)
{
$this->trace[] = [$method, $param];
if ($this->returnsPos < count($this->returns)) {
$currentReturn = $this->returns[$this->returnsPos];
if ($currentReturn[0] === $method && $currentReturn[1] === $param) {
$this->returnsPos++;
return $currentReturn[2];
}
}
return null;
}
public function __destruct()
{
if ($this->returnsPos !== count($this->returns)) {
throw new Exception('Expected event did not occur');
}
}
}
|
