From 0e5f568fb490abcd0c4761e6c7f4fe486b91b189 Mon Sep 17 00:00:00 2001
From: Patryk <drapikpatryk@gmail.com>
Date: Fri, 6 Jul 2018 13:00:38 +0200
Subject: [PATCH] Fix tests

---
 src/Resolver/MediaProviderResolver.php     | 2 +-
 tests/Behat/Context/Setup/MediaContext.php | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Resolver/MediaProviderResolver.php b/src/Resolver/MediaProviderResolver.php
index 157b62549..5d640cef7 100755
--- a/src/Resolver/MediaProviderResolver.php
+++ b/src/Resolver/MediaProviderResolver.php
@@ -29,7 +29,7 @@ public function __construct(ServiceRegistryInterface $providerRegistry)
     public function resolveProvider(MediaInterface $media): ProviderInterface
     {
         /** @var ProviderInterface $provider */
-        $provider = $this->providerRegistry->get($media->getFileType());
+        $provider = $this->providerRegistry->get($media->getType());
 
         return $provider;
     }
diff --git a/tests/Behat/Context/Setup/MediaContext.php b/tests/Behat/Context/Setup/MediaContext.php
index aef99394d..5366d3239 100755
--- a/tests/Behat/Context/Setup/MediaContext.php
+++ b/tests/Behat/Context/Setup/MediaContext.php
@@ -127,7 +127,7 @@ private function createMedia(
         $media->setCurrentLocale('en_US');
         $media->setName($name);
         $media->setDescription($description);
-        $media->setFileType($fileType);
+        $media->setType($fileType);
 
         return $media;
     }