1
0
mirror of https://github.com/django/django.git synced 2025-10-23 21:59:11 +00:00

Fixed #19918 -- Modified select_for_update to run on the write database.

This commit is contained in:
Alex Cucu
2013-02-26 11:51:56 +02:00
committed by Tim Graham
parent 1280675834
commit 1c64a0f29e
2 changed files with 14 additions and 1 deletions

View File

@@ -5,11 +5,13 @@ import time
import unittest
from django.conf import settings
from django.db import transaction, connection
from django.db import transaction, connection, router
from django.db.utils import ConnectionHandler, DEFAULT_DB_ALIAS, DatabaseError
from django.test import (TransactionTestCase, skipIfDBFeature,
skipUnlessDBFeature)
from multiple_database.tests import TestRouter
from .models import Person
# Some tests require threading, which might not be available. So create a
@@ -254,3 +256,13 @@ class SelectForUpdateTests(TransactionTestCase):
"""
people = list(Person.objects.select_for_update())
self.assertTrue(transaction.is_dirty())
@skipUnlessDBFeature('has_select_for_update')
def test_select_for_update_on_multidb(self):
old_routers = router.routers
try:
router.routers = [TestRouter()]
query = Person.objects.select_for_update()
self.assertEqual(router.db_for_write(Person), query.db)
finally:
router.routers = old_routers