DrupalConsole/Test/Command/GenerateCommandTest.php at master · vagabundeo/DrupalConsole · GitHub

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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
<?php

namespace Drupal\AppConsole\Test\Command;

use Symfony\Component\Console\Helper\FormatterHelper;
use Symfony\Component\DependencyInjection\Container;
use Symfony\Component\Console\Helper\HelperSet;
use Drupal\AppConsole\Command\Helper\DialogHelper;

abstract class GenerateCommandTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @return \Symfony\Component\DependencyInjection\Container Drupal container
     */
    protected function getContainer()
    {
        $container = new Container();
        $container->set('twig', new \Twig_Environment());
        return $container;
    }

    protected function getHelperSet($input)
    {
        $dialog = new DialogHelper();
        $dialog->setInputStream($this->getInputStream($input));

        $autoload = $this
          ->getMockBuilder('Drupal\AppConsole\Command\Helper\DrupalAutoloadHelper')
          ->setMethods(['findAutoload', 'getDrupalRoot'])
          ->getMock();

        $stringUtils = $this->getMockBuilder('Drupal\AppConsole\Utils\StringUtils')
          ->disableOriginalConstructor()
          ->setMethods(['createMachineName'])
          ->getMock();

        $stringUtils->expects($this->any())
          ->method('createMachineName')
          ->will($this->returnArgument(0));

        $validators = $this->getMockBuilder('Drupal\AppConsole\Utils\Validators')
          ->disableOriginalConstructor()
          ->setMethods(['validateModuleName'])
          ->getMock();

        $validators->expects($this->any())
          ->method('validateModuleName')
          ->will($this->returnArgument(0));

        $translator = $this->getTranslationHelper();

        $message = $this
          ->getMockBuilder('Drupal\AppConsole\Command\Helper\MessageHelper')
          ->disableOriginalConstructor()
          ->setMethods(['showMessages', 'showMessage'])
          ->getMock();

        $chain = $this
          ->getMockBuilder('Drupal\AppConsole\Command\Helper\ChainCommandHelper')
          ->disableOriginalConstructor()
          ->setMethods(['addCommand', 'getCommands'])
          ->getMock();

        return new HelperSet([
          'formatter' => new FormatterHelper(),
          'drupal-autoload' => $autoload,
          'dialog' => $dialog,
          'stringUtils' => $stringUtils,
          'validators' => $validators,
          'translator' => $translator,
          'message' => $message,
          'chain' => $chain,
        ]);
    }

    protected function getInputStream($input)
    {
        $stream = fopen('php://memory', 'r+', false);
        fputs($stream, $input . str_repeat("\n", 10));
        rewind($stream);

        return $stream;
    }

    protected function getTranslationHelper()
    {
        return $this
          ->getMockBuilder('Drupal\AppConsole\Command\Helper\TranslatorHelper')
          ->disableOriginalConstructor()
          ->setMethods(['loadResource', 'trans'])
          ->getMock();
    }
}