Add method forUpdate (#465)

cycle · Mar 29, 2024 · 9ce5e6f · 9ce5e6f
1 parent 230dc9f
commit 9ce5e6f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/Select/Repository.php b/src/Select/Repository.php
@@ -59,4 +59,12 @@ public function select(): Select
     {
         return clone $this->select;
     }
+
+    public function forUpdate(): static
+    {
+        $repository = clone $this;
+        $repository->select->forUpdate();
+
+        return $repository;
+    }
 }
diff --git a/tests/ORM/Functional/Driver/SQLite/Select/RepositoryTest.php b/tests/ORM/Functional/Driver/SQLite/Select/RepositoryTest.php
@@ -5,6 +5,7 @@
 namespace Cycle\ORM\Tests\Functional\Driver\SQLite\Select;
 
 // phpcs:ignore
+use Cycle\ORM\Tests\Fixtures\User;
 use Cycle\ORM\Tests\Functional\Driver\Common\Select\RepositoryTest as CommonClass;
 
 /**
@@ -14,4 +15,19 @@
 class RepositoryTest extends CommonClass
 {
     public const DRIVER = 'sqlite';
+
+    /**
+     * This test does not need to be executed for each database driver.
+     */
+    public function testForUpdate(): void
+    {
+        $repository = $this->orm->getRepository(User::class);
+
+        $this->assertFalse($repository->select()->getBuilder()->getQuery()->getTokens()['forUpdate']);
+
+        $forUpdate = $repository->forUpdate();
+
+        $this->assertTrue($forUpdate->select()->getBuilder()->getQuery()->getTokens()['forUpdate']);
+        $this->assertNotSame($repository, $forUpdate);
+    }
 }