Merge pull request #15 from moufmouf/implement_interface

Adding useTrait and useTraitOnDao methods

Author: moufmouf

src/TdbmFluidTable.php

@@ -134,6 +134,32 @@ public function implementsInterfaceOnDao(string $interfaceName): self
         return $this;
     }
 
+    /**
+     * Makes the generated bean implement the trait $traitName
+     *
+     * @param string $traitName
+     * @param string[] $modifiers An array of modifiers. For instance: ['A::foo insteadof B']
+     * @return TdbmFluidTable
+     */
+    public function useTrait(string $traitName, array $modifiers = []): self
+    {
+        $this->addAnnotation('AddTrait', ['name' => $traitName, 'modifiers' => $modifiers], false);
+        return $this;
+    }
+
+    /**
+     * Makes the generated DAO implement the trait $traitName
+     *
+     * @param string $traitName
+     * @param string[] $modifiers An array of modifiers. For instance: ['A::foo insteadof B']
+     * @return TdbmFluidTable
+     */
+    public function useTraitOnDao(string $traitName, array $modifiers = []): self
+    {
+        $this->addAnnotation('AddTraitOnDao', ['name' => $traitName, 'modifiers' => $modifiers], false);
+        return $this;
+    }
+
     private function getComment(): Comment
     {
         $options = $this->fluidTable->getDbalTable()->getOptions();

tests/TdbmFluidTableTest.php

@@ -148,6 +148,31 @@ public function testImplementsInterfaceOnDao()
         $this->assertSame("\n@AddInterfaceOnDao(name = \"Foo\\Bar\")", $schema->getTable('posts')->getOptions()['comment']);
     }
 
+    public function testUseTrait()
+    {
+        $schema = new Schema();
+        $fluid = new TdbmFluidSchema($schema);
+
+        $posts = $fluid->table('posts');
+
+        $posts->useTrait('Foo\\Bar');
+        $posts->useTrait('Foo\\Bar2');
+
+        $this->assertSame("\n@AddTrait(name = \"Foo\\Bar\", modifiers = {})\n@AddTrait(name = \"Foo\\Bar2\", modifiers = {})", $schema->getTable('posts')->getOptions()['comment']);
+    }
+
+    public function testUseTraitOnDao()
+    {
+        $schema = new Schema();
+        $fluid = new TdbmFluidSchema($schema);
+
+        $posts = $fluid->table('posts');
+
+        $posts->useTraitOnDao('Foo\\Bar');
+
+        $this->assertSame("\n@AddTraitOnDao(name = \"Foo\\Bar\", modifiers = {})", $schema->getTable('posts')->getOptions()['comment']);
+    }
+
     public function testTimestamps()
     {
         if (defined('Doctrine\\DBAL\\Types\\Type::DATE_IMMUTABLE')) {