fixtures for tag

1 files changed, 19 insertions, 0 deletions

diff --git a/src/Wallabag/CoreBundle/DataFixtures/ORM/LoadTagData.php b/src/Wallabag/CoreBundle/DataFixtures/ORM/LoadTagData.php
index 6b13c2be..4d9846b6 100644
--- a/src/Wallabag/CoreBundle/DataFixtures/ORM/LoadTagData.php
+++ b/src/Wallabag/CoreBundle/DataFixtures/ORM/LoadTagData.php
@@ -6,6 +6,7 @@ use Doctrine\Common\DataFixtures\AbstractFixture;
 use Doctrine\Common\DataFixtures\OrderedFixtureInterface;
 use Doctrine\Common\Persistence\ObjectManager;
 use Wallabag\CoreBundle\Entity\Tag;
+use Wallabag\CoreBundle\Entity\TagsEntries;
 
 class LoadTagData extends AbstractFixture implements OrderedFixtureInterface
 {
@@ -19,16 +20,34 @@ class LoadTagData extends AbstractFixture implements OrderedFixtureInterface
 
         $manager->persist($tag1);
 
+        $this->addReference('tag1', $tag1);
+
+        $tagsEntries1 = new TagsEntries();
+        $tagsEntries1->setEntryId($this->getReference('entry1'));
+        $manager->persist($tagsEntries1);
+
         $tag2 = new Tag();
         $tag2->setLabel('bar');
 
         $manager->persist($tag2);
 
+        $this->addReference('tag2', $tag2);
+
+        $tagsEntries2 = new TagsEntries();
+        $tagsEntries2->setEntryId($this->getReference('entry2'));
+        $manager->persist($tagsEntries2);
+
         $tag3 = new Tag();
         $tag3->setLabel('baz');
 
         $manager->persist($tag3);
 
+        $this->addReference('tag3', $tag3);
+
+        $tagsEntries3 = new TagsEntries();
+        $tagsEntries3->setEntryId($this->getReference('entry2'));
+        $manager->persist($tagsEntries3);
+
         $manager->flush();
     }