/**
  * Creates and returns Products array filled with test data.
  *
  * @param Repository $repository
  *
  * @return array
  */
 protected function getDocumentsData($repository)
 {
     $document = $repository->createDocument();
     $document->setId('test-product-1');
     $document->title = 'Test title';
     $document->setScore(0.0);
     $this->getManager()->persist($document);
     $document2 = $repository->createDocument();
     $document2->setId('test-product-2');
     $document2->title = 'Test title2';
     $document2->setScore(0.0);
     $this->getManager()->persist($document2);
     $this->getManager()->commit();
     $return[$document->getId()] = $document;
     $return[$document2->getId()] = $document2;
     return $return;
 }
 /**
  * {@inheritdoc}
  */
 public function current()
 {
     $doctrineEntity = parent::current();
     return new ImportItem($doctrineEntity[0], $this->repository->createDocument());
 }