| 148 |
lars |
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 padraic@php.net 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\Generator\StringManipulation\Pass;
|
|
|
22 |
|
|
|
23 |
use Mockery\Generator\MockConfiguration;
|
|
|
24 |
|
|
|
25 |
class TraitPass implements Pass
|
|
|
26 |
{
|
|
|
27 |
public function apply($code, MockConfiguration $config)
|
|
|
28 |
{
|
|
|
29 |
$traits = $config->getTargetTraits();
|
|
|
30 |
|
|
|
31 |
if (empty($traits)) {
|
|
|
32 |
return $code;
|
|
|
33 |
}
|
|
|
34 |
|
|
|
35 |
$useStatements = array_map(function ($trait) {
|
|
|
36 |
return "use \\\\" . ltrim($trait->getName(), "\\") . ";";
|
|
|
37 |
}, $traits);
|
|
|
38 |
|
|
|
39 |
$code = preg_replace(
|
|
|
40 |
'/^{$/m',
|
|
|
41 |
"{\n " . implode("\n ", $useStatements) . "\n",
|
|
|
42 |
$code
|
|
|
43 |
);
|
|
|
44 |
|
|
|
45 |
return $code;
|
|
|
46 |
}
|
|
|
47 |
}
|