1 files changed, 10 insertions, 0 deletions
diff --git a/tests/test_build.py b/tests/test_build.py
index a1a268d..fbfbb62 100644
--- a/tests/test_build.py
+++ b/tests/test_build.py
@@ -63,6 +63,16 @@ class TestBuild(unittest.TestCase):
                 "SELECT x FROM tbl WHERE x < 9",
             ),
             (
+                lambda: select("x").from_("tbl").where("x > 0").lock(),
+                "SELECT x FROM tbl WHERE x > 0 FOR UPDATE",
+                "mysql",
+            ),
+            (
+                lambda: select("x").from_("tbl").where("x > 0").lock(update=False),
+                "SELECT x FROM tbl WHERE x > 0 FOR SHARE",
+                "postgres",
+            ),
+            (
                 lambda: select("x", "y").from_("tbl").group_by("x"),
                 "SELECT x, y FROM tbl GROUP BY x",
             ),