Merge pull request #108 from linaori/fix/lazy-factory-recursion

Prevent recursion when the lazy adapter refers to itself

src/Lazy/LazyFactory.php

@@ -29,6 +29,10 @@ public function __construct(ContainerInterface $storages)
 
     public function createStorage(string $source, string $storageName)
     {
+        if ($source === $storageName) {
+            throw new \InvalidArgumentException('The "lazy" adapter source is referring to itself as "'.$source.'", which would lead to infinite recursion.');
+        }
+
         if (!$this->storages->has($source)) {
             throw new \InvalidArgumentException('You have requested a non-existent source storage "'.$source.'" in lazy storage "'.$storageName.'".');
         }

tests/Lazy/LazyFactoryTest.php

@@ -0,0 +1,55 @@
+<?php
+
+/*
+ * This file is part of the flysystem-bundle project.
+ *
+ * (c) Titouan Galopin <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Tests\League\FlysystemBundle\Lazy;
+
+use League\Flysystem\InMemory\InMemoryFilesystemAdapter;
+use League\FlysystemBundle\Lazy\LazyFactory;
+use PHPUnit\Framework\TestCase;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+
+class LazyFactoryTest extends TestCase
+{
+    public function testItPreventsInfiniteRecursion(): void
+    {
+        $containerBuilder = new ContainerBuilder();
+        $factory = new LazyFactory($containerBuilder);
+
+        $this->expectException(\InvalidArgumentException::class);
+        $this->expectExceptionMessage('The "lazy" adapter source is referring to itself as "lazy_storage", which would lead to infinite recursion.');
+        $factory->createStorage('lazy_storage', 'lazy_storage');
+    }
+
+    public function testItErrorsWhenServiceIsNotAvailable(): void
+    {
+        $containerBuilder = new ContainerBuilder();
+        $factory = new LazyFactory($containerBuilder);
+
+        $this->expectException(\InvalidArgumentException::class);
+        $this->expectExceptionMessage('You have requested a non-existent source storage "source_storage" in lazy storage "lazy_storage".');
+        $factory->createStorage('source_storage', 'lazy_storage');
+    }
+
+    public function testItReturnsTheRequestedStorageService(): void
+    {
+        $actual = new InMemoryFilesystemAdapter();
+        $lazy = new InMemoryFilesystemAdapter();
+
+        $containerBuilder = new ContainerBuilder();
+        $containerBuilder->set('source_storage', $actual);
+        $containerBuilder->set('lazy_storage', $lazy);
+
+        $factory = new LazyFactory($containerBuilder);
+
+        $adapter = $factory->createStorage('source_storage', 'lazy_storage');
+        self::assertSame($actual, $adapter);
+    }
+}