Ticket #26933: 26933.diff

tests/get_or_create/tests.py

@@ -441 +441 @@
         while it holds the lock. The updated field isn't a field in 'defaults',
         so update_or_create() shouldn't have an effect on it.
         """
+
+        lock_status = {'has_grabbed_lock': False}
+
         def birthday_sleep():
-            time.sleep(0.3)
+            lock_status['has_grabbed_lock'] = True
+            time.sleep(0.5)
             return date(1940, 10, 10)
 
         def update_birthday_slowly():
@@ -450 +454 @@
                 first_name='John', defaults={'birthday': birthday_sleep}
             )
 
+        def lock_wait():
+            # timeout after ~0.5 seconds
+            for i in range(20):
+                time.sleep(0.025)
+                if lock_status['has_grabbed_lock']:
+                    break
+            else:
+                return False
+            return True
+
         Person.objects.create(first_name='John', last_name='Lennon', birthday=date(1940, 10, 9))
 
         # update_or_create in a separate thread
@@ -457 +471 @@
         before_start = datetime.now()
         t.start()
 
-        # Wait for lock to begin
-        time.sleep(0.05)
+        if not lock_wait():
+            self.skipTest('Database took too long to lock the row')
 
         # Update during lock
         Person.objects.filter(first_name='John').update(last_name='NotLennon')
@@ -469 +483 @@
 
         # The update remains and it blocked.
         updated_person = Person.objects.get(first_name='John')
-        self.assertGreater(after_update - before_start, timedelta(seconds=0.3))
+        self.assertGreater(after_update - before_start, timedelta(seconds=0.5))
         self.assertEqual(updated_person.last_name, 'NotLennon')