Blame view

vendor/phpspec/prophecy/src/Prophecy/Doubler/ClassPatch/ProphecySubjectPatch.php 2.74 KB
ad2e91f7   Mihail   move multyparser ...
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
  <?php
  
  /*
   * This file is part of the Prophecy.
   * (c) Konstantin Kudryashov <ever.zet@gmail.com>
   *     Marcello Duarte <marcello.duarte@gmail.com>
   *
   * For the full copyright and license information, please view the LICENSE
   * file that was distributed with this source code.
   */
  
  namespace Prophecy\Doubler\ClassPatch;
  
  use Prophecy\Doubler\Generator\Node\ClassNode;
  use Prophecy\Doubler\Generator\Node\MethodNode;
  use Prophecy\Doubler\Generator\Node\ArgumentNode;
  
  /**
   * Add Prophecy functionality to the double.
   * This is a core class patch for Prophecy.
   *
   * @author Konstantin Kudryashov <ever.zet@gmail.com>
   */
  class ProphecySubjectPatch implements ClassPatchInterface
  {
      /**
       * Always returns true.
       *
       * @param ClassNode $node
       *
       * @return bool
       */
      public function supports(ClassNode $node)
      {
          return true;
      }
  
      /**
       * Apply Prophecy functionality to class node.
       *
       * @param ClassNode $node
       */
      public function apply(ClassNode $node)
      {
          $node->addInterface('Prophecy\Prophecy\ProphecySubjectInterface');
          $node->addProperty('objectProphecy', 'private');
  
          foreach ($node->getMethods() as $name => $method) {
              if ('__construct' === strtolower($name)) {
                  continue;
              }
  
              $method->setCode(
                  'return $this->getProphecy()->makeProphecyMethodCall(__FUNCTION__, func_get_args());'
              );
          }
  
          $prophecySetter = new MethodNode('setProphecy');
          $prophecyArgument = new ArgumentNode('prophecy');
          $prophecyArgument->setTypeHint('Prophecy\Prophecy\ProphecyInterface');
          $prophecySetter->addArgument($prophecyArgument);
          $prophecySetter->setCode('$this->objectProphecy = $prophecy;');
  
          $prophecyGetter = new MethodNode('getProphecy');
          $prophecyGetter->setCode('return $this->objectProphecy;');
  
          if ($node->hasMethod('__call')) {
              $__call = $node->getMethod('__call');
          } else {
              $__call = new MethodNode('__call');
              $__call->addArgument(new ArgumentNode('name'));
              $__call->addArgument(new ArgumentNode('arguments'));
  
              $node->addMethod($__call);
          }
  
          $__call->setCode(<<<PHP
  throw new \Prophecy\Exception\Doubler\MethodNotFoundException(
      sprintf('Method `%s::%s()` not found.', get_class(\$this), func_get_arg(0)),
      \$this->getProphecy(), func_get_arg(0)
  );
  PHP
          );
  
          $node->addMethod($prophecySetter);
          $node->addMethod($prophecyGetter);
      }
  
      /**
       * Returns patch priority, which determines when patch will be applied.
       *
       * @return int Priority number (higher - earlier)
       */
      public function getPriority()
      {
          return 0;
      }
  }