Add tests

VincentLanglet · VincentLanglet · commit 12b0f44a0c4b · 2024-03-03T21:27:17.000+01:00
diff --git a/src/Symfony/Component/Console/Command/LockableTrait.php b/src/Symfony/Component/Console/Command/LockableTrait.php
@@ -28,15 +28,6 @@ trait LockableTrait
 
     private ?LockFactory $lockFactory = null;
 
-    public function setLockFactory(?LockFactory $lockFactory): void
-    {
-        if (null !== $this->lock) {
-            throw new LogicException('You cannot set the lockFactory if a lock is already in place.');
-        }
-
-        $this->lockFactory = $lockFactory;
-    }
-
     /**
      * Locks a command.
      */
diff --git a/src/Symfony/Component/Console/Tests/Command/LockableTraitTest.php b/src/Symfony/Component/Console/Tests/Command/LockableTraitTest.php
@@ -14,6 +14,7 @@
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\Console\Tester\CommandTester;
 use Symfony\Component\Lock\LockFactory;
+use Symfony\Component\Lock\LockInterface;
 use Symfony\Component\Lock\Store\FlockStore;
 use Symfony\Component\Lock\Store\SemaphoreStore;
 
@@ -26,6 +27,7 @@ public static function setUpBeforeClass(): void
         self::$fixturesPath = __DIR__.'/../Fixtures/';
         require_once self::$fixturesPath.'/FooLockCommand.php';
         require_once self::$fixturesPath.'/FooLock2Command.php';
+        require_once self::$fixturesPath.'/FooLock3Command.php';
     }
 
     public function testLockIsReleased()
@@ -64,4 +66,18 @@ public function testMultipleLockCallsThrowLogicException()
         $tester = new CommandTester($command);
         $this->assertSame(1, $tester->execute([]));
     }
+
+    public function testCustomLockFactoryIsUsed()
+    {
+        $lockFactory = $this->createMock(LockFactory::class);
+        $command = new \FooLock3Command($lockFactory);
+
+        $tester = new CommandTester($command);
+
+        $lock = $this->createMock(LockInterface::class);
+        $lock->method('acquire')->willReturn(false);
+
+        $lockFactory->expects(static::once())->method('createLock')->willReturn($lock);
+        $this->assertSame(1, $tester->execute([]));
+    }
 }
diff --git a/src/Symfony/Component/Console/Tests/Fixtures/FooLock3Command.php b/src/Symfony/Component/Console/Tests/Fixtures/FooLock3Command.php
@@ -0,0 +1,35 @@
+<?php
+
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Command\LockableTrait;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Lock\LockFactory;
+
+class FooLock3Command extends Command
+{
+    use LockableTrait;
+
+    public function __construct(LockFactory $lockFactory)
+    {
+        parent::__construct();
+
+        $this->lockFactory = $lockFactory;
+    }
+
+    protected function configure(): void
+    {
+        $this->setName('foo:lock3');
+    }
+
+    protected function execute(InputInterface $input, OutputInterface $output): int
+    {
+        if (!$this->lock()) {
+            return 1;
+        }
+
+        $this->release();
+
+        return 2;
+    }
+}
