Completed
Pull Request — master (#668)
by Dave
03:50
created

Expectation::andReturns()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 4
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 2

Importance

Changes 0
Metric Value
dl 0
loc 4
ccs 0
cts 2
cp 0
rs 10
c 0
b 0
f 0
cc 1
eloc 2
nc 1
nop 0
crap 2
1
<?php
2
/**
3
 * Mockery
4
 *
5
 * LICENSE
6
 *
7
 * This source file is subject to the new BSD license that is bundled
8
 * with this package in the file LICENSE.txt.
9
 * It is also available through the world-wide-web at this URL:
10
 * http://github.com/padraic/mockery/blob/master/LICENSE
11
 * If you did not receive a copy of the license and are unable to
12
 * obtain it through the world-wide-web, please send an email
13
 * to [email protected] so we can send you a copy immediately.
14
 *
15
 * @category   Mockery
16
 * @package    Mockery
17
 * @copyright  Copyright (c) 2010 Pádraic Brady (http://blog.astrumfutura.com)
18
 * @license    http://github.com/padraic/mockery/blob/master/LICENSE New BSD License
19
 */
20
21
namespace Mockery;
22
23
use Closure;
24
use Mockery\Matcher\MultiArgumentClosure;
25
26
class Expectation implements ExpectationInterface
27
{
28
    /**
29
     * Mock object to which this expectation belongs
30
     *
31
     * @var object
32
     */
33
    protected $_mock = null;
34
35
    /**
36
     * Method name
37
     *
38
     * @var string
39
     */
40
    protected $_name = null;
41
42
    /**
43
     * Arguments expected by this expectation
44
     *
45
     * @var array
46
     */
47
    protected $_expectedArgs = array();
48
49
    /**
50
     * Count validator store
51
     *
52
     * @var array
53
     */
54
    protected $_countValidators = array();
55
56
    /**
57
     * The count validator class to use
58
     *
59
     * @var string
60
     */
61
    protected $_countValidatorClass = 'Mockery\CountValidator\Exact';
62
63
    /**
64
     * Actual count of calls to this expectation
65
     *
66
     * @var int
67
     */
68
    protected $_actualCount = 0;
69
70
    /**
71
     * Value to return from this expectation
72
     *
73
     * @var mixed
74
     */
75
    protected $_returnValue = null;
76
77
    /**
78
     * Array of return values as a queue for multiple return sequence
79
     *
80
     * @var array
81
     */
82
    protected $_returnQueue = array();
83
84
    /**
85
     * Array of closures executed with given arguments to generate a result
86
     * to be returned
87
     *
88
     * @var array
89
     */
90
    protected $_closureQueue = array();
91
92
    /**
93
     * Array of values to be set when this expectation matches
94
     *
95
     * @var array
96
     */
97
    protected $_setQueue = array();
98
99
    /**
100
     * Integer representing the call order of this expectation
101
     *
102
     * @var int
103
     */
104
    protected $_orderNumber = null;
105
106
    /**
107
     * Integer representing the call order of this expectation on a global basis
108
     *
109
     * @var int
110
     */
111
    protected $_globalOrderNumber = null;
112
113
    /**
114
     * Flag indicating that an exception is expected to be throw (not returned)
115
     *
116
     * @var bool
117
     */
118
    protected $_throw = false;
119
120
    /**
121
     * Flag indicating whether the order of calling is determined locally or
122
     * globally
123
     *
124
     * @var bool
125
     */
126
    protected $_globally = false;
127
128
    /**
129
     * Flag indicating we expect no arguments
130
     *
131
     * @var bool
132
     */
133
    protected $_noArgsExpectation = false;
134
135
    /**
136
     * Flag indicating if the return value should be obtained from the original
137
     * class method instead of returning predefined values from the return queue
138
     *
139
     * @var bool
140
     */
141
    protected $_passthru = false;
142
143
    /**
144
     * Constructor
145
     *
146
     * @param \Mockery\MockInterface $mock
147
     * @param string $name
148
     */
149 306
    public function __construct(\Mockery\MockInterface $mock, $name)
150
    {
151 306
        $this->_mock = $mock;
152 306
        $this->_name = $name;
153 306
    }
154
155
    /**
156
     * Return a string with the method name and arguments formatted
157
     *
158
     * @param string $name Name of the expected method
159
     * @param array $args List of arguments to the method
160
     * @return string
161
     */
162 45
    public function __toString()
163
    {
164 45
        return \Mockery::formatArgs($this->_name, $this->_expectedArgs);
165
    }
166
167
    /**
168
     * Verify the current call, i.e. that the given arguments match those
169
     * of this expectation
170
     *
171
     * @param array $args
172
     * @return mixed
173
     */
174 235
    public function verifyCall(array $args)
175
    {
176 235
        $this->validateOrder();
177 235
        $this->_actualCount++;
178 235
        if (true === $this->_passthru) {
179 3
            return $this->_mock->mockery_callSubjectMethod($this->_name, $args);
180
        }
181 233
        $return = $this->_getReturnValue($args);
182 233
        if ($return instanceof \Exception && $this->_throw === true) {
183 7
            throw $return;
184
        }
185 227
        $this->_setValues();
186 227
        return $return;
187
    }
188
189
    /**
190
     * Sets public properties with queued values to the mock object
191
     *
192
     * @param array $args
0 ignored issues
show
Bug introduced by
There is no parameter named $args. Was it maybe removed?

This check looks for PHPDoc comments describing methods or function parameters that do not exist on the corresponding method or function.

Consider the following example. The parameter $italy is not defined by the method finale(...).

/**
 * @param array $germany
 * @param array $island
 * @param array $italy
 */
function finale($germany, $island) {
    return "2:1";
}

The most likely cause is that the parameter was removed, but the annotation was not.

Loading history...
193
     * @return mixed
194
     */
195 227
    protected function _setValues()
196
    {
197 227
        foreach ($this->_setQueue as $name => &$values) {
198 8
            if (count($values) > 0) {
199 8
                $value = array_shift($values);
200 8
                $this->_mock->{$name} = $value;
201 8
            }
202 227
        }
203 227
    }
204
205
    /**
206
     * Fetch the return value for the matching args
207
     *
208
     * @param array $args
209
     * @return mixed
210
     */
211 233
    protected function _getReturnValue(array $args)
212
    {
213 233
        if (count($this->_closureQueue) > 1) {
214
            return call_user_func_array(array_shift($this->_closureQueue), $args);
215 233
        } elseif (count($this->_closureQueue) > 0) {
216 1
            return call_user_func_array(current($this->_closureQueue), $args);
217 232
        } elseif (count($this->_returnQueue) > 1) {
218 5
            return array_shift($this->_returnQueue);
219 231
        } elseif (count($this->_returnQueue) > 0) {
220 93
            return current($this->_returnQueue);
221
        }
222
223 139
        return $this->_mock->mockery_returnValueForMethod($this->_name);
224
    }
225
226
    /**
227
     * Checks if this expectation is eligible for additional calls
228
     *
229
     * @return bool
230
     */
231 273
    public function isEligible()
232
    {
233 273
        foreach ($this->_countValidators as $validator) {
234 139
            if (!$validator->isEligible($this->_actualCount)) {
235 22
                return false;
236
            }
237 270
        }
238 270
        return true;
239
    }
240
241
    /**
242
     * Check if there is a constraint on call count
243
     *
244
     * @return bool
245
     */
246
    public function isCallCountConstrained()
247
    {
248
        return (count($this->_countValidators) > 0);
249
    }
250
251
    /**
252
     * Verify call order
253
     *
254
     * @return void
255
     */
256 235
    public function validateOrder()
257
    {
258 235
        if ($this->_orderNumber) {
259 15
            $this->_mock->mockery_validateOrder((string) $this, $this->_orderNumber, $this->_mock);
260 15
        }
261 235
        if ($this->_globalOrderNumber) {
262 1
            $this->_mock->mockery_getContainer()
263 1
                ->mockery_validateOrder((string) $this, $this->_globalOrderNumber, $this->_mock);
264 1
        }
265 235
    }
266
267
    /**
268
     * Verify this expectation
269
     *
270
     * @return bool
271
     */
272 144
    public function verify()
273
    {
274 144
        foreach ($this->_countValidators as $validator) {
275 121
            $validator->validate($this->_actualCount);
276 129
        }
277 126
    }
278
279
    /**
280
     * Check if the registered expectation is a MultiArgumentClosureExpectation.
281
     * @return bool
282
     */
283 116
    private function isMultiArgumentClosureExpectation()
284
    {
285 116
        return (count($this->_expectedArgs) === 1 && ($this->_expectedArgs[0] instanceof \Mockery\Matcher\MultiArgumentClosure));
286
    }
287
288
    /**
289
     * Check if passed arguments match an argument expectation
290
     *
291
     * @param array $args
292
     * @return bool
293
     */
294 279
    public function matchArgs(array $args)
295
    {
296 279
        if (empty($this->_expectedArgs) && !$this->_noArgsExpectation) {
297 167
            return true;
298
        }
299 116
        if ($this->isMultiArgumentClosureExpectation()) {
300 6
            return $this->_matchArg($this->_expectedArgs[0], $args);
301
        }
302 110
        $argCount = count($args);
303 110
        if ($argCount !== count($this->_expectedArgs)) {
304 8
            return false;
305
        }
306 104
        for ($i=0; $i<$argCount; $i++) {
307 102
            $param =& $args[$i];
308 102
            if (!$this->_matchArg($this->_expectedArgs[$i], $param)) {
309 45
                return false;
310
            }
311 66
        }
312
313 68
        return true;
314
    }
315
316
    /**
317
     * Check if passed argument matches an argument expectation
318
     *
319
     * @param mixed $expected
320
     * @param mixed &$actual
321
     * @return bool
322
     */
323 108
    protected function _matchArg($expected, &$actual)
324
    {
325 108
        if ($expected === $actual) {
326 30
            return true;
327
        }
328 90
        if (!is_object($expected) && !is_object($actual) && $expected == $actual) {
329
            return true;
330
        }
331 90
        if (is_string($expected) && !is_array($actual) && !is_object($actual)) {
332
            # push/pop an error handler here to to make sure no error/exception thrown if $expected is not a regex
333 6
            set_error_handler(function () {
334 6
            });
335 6
            $result = preg_match($expected, (string) $actual);
336 6
            restore_error_handler();
337
338 6
            if ($result) {
339 3
                return true;
340
            }
341 3
        }
342 89
        if (is_string($expected) && is_object($actual)) {
343 1
            $result = $actual instanceof $expected;
344 1
            if ($result) {
345 1
                return true;
346
            }
347
        }
348 88
        if ($expected instanceof \Mockery\Matcher\MatcherAbstract) {
349 67
            return $expected->match($actual);
350
        }
351 21
        if ($expected instanceof \Hamcrest\Matcher || $expected instanceof \Hamcrest_Matcher) {
0 ignored issues
show
Bug introduced by
The class Hamcrest_Matcher does not exist. Did you forget a USE statement, or did you not list all dependencies?

This error could be the result of:

1. Missing dependencies

PHP Analyzer uses your composer.json file (if available) to determine the dependencies of your project and to determine all the available classes and functions. It expects the composer.json to be in the root folder of your repository.

Are you sure this class is defined by one of your dependencies, or did you maybe not list a dependency in either the require or require-dev section?

2. Missing use statement

PHP does not complain about undefined classes in ìnstanceof checks. For example, the following PHP code will work perfectly fine:

if ($x instanceof DoesNotExist) {
    // Do something.
}

If you have not tested against this specific condition, such errors might go unnoticed.

Loading history...
352 3
            return $expected->matches($actual);
353
        }
354 18
        return false;
355
    }
356
357
    /**
358
     * Expected argument setter for the expectation
359
     *
360
     * @param mixed ...
361
     * @return self
362
     */
363 130
    public function with()
364
    {
365 130
        return $this->withArgs(func_get_args());
366
    }
367
368
    /**
369
     * Expected arguments for the expectation passed as an array
370
     *
371
     * @param array $arguments
372
     * @return self
373
     */
374 139
    private function withArgsInArray(array $arguments)
375
    {
376 139
        if (empty($arguments)) {
377 3
            return $this->withNoArgs();
378
        }
379 136
        $this->_expectedArgs = $arguments;
380 136
        $this->_noArgsExpectation = false;
381 136
        return $this;
382
    }
383
384
    /**
385
     * Expected arguments have to be matched by the given closure.
386
     *
387
     * @param Closure $closure
388
     * @return self
389
     */
390 6
    private function withArgsMatchedByClosure(Closure $closure)
391
    {
392 6
        $this->_expectedArgs = [new MultiArgumentClosure($closure)];
393 6
        $this->_noArgsExpectation = false;
394 6
        return $this;
395
    }
396
397
    /**
398
     * Expected arguments for the expectation passed as an array or a closure that matches each passed argument on
399
     * each function call.
400
     *
401
     * @param array|Closure $argsOrClosure
402
     * @return self
403
     */
404 146
    public function withArgs($argsOrClosure)
405
    {
406 146
        if (is_array($argsOrClosure)) {
407 139
            $this->withArgsInArray($argsOrClosure);
408 146
        } elseif ($argsOrClosure instanceof Closure) {
409 6
            $this->withArgsMatchedByClosure($argsOrClosure);
410 6
        } else {
411 1
            throw new \InvalidArgumentException(sprintf('Call to %s with an invalid argument (%s), only array and '.
412 1
                'closure are allowed', __METHOD__, $argsOrClosure));
413
        }
414 145
        return $this;
415
    }
416
417
    /**
418
     * Set with() as no arguments expected
419
     *
420
     * @return self
421
     */
422 7
    public function withNoArgs()
423
    {
424 7
        $this->_noArgsExpectation = true;
425 7
        $this->_expectedArgs = [];
426 7
        return $this;
427
    }
428
429
    /**
430
     * Set expectation that any arguments are acceptable
431
     *
432
     * @return self
433
     */
434 3
    public function withAnyArgs()
435
    {
436 3
        $this->_noArgsExpectation = false;
437 3
        $this->_expectedArgs = array();
438 3
        return $this;
439
    }
440
441
    /**
442
     * Set a return value, or sequential queue of return values
443
     *
444
     * @param mixed ...
445
     * @return self
446
     */
447 106
    public function andReturn()
448
    {
449 106
        $this->_returnQueue = func_get_args();
450 106
        return $this;
451
    }
452
453
    /**
454
     * Set a return value, or sequential queue of return values
455
     *
456
     * @param mixed ...
457
     * @return self
458
     */
459
    public function andReturns()
460
    {
461
        return call_user_func_array([$this, 'andReturn'], func_get_args());
462
    }
463
464
    /**
465
     * Return this mock, like a fluent interface
466
     *
467
     * @return self
468
     */
469 1
    public function andReturnSelf()
470
    {
471 1
        return $this->andReturn($this->_mock);
472
    }
473
474
    /**
475
     * Set a sequential queue of return values with an array
476
     *
477
     * @param array $values
478
     * @return self
479
     */
480 2
    public function andReturnValues(array $values)
481
    {
482 2
        call_user_func_array(array($this, 'andReturn'), $values);
483 2
        return $this;
484
    }
485
486
    /**
487
     * Set a closure or sequence of closures with which to generate return
488
     * values. The arguments passed to the expected method are passed to the
489
     * closures as parameters.
490
     *
491
     * @param callable ...
492
     * @return self
493
     */
494 1
    public function andReturnUsing()
495
    {
496 1
        $this->_closureQueue = func_get_args();
497 1
        return $this;
498
    }
499
500
    /**
501
     * Return a self-returning black hole object.
502
     *
503
     * @return self
504
     */
505 1
    public function andReturnUndefined()
506
    {
507 1
        $this->andReturn(new \Mockery\Undefined);
508 1
        return $this;
509
    }
510
511
    /**
512
     * Return null. This is merely a language construct for Mock describing.
513
     *
514
     * @return self
515
     */
516 1
    public function andReturnNull()
517
    {
518 1
        return $this;
519
    }
520
521 1
    public function andReturnFalse()
522
    {
523 1
        return $this->andReturn(false);
524
    }
525
526 1
    public function andReturnTrue()
527
    {
528 1
        return $this->andReturn(true);
529
    }
530
531
    /**
532
     * Set Exception class and arguments to that class to be thrown
533
     *
534
     * @param string|\Exception $exception
535
     * @param string $message
536
     * @param int $code
537
     * @param \Exception $previous
538
     * @return self
539
     */
540 6
    public function andThrow($exception, $message = '', $code = 0, \Exception $previous = null)
541
    {
542 6
        $this->_throw = true;
543 6
        if (is_object($exception)) {
544 2
            $this->andReturn($exception);
545 2
        } else {
546 4
            $this->andReturn(new $exception($message, $code, $previous));
547
        }
548 6
        return $this;
549
    }
550
551
    /**
552
     * Set Exception classes to be thrown
553
     *
554
     * @param array $exceptions
555
     * @return self
556
     */
557 2
    public function andThrowExceptions(array $exceptions)
558
    {
559 2
        $this->_throw = true;
560 2
        foreach ($exceptions as $exception) {
561 2
            if (!is_object($exception)) {
562 1
                throw new Exception('You must pass an array of exception objects to andThrowExceptions');
563
            }
564 1
        }
565 1
        return $this->andReturnValues($exceptions);
566
    }
567
568
    /**
569
     * Register values to be set to a public property each time this expectation occurs
570
     *
571
     * @param string $name
572
     * @param mixed $value
573
     * @return self
574
     */
575 8
    public function andSet($name, $value)
576
    {
577 8
        $values = func_get_args();
578 8
        array_shift($values);
579 8
        $this->_setQueue[$name] = $values;
580 8
        return $this;
581
    }
582
583
    /**
584
     * Alias to andSet(). Allows the natural English construct
585
     * - set('foo', 'bar')->andReturn('bar')
586
     *
587
     * @param string $name
588
     * @param mixed $value
589
     * @return self
590
     */
591 3
    public function set($name, $value)
592
    {
593 3
        return call_user_func_array(array($this, 'andSet'), func_get_args());
594
    }
595
596
    /**
597
     * Indicates this expectation should occur zero or more times
598
     *
599
     * @return self
600
     */
601 2
    public function zeroOrMoreTimes()
602
    {
603 2
        $this->atLeast()->never();
604 2
    }
605
606
    /**
607
     * Indicates the number of times this expectation should occur
608
     *
609
     * @param int $limit
610
     * @throws InvalidArgumentException
611
     * @return self
612
     */
613 158
    public function times($limit = null)
614
    {
615 158
        if (is_null($limit)) {
616
            return $this;
617
        }
618 158
        if (!is_int($limit)) {
619 1
            throw new \InvalidArgumentException('The passed Times limit should be an integer value');
620
        }
621 157
        $this->_countValidators[] = new $this->_countValidatorClass($this, $limit);
622 157
        $this->_countValidatorClass = 'Mockery\CountValidator\Exact';
623 157
        return $this;
624
    }
625
626
    /**
627
     * Indicates that this expectation is never expected to be called
628
     *
629
     * @return self
630
     */
631 38
    public function never()
632
    {
633 38
        return $this->times(0);
634
    }
635
636
    /**
637
     * Indicates that this expectation is expected exactly once
638
     *
639
     * @return self
640
     */
641 108
    public function once()
642
    {
643 108
        return $this->times(1);
644
    }
645
646
    /**
647
     * Indicates that this expectation is expected exactly twice
648
     *
649
     * @return self
650
     */
651 19
    public function twice()
652
    {
653 19
        return $this->times(2);
654
    }
655
656
    /**
657
     * Sets next count validator to the AtLeast instance
658
     *
659
     * @return self
660
     */
661 14
    public function atLeast()
662
    {
663 14
        $this->_countValidatorClass = 'Mockery\CountValidator\AtLeast';
664 14
        return $this;
665
    }
666
667
    /**
668
     * Sets next count validator to the AtMost instance
669
     *
670
     * @return self
671
     */
672 7
    public function atMost()
673
    {
674 7
        $this->_countValidatorClass = 'Mockery\CountValidator\AtMost';
675 7
        return $this;
676
    }
677
678
    /**
679
     * Shorthand for setting minimum and maximum constraints on call counts
680
     *
681
     * @param int $minimum
682
     * @param int $maximum
683
     */
684
    public function between($minimum, $maximum)
685
    {
686
        return $this->atLeast()->times($minimum)->atMost()->times($maximum);
687
    }
688
689
    /**
690
     * Indicates that this expectation must be called in a specific given order
691
     *
692
     * @param string $group Name of the ordered group
693
     * @return self
694
     */
695 17
    public function ordered($group = null)
696
    {
697 17
        if ($this->_globally) {
698 1
            $this->_globalOrderNumber = $this->_defineOrdered($group, $this->_mock->mockery_getContainer());
699 1
        } else {
700 16
            $this->_orderNumber = $this->_defineOrdered($group, $this->_mock);
701
        }
702 17
        $this->_globally = false;
703 17
        return $this;
704
    }
705
706
    /**
707
     * Indicates call order should apply globally
708
     *
709
     * @return self
710
     */
711 1
    public function globally()
712
    {
713 1
        $this->_globally = true;
714 1
        return $this;
715
    }
716
717
    /**
718
     * Setup the ordering tracking on the mock or mock container
719
     *
720
     * @param string $group
721
     * @param object $ordering
722
     * @return int
723
     */
724 17
    protected function _defineOrdered($group, $ordering)
725
    {
726 17
        $groups = $ordering->mockery_getGroups();
727 17
        if (is_null($group)) {
728 16
            $result = $ordering->mockery_allocateOrder();
729 17
        } elseif (isset($groups[$group])) {
730 2
            $result = $groups[$group];
731 2
        } else {
732 4
            $result = $ordering->mockery_allocateOrder();
733 4
            $ordering->mockery_setGroup($group, $result);
734
        }
735 17
        return $result;
736
    }
737
738
    /**
739
     * Return order number
740
     *
741
     * @return int
742
     */
743 1
    public function getOrderNumber()
744
    {
745 1
        return $this->_orderNumber;
746
    }
747
748
    /**
749
     * Mark this expectation as being a default
750
     *
751
     * @return self
752
     */
753 25
    public function byDefault()
754
    {
755 25
        $director = $this->_mock->mockery_getExpectationsFor($this->_name);
756 25
        if (!empty($director)) {
757 25
            $director->makeExpectationDefault($this);
758 24
        }
759 24
        return $this;
760
    }
761
762
    /**
763
     * Return the parent mock of the expectation
764
     *
765
     * @return \Mockery\MockInterface
766
     */
767 29
    public function getMock()
768
    {
769 29
        return $this->_mock;
770
    }
771
772
    /**
773
     * Flag this expectation as calling the original class method with the
774
     * any provided arguments instead of using a return value queue.
775
     *
776
     * @return self
777
     */
778 3
    public function passthru()
779
    {
780 3
        if ($this->_mock instanceof Mock) {
781
            throw new Exception(
782
                'Mock Objects not created from a loaded/existing class are '
783
                . 'incapable of passing method calls through to a parent class'
784
            );
785
        }
786 3
        $this->_passthru = true;
787 3
        return $this;
788
    }
789
790
    /**
791
     * Cloning logic
792
     *
793
     */
794 9
    public function __clone()
795
    {
796 9
        $newValidators = array();
797 9
        $countValidators = $this->_countValidators;
798 9
        foreach ($countValidators as $validator) {
799 6
            $newValidators[] = clone $validator;
800 9
        }
801 9
        $this->_countValidators = $newValidators;
802 9
    }
803
804 6
    public function getName()
805
    {
806 6
        return $this->_name;
807
    }
808
}
809