From 44ee7588ca8841f24cd1bad1ca52d38ecac6b9cb Mon Sep 17 00:00:00 2001
From: fsevestre <aniki.taicho@gmail.com>
Date: Thu, 1 Oct 2015 21:56:58 +0200
Subject: [PATCH] Return entity after writeItem() method call on DoctrineWriter

---
 src/Writer/DoctrineWriter.php       | 6 +++++-
 tests/Writer/DoctrineWriterTest.php | 4 +++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/Writer/DoctrineWriter.php b/src/Writer/DoctrineWriter.php
index fca7cc38..88d97771 100644
--- a/src/Writer/DoctrineWriter.php
+++ b/src/Writer/DoctrineWriter.php
@@ -170,7 +170,9 @@ public function finish()
     }
 
     /**
-     * {@inheritdoc}
+     * @param array $item
+     *
+     * @return object
      */
     public function writeItem(array $item)
     {
@@ -180,6 +182,8 @@ public function writeItem(array $item)
         $this->updateEntity($item, $entity);
 
         $this->entityManager->persist($entity);
+
+        return $entity;
     }
 
     /**
diff --git a/tests/Writer/DoctrineWriterTest.php b/tests/Writer/DoctrineWriterTest.php
index 5792512e..62335689 100644
--- a/tests/Writer/DoctrineWriterTest.php
+++ b/tests/Writer/DoctrineWriterTest.php
@@ -22,7 +22,9 @@ public function testWriteItem()
             'secondProperty'  => 'some other value',
             'firstAssociation'=> $association
         );
-        $writer->writeItem($item);
+        $entity = $writer->writeItem($item);
+
+        $this->assertInstanceOf('Ddeboer\DataImport\Tests\Fixtures\Entity\TestEntity', $entity);
     }
 
     protected function getEntityManager()