Completed
Pull Request — master (#685)
by Dave
02:20
created

TraitPass::apply()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 20
Code Lines 12

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 4
CRAP Score 2.864

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 2
eloc 12
nc 2
nop 2
dl 0
loc 20
ccs 4
cts 10
cp 0.4
crap 2.864
rs 9.4285
c 1
b 0
f 0
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\Generator\StringManipulation\Pass;
22
23
use Mockery\Generator\MockConfiguration;
24
25
class TraitPass implements Pass
26
{
27 420
    public function apply($code, MockConfiguration $config)
28
    {
29 420
        $traits = $config->getTargetTraits();
30
31 420
        if (empty($traits)) {
32 420
            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
}
48