Add a couple tests on read database

typeorm · Aug 25, 2022 · 9e7ecf7 · 9e7ecf7
1 parent 5a9ae8f
commit 9e7ecf7
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/test/functional/query-builder/locking/query-builder-locking.ts b/test/functional/query-builder/locking/query-builder-locking.ts
@@ -1169,6 +1169,49 @@ describe("query builder > locking", () => {
             }),
         ))
 
+    it('skip_locked with "pessimistic_read" check getOne', () =>
+        Promise.all(
+            connections.map(async (connection) => {
+                if (
+                    connection.driver.options.type === "postgres" ||
+                    (connection.driver.options.type === "mysql" &&
+                        DriverUtils.isReleaseVersionOrGreater(
+                            connection.driver,
+                            "8.0.0",
+                        ))
+                ) {
+                    return connection.manager.transaction((entityManager) => {
+                        return Promise.resolve(
+                            entityManager
+                                .createQueryBuilder(PostWithVersion, "post")
+                                .setLock("pessimistic_read")
+                                .setOnLocked("skip_locked")
+                                .where("post.id = :id", { id: 1 })
+                                .getOne().should.not.be.rejected,
+                        )
+                    })
+                }
+            }),
+        ))
+
+    it('skip_locked with "for_key_share" check getOne', () =>
+        Promise.all(
+            connections.map(async (connection) => {
+                if (connection.driver.options.type === "postgres") {
+                    return connection.manager.transaction((entityManager) => {
+                        return Promise.resolve(
+                            entityManager
+                                .createQueryBuilder(PostWithVersion, "post")
+                                .setLock("for_key_share")
+                                .setOnLocked("skip_locked")
+                                .where("post.id = :id", { id: 1 })
+                                .getOne().should.not.be.rejected,
+                        )
+                    })
+                }
+            }),
+        ))
+
     it('skip_locked with "pessimistic_read" fails on early versions of MySQL', () =>
         Promise.all(
             connections.map(async (connection) => {