/**
  * @covers Kunstmaan\MediaBundle\Helper\MediaManager::getHandler
  */
 public function testGetHandlerWithCustomHandler()
 {
     $media = new Media();
     $handler = $this->getCustomHandler($media);
     $this->object->addHandler($handler);
     $this->assertEquals($handler, $this->object->getHandler($media));
 }
 /**
  * @param object $entity The entity
  * @param bool $new Is new
  */
 private function saveMedia($entity, $new = false)
 {
     if (!$entity instanceof Media) {
         return;
     }
     $this->mediaManager->saveMedia($entity, $new);
     $url = $entity->getUrl();
     $handler = $this->mediaManager->getHandler($entity);
     if (isset($this->fileUrlMap[$url]) && $handler instanceof FileHandler) {
         $handler->fileSystem->rename(preg_replace('~^' . preg_quote($handler->mediaPath, '~') . '~', '/', $this->fileUrlMap[$url]), preg_replace('~^' . preg_quote($handler->mediaPath, '~') . '~', '/', $url));
         unset($this->fileUrlMap[$url]);
     }
 }