public function testGetSet()
 {
     $obj = new TypesToDocumentClasses();
     $obj->set('my_real_index', 'my_real_type', 'AppBundle:Type');
     $res = $obj->get('my_real_index', 'my_real_type');
     $this->assertEquals('AppBundle:Type', $res);
     $obj->set(null, 'second_real_type', 'AppBundle:Type');
     $res = $obj->get('second_real_index', 'second_real_type');
     $this->assertEquals('AppBundle:Type', $res);
     $this->setExpectedException('InvalidArgumentException');
     $obj->get('non_existing_index', 'my_real_type');
 }
 /**
  * Converts raw array to document.
  *
  * @param array $rawData
  *
  * @return DocumentInterface
  *
  * @throws \LogicException
  */
 private function convertToDocument($rawData)
 {
     $documentClass = $this->typesToDocumentClasses->get($rawData['_index'], $rawData['_type']);
     return $this->documentConverter->convertToDocument($rawData, $documentClass);
 }