fedecarg / ReturnValue.php

Created April 05, 2014


    
    View ReturnValue.php 
  
  
      ReturnValue.php
    
    
        Raw
      
    
    File suppressed. Click to show.
  
  
            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
          
          
            <?php

/**
 * Stubs a method by returning a user-defined value.
 */
class ReturnValue implements PHPUnit_Framework_MockObject_Stub
{
    protected $value;

    public function __construct($value)
    {
        $this->value = $value;
    }

    public function invoke(PHPUnit_Framework_MockObject_Invocation $invocation)
    {
        $annotations = PHPUnit_Util_Test::parseTestMethodAnnotations($invocation->className, $invocation->methodName);
        if (!isset($annotations['method']['return'][0])) {
            throw new PHPUnit_Framework_Exception('Invalid method declaration; return type required');
        } else {
            $actual = strtolower(gettype($this->value));
            $expected = explode('|', $annotations['method']['return'][0]);
            if (count($expected) > 1) {
                throw new PHPUnit_Framework_Exception(
                    sprintf(
                        '%s cannot return more than one type, %d given (%s)',
                        $invocation->methodName,
                        count($expected),
                        implode(', ', $expected)
                    )
                );
            } elseif (!in_array($actual, $expected)) {
                throw new PHPUnit_Framework_Exception(
                    sprintf(
                        '%s does not match expected type "%s"',
                        $actual,
                        $annotations['method']['return'][0]
                    )
                );
            }
        }

        return $this->value;
    }

    public function toString()
    {
        return sprintf('return user-specified value %s', PHPUnit_Util_Type::export($this->value));
    }
}