Merge pull request #671 from CakeDC/feature/add-table-to-behavior-con…

…structor add table to behavior constructor
cakephp · Apr 10, 2020 · a44c956 · a44c956
2 parents 6491dc1 + 80835f8
commit a44c956
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/src/Command/TestCommand.php b/src/Command/TestCommand.php
@@ -550,7 +550,11 @@ public function generateConstructor(string $type, string $fullClassName): array
                 "? [] : ['className' => {$className}::class];";
             $construct = "TableRegistry::getTableLocator()->get('{$tableName}', \$config);";
         }
-        if ($type === 'Behavior' || $type === 'Entity' || $type === 'Form') {
+        if ($type === 'Behavior') {
+            $pre = "\$table = new Table();";
+            $construct = "new {$className}(\$table);";
+        }
+        if ($type === 'Entity' || $type === 'Form') {
             $construct = "new {$className}();";
         }
         if ($type === 'Helper') {
@@ -675,6 +679,9 @@ public function generateUses(string $type, string $fullClassName): array
             $uses[] = 'Cake\TestSuite\Stub\ConsoleOutput';
             $uses[] = 'Cake\Console\ConsoleIo';
         }
+        if ($type === 'Behavior') {
+            $uses[] = 'Cake\ORM\Table';
+        }
         $uses[] = $fullClassName;
 
         return $uses;

diff --git a/tests/TestCase/Command/TestCommandTest.php b/tests/TestCase/Command/TestCommandTest.php
@@ -648,6 +648,14 @@ public function testGenerateConstructor()
             '',
         ];
         $this->assertEquals($expected, $result);
+
+        $result = $command->generateConstructor('Behavior', 'App\Model\Behavior\PostsBehavior');
+        $expected = [
+            '$table = new Table();',
+            'new PostsBehavior($table);',
+            '',
+        ];
+        $this->assertEquals($expected, $result);
     }
 
     /**

diff --git a/tests/comparisons/Test/testBakeBehaviorTest.php b/tests/comparisons/Test/testBakeBehaviorTest.php
@@ -4,6 +4,7 @@
 namespace Bake\Test\App\Test\TestCase\Model\Behavior;
 
 use Bake\Test\App\Model\Behavior\ExampleBehavior;
+use Cake\ORM\Table;
 use Cake\TestSuite\TestCase;
 
 /**
@@ -26,7 +27,8 @@ class ExampleBehaviorTest extends TestCase
     public function setUp(): void
     {
         parent::setUp();
-        $this->Example = new ExampleBehavior();
+        $table = new Table();
+        $this->Example = new ExampleBehavior($table);
     }
 
     /**