Add test for lock_migration_table

1 files changed, 17 insertions, 1 deletions

diff --git a/yoyo/tests/test_backends.py b/yoyo/tests/test_backends.py
index 8b8ff81..d567689 100644
--- a/yoyo/tests/test_backends.py
+++ b/yoyo/tests/test_backends.py
@@ -1,6 +1,8 @@
 import pytest
+import time
+from threading import Thread
 
-from yoyo import backends
+from yoyo import backends, read_migrations
 from yoyo.tests import get_test_backends
 from yoyo.tests import with_migrations
 
@@ -112,3 +114,17 @@ class TestTransactionHandling(object):
             migrations = read_migrations(tmpdir)
             backend.apply_migrations(migrations)
             backend.rollback_migrations(migrations)
+
+
+    @with_migrations(a="""
+		steps = [
+			step("SELECT pg_sleep(1)"),
+		]
+    """)
+    def test_lock_migration_table(self, tmpdir):
+        backend = get_test_backends(only={'postgresql'})[0]
+        migrations = read_migrations(tmpdir)
+        Thread(target = backend.apply_migrations, args = (migrations,)).start()
+        # give a chance to start, but wake up in the middle of applying
+        time.sleep(0.1)
+        assert backend.get_applied_migration_ids() == ['a']