--- /dev/null
+<?php
+/**
+ * @file
+ * Contains \Drupal\Console\Test\Command\GeneratorEntityCommandTest.
+ */
+
+namespace Drupal\Console\Test\Command;
+
+use Drupal\Console\Command\Generate\EntityConfigCommand;
+use Symfony\Component\Console\Tester\CommandTester;
+use Drupal\Console\Test\DataProvider\EntityDataProviderTrait;
+
+class GeneratorEntityCommandTest extends GenerateCommandTest
+{
+ use EntityDataProviderTrait;
+
+ /**
+ * EntityConfig generator test
+ *
+ * @param $module
+ * @param $entity_class
+ * @param $entity_name
+ * @param $label
+ *
+ * @dataProvider commandData
+ */
+ public function testGenerateEntityConfig(
+ $module,
+ $entity_class,
+ $entity_name,
+ $label
+ ) {
+ $command = new EntityConfigCommand($this->getHelperSet());
+ $command->setHelperSet($this->getHelperSet());
+ $command->setGenerator($this->getGenerator());
+
+ $commandTester = new CommandTester($command);
+
+ $code = $commandTester->execute(
+ [
+ '--module' => $module,
+ '--entity-class' => $entity_class,
+ '--entity-name' => $entity_name,
+ '--label' => $label
+ ],
+ ['interactive' => false]
+ );
+
+ $this->assertEquals(0, $code);
+ }
+
+ private function getGenerator()
+ {
+ return $this
+ ->getMockBuilder('Drupal\Console\Generator\EntityConfigGenerator')
+ ->disableOriginalConstructor()
+ ->setMethods(['generate'])
+ ->getMock();
+ }
+}