ProphecySubjectPatch.php in Zircon Profile 8

Namespace

Prophecy\Doubler\ClassPatch

File

vendor/phpspec/prophecy/src/Prophecy/Doubler/ClassPatch/ProphecySubjectPatch.php

<?php

/*
 * This file is part of the Prophecy.
 * (c) Konstantin Kudryashov <[e-mail]>
 *     Marcello Duarte <[e-mail]>
 *
 * 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 <[e-mail]>
 */
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;
  }

}

Classes

Name	Description
ProphecySubjectPatch	Add Prophecy functionality to the double. This is a core class patch for Prophecy.