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()