2110 lines
90 KiB
Python
2110 lines
90 KiB
Python
import datetime
|
|
import pickle
|
|
from io import StringIO
|
|
from operator import attrgetter
|
|
from unittest.mock import Mock
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.core import management
|
|
from django.db import DEFAULT_DB_ALIAS, connections, router, transaction
|
|
from django.db.models import signals
|
|
from django.db.utils import ConnectionRouter
|
|
from django.test import SimpleTestCase, TestCase, override_settings
|
|
|
|
from .models import Book, Person, Pet, Review, UserProfile
|
|
from .routers import AuthRouter, TestRouter, WriteRouter
|
|
|
|
|
|
class QueryTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def test_db_selection(self):
|
|
"Querysets will use the default database by default"
|
|
self.assertEqual(Book.objects.db, DEFAULT_DB_ALIAS)
|
|
self.assertEqual(Book.objects.all().db, DEFAULT_DB_ALIAS)
|
|
|
|
self.assertEqual(Book.objects.using('other').db, 'other')
|
|
|
|
self.assertEqual(Book.objects.db_manager('other').db, 'other')
|
|
self.assertEqual(Book.objects.db_manager('other').all().db, 'other')
|
|
|
|
def test_default_creation(self):
|
|
"Objects created on the default database don't leak onto other databases"
|
|
# Create a book on the default database using create()
|
|
Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
|
|
# Create a book on the default database using a save
|
|
dive = Book()
|
|
dive.title = "Dive into Python"
|
|
dive.published = datetime.date(2009, 5, 4)
|
|
dive.save()
|
|
|
|
# Book exists on the default database, but not on other database
|
|
try:
|
|
Book.objects.get(title="Pro Django")
|
|
Book.objects.using('default').get(title="Pro Django")
|
|
except Book.DoesNotExist:
|
|
self.fail('"Pro Django" should exist on default database')
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('other').get(title="Pro Django")
|
|
|
|
try:
|
|
Book.objects.get(title="Dive into Python")
|
|
Book.objects.using('default').get(title="Dive into Python")
|
|
except Book.DoesNotExist:
|
|
self.fail('"Dive into Python" should exist on default database')
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('other').get(title="Dive into Python")
|
|
|
|
def test_other_creation(self):
|
|
"Objects created on another database don't leak onto the default database"
|
|
# Create a book on the second database
|
|
Book.objects.using('other').create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
# Create a book on the default database using a save
|
|
dive = Book()
|
|
dive.title = "Dive into Python"
|
|
dive.published = datetime.date(2009, 5, 4)
|
|
dive.save(using='other')
|
|
|
|
# Book exists on the default database, but not on other database
|
|
try:
|
|
Book.objects.using('other').get(title="Pro Django")
|
|
except Book.DoesNotExist:
|
|
self.fail('"Pro Django" should exist on other database')
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.get(title="Pro Django")
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(title="Pro Django")
|
|
|
|
try:
|
|
Book.objects.using('other').get(title="Dive into Python")
|
|
except Book.DoesNotExist:
|
|
self.fail('"Dive into Python" should exist on other database')
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.get(title="Dive into Python")
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(title="Dive into Python")
|
|
|
|
def test_refresh(self):
|
|
dive = Book(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
dive.save(using='other')
|
|
dive2 = Book.objects.using('other').get()
|
|
dive2.title = "Dive into Python (on default)"
|
|
dive2.save(using='default')
|
|
dive.refresh_from_db()
|
|
self.assertEqual(dive.title, "Dive into Python")
|
|
dive.refresh_from_db(using='default')
|
|
self.assertEqual(dive.title, "Dive into Python (on default)")
|
|
self.assertEqual(dive._state.db, "default")
|
|
|
|
def test_refresh_router_instance_hint(self):
|
|
router = Mock()
|
|
router.db_for_read.return_value = None
|
|
book = Book.objects.create(title='Dive Into Python', published=datetime.date(1957, 10, 12))
|
|
with self.settings(DATABASE_ROUTERS=[router]):
|
|
book.refresh_from_db()
|
|
router.db_for_read.assert_called_once_with(Book, instance=book)
|
|
|
|
def test_basic_queries(self):
|
|
"Queries are constrained to a single database"
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
dive = Book.objects.using('other').get(published=datetime.date(2009, 5, 4))
|
|
self.assertEqual(dive.title, "Dive into Python")
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(published=datetime.date(2009, 5, 4))
|
|
|
|
dive = Book.objects.using('other').get(title__icontains="dive")
|
|
self.assertEqual(dive.title, "Dive into Python")
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(title__icontains="dive")
|
|
|
|
dive = Book.objects.using('other').get(title__iexact="dive INTO python")
|
|
self.assertEqual(dive.title, "Dive into Python")
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(title__iexact="dive INTO python")
|
|
|
|
dive = Book.objects.using('other').get(published__year=2009)
|
|
self.assertEqual(dive.title, "Dive into Python")
|
|
self.assertEqual(dive.published, datetime.date(2009, 5, 4))
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(published__year=2009)
|
|
|
|
years = Book.objects.using('other').dates('published', 'year')
|
|
self.assertEqual([o.year for o in years], [2009])
|
|
years = Book.objects.using('default').dates('published', 'year')
|
|
self.assertEqual([o.year for o in years], [])
|
|
|
|
months = Book.objects.using('other').dates('published', 'month')
|
|
self.assertEqual([o.month for o in months], [5])
|
|
months = Book.objects.using('default').dates('published', 'month')
|
|
self.assertEqual([o.month for o in months], [])
|
|
|
|
def test_m2m_separation(self):
|
|
"M2M fields are constrained to a single database"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
marty = Person.objects.create(name="Marty Alchin")
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
|
|
# Save the author relations
|
|
pro.authors.set([marty])
|
|
dive.authors.set([mark])
|
|
|
|
# Inspect the m2m tables directly.
|
|
# There should be 1 entry in each database
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 1)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 1)
|
|
|
|
# Queries work across m2m joins
|
|
self.assertEqual(
|
|
list(Book.objects.using('default').filter(authors__name='Marty Alchin').values_list('title', flat=True)),
|
|
['Pro Django']
|
|
)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Marty Alchin').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
|
|
self.assertEqual(
|
|
list(Book.objects.using('default').filter(authors__name='Mark Pilgrim').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Mark Pilgrim').values_list('title', flat=True)),
|
|
['Dive into Python']
|
|
)
|
|
|
|
# Reget the objects to clear caches
|
|
dive = Book.objects.using('other').get(title="Dive into Python")
|
|
mark = Person.objects.using('other').get(name="Mark Pilgrim")
|
|
|
|
# Retrieve related object by descriptor. Related objects should be database-bound
|
|
self.assertEqual(list(dive.authors.all().values_list('name', flat=True)), ['Mark Pilgrim'])
|
|
|
|
self.assertEqual(list(mark.book_set.all().values_list('title', flat=True)), ['Dive into Python'])
|
|
|
|
def test_m2m_forward_operations(self):
|
|
"M2M forward manipulations are all constrained to a single DB"
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
|
|
# Save the author relations
|
|
dive.authors.set([mark])
|
|
|
|
# Add a second author
|
|
john = Person.objects.using('other').create(name="John Smith")
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='John Smith').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
|
|
dive.authors.add(john)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Mark Pilgrim').values_list('title', flat=True)),
|
|
['Dive into Python']
|
|
)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='John Smith').values_list('title', flat=True)),
|
|
['Dive into Python']
|
|
)
|
|
|
|
# Remove the second author
|
|
dive.authors.remove(john)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Mark Pilgrim').values_list('title', flat=True)),
|
|
['Dive into Python']
|
|
)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='John Smith').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
|
|
# Clear all authors
|
|
dive.authors.clear()
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Mark Pilgrim').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='John Smith').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
|
|
# Create an author through the m2m interface
|
|
dive.authors.create(name='Jane Brown')
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Mark Pilgrim').values_list('title', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Book.objects.using('other').filter(authors__name='Jane Brown').values_list('title', flat=True)),
|
|
['Dive into Python']
|
|
)
|
|
|
|
def test_m2m_reverse_operations(self):
|
|
"M2M reverse manipulations are all constrained to a single DB"
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
|
|
# Save the author relations
|
|
dive.authors.set([mark])
|
|
|
|
# Create a second book on the other database
|
|
grease = Book.objects.using('other').create(title="Greasemonkey Hacks", published=datetime.date(2005, 11, 1))
|
|
|
|
# Add a books to the m2m
|
|
mark.book_set.add(grease)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(book__title='Dive into Python').values_list('name', flat=True)),
|
|
['Mark Pilgrim']
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(book__title='Greasemonkey Hacks').values_list('name', flat=True)
|
|
),
|
|
['Mark Pilgrim']
|
|
)
|
|
|
|
# Remove a book from the m2m
|
|
mark.book_set.remove(grease)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(book__title='Dive into Python').values_list('name', flat=True)),
|
|
['Mark Pilgrim']
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(book__title='Greasemonkey Hacks').values_list('name', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
|
|
# Clear the books associated with mark
|
|
mark.book_set.clear()
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(book__title='Dive into Python').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(book__title='Greasemonkey Hacks').values_list('name', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
|
|
# Create a book through the m2m interface
|
|
mark.book_set.create(title="Dive into HTML5", published=datetime.date(2020, 1, 1))
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(book__title='Dive into Python').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(book__title='Dive into HTML5').values_list('name', flat=True)),
|
|
['Mark Pilgrim']
|
|
)
|
|
|
|
def test_m2m_cross_database_protection(self):
|
|
"Operations that involve sharing M2M objects across databases raise an error"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
|
|
marty = Person.objects.create(name="Marty Alchin")
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
# Set a foreign key set with an object from a different database
|
|
msg = (
|
|
'Cannot assign "<Person: Marty Alchin>": the current database '
|
|
'router prevents this relation.'
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='default'):
|
|
marty.edited.set([pro, dive])
|
|
|
|
# Add to an m2m with an object from a different database
|
|
msg = (
|
|
'Cannot add "<Book: Dive into Python>": instance is on '
|
|
'database "default", value is on database "other"'
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='default'):
|
|
marty.book_set.add(dive)
|
|
|
|
# Set a m2m with an object from a different database
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='default'):
|
|
marty.book_set.set([pro, dive])
|
|
|
|
# Add to a reverse m2m with an object from a different database
|
|
msg = (
|
|
'Cannot add "<Person: Marty Alchin>": instance is on '
|
|
'database "other", value is on database "default"'
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='other'):
|
|
dive.authors.add(marty)
|
|
|
|
# Set a reverse m2m with an object from a different database
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='other'):
|
|
dive.authors.set([mark, marty])
|
|
|
|
def test_m2m_deletion(self):
|
|
"Cascaded deletions of m2m relations issue queries on the right database"
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
dive.authors.set([mark])
|
|
|
|
# Check the initial state
|
|
self.assertEqual(Person.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 0)
|
|
|
|
self.assertEqual(Person.objects.using('other').count(), 1)
|
|
self.assertEqual(Book.objects.using('other').count(), 1)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 1)
|
|
|
|
# Delete the object on the other database
|
|
dive.delete(using='other')
|
|
|
|
self.assertEqual(Person.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 0)
|
|
|
|
# The person still exists ...
|
|
self.assertEqual(Person.objects.using('other').count(), 1)
|
|
# ... but the book has been deleted
|
|
self.assertEqual(Book.objects.using('other').count(), 0)
|
|
# ... and the relationship object has also been deleted.
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
# Now try deletion in the reverse direction. Set up the relation again
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
dive.authors.set([mark])
|
|
|
|
# Check the initial state
|
|
self.assertEqual(Person.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 0)
|
|
|
|
self.assertEqual(Person.objects.using('other').count(), 1)
|
|
self.assertEqual(Book.objects.using('other').count(), 1)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 1)
|
|
|
|
# Delete the object on the other database
|
|
mark.delete(using='other')
|
|
|
|
self.assertEqual(Person.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 0)
|
|
|
|
# The person has been deleted ...
|
|
self.assertEqual(Person.objects.using('other').count(), 0)
|
|
# ... but the book still exists
|
|
self.assertEqual(Book.objects.using('other').count(), 1)
|
|
# ... and the relationship object has been deleted.
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
def test_foreign_key_separation(self):
|
|
"FK fields are constrained to a single database"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
|
|
george = Person.objects.create(name="George Vilches")
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
chris = Person.objects.using('other').create(name="Chris Mills")
|
|
|
|
# Save the author's favorite books
|
|
pro.editor = george
|
|
pro.save()
|
|
|
|
dive.editor = chris
|
|
dive.save()
|
|
|
|
pro = Book.objects.using('default').get(title="Pro Django")
|
|
self.assertEqual(pro.editor.name, "George Vilches")
|
|
|
|
dive = Book.objects.using('other').get(title="Dive into Python")
|
|
self.assertEqual(dive.editor.name, "Chris Mills")
|
|
|
|
# Queries work across foreign key joins
|
|
self.assertEqual(
|
|
list(Person.objects.using('default').filter(edited__title='Pro Django').values_list('name', flat=True)),
|
|
['George Vilches']
|
|
)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Pro Django').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('default').filter(edited__title='Dive into Python').values_list('name', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(edited__title='Dive into Python').values_list('name', flat=True)
|
|
),
|
|
['Chris Mills']
|
|
)
|
|
|
|
# Reget the objects to clear caches
|
|
chris = Person.objects.using('other').get(name="Chris Mills")
|
|
dive = Book.objects.using('other').get(title="Dive into Python")
|
|
|
|
# Retrieve related object by descriptor. Related objects should be database-bound
|
|
self.assertEqual(list(chris.edited.values_list('title', flat=True)), ['Dive into Python'])
|
|
|
|
def test_foreign_key_reverse_operations(self):
|
|
"FK reverse manipulations are all constrained to a single DB"
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
chris = Person.objects.using('other').create(name="Chris Mills")
|
|
|
|
# Save the author relations
|
|
dive.editor = chris
|
|
dive.save()
|
|
|
|
# Add a second book edited by chris
|
|
html5 = Book.objects.using('other').create(title="Dive into HTML5", published=datetime.date(2010, 3, 15))
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Dive into HTML5').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
|
|
chris.edited.add(html5)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Dive into HTML5').values_list('name', flat=True)),
|
|
['Chris Mills']
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(edited__title='Dive into Python').values_list('name', flat=True)
|
|
),
|
|
['Chris Mills']
|
|
)
|
|
|
|
# Remove the second editor
|
|
chris.edited.remove(html5)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Dive into HTML5').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(edited__title='Dive into Python').values_list('name', flat=True)
|
|
),
|
|
['Chris Mills']
|
|
)
|
|
|
|
# Clear all edited books
|
|
chris.edited.clear()
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Dive into HTML5').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(edited__title='Dive into Python').values_list('name', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
|
|
# Create an author through the m2m interface
|
|
chris.edited.create(title='Dive into Water', published=datetime.date(2010, 3, 15))
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Dive into HTML5').values_list('name', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Person.objects.using('other').filter(edited__title='Dive into Water').values_list('name', flat=True)),
|
|
['Chris Mills']
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
Person.objects.using('other').filter(edited__title='Dive into Python').values_list('name', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
|
|
def test_foreign_key_cross_database_protection(self):
|
|
"Operations that involve sharing FK objects across databases raise an error"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
marty = Person.objects.create(name="Marty Alchin")
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
# Set a foreign key with an object from a different database
|
|
msg = (
|
|
'Cannot assign "<Person: Marty Alchin>": the current database '
|
|
'router prevents this relation.'
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
dive.editor = marty
|
|
|
|
# Set a foreign key set with an object from a different database
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='default'):
|
|
marty.edited.set([pro, dive])
|
|
|
|
# Add to a foreign key set with an object from a different database
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='default'):
|
|
marty.edited.add(dive)
|
|
|
|
def test_foreign_key_deletion(self):
|
|
"Cascaded deletions of Foreign Key relations issue queries on the right database"
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
Pet.objects.using('other').create(name="Fido", owner=mark)
|
|
|
|
# Check the initial state
|
|
self.assertEqual(Person.objects.using('default').count(), 0)
|
|
self.assertEqual(Pet.objects.using('default').count(), 0)
|
|
|
|
self.assertEqual(Person.objects.using('other').count(), 1)
|
|
self.assertEqual(Pet.objects.using('other').count(), 1)
|
|
|
|
# Delete the person object, which will cascade onto the pet
|
|
mark.delete(using='other')
|
|
|
|
self.assertEqual(Person.objects.using('default').count(), 0)
|
|
self.assertEqual(Pet.objects.using('default').count(), 0)
|
|
|
|
# Both the pet and the person have been deleted from the right database
|
|
self.assertEqual(Person.objects.using('other').count(), 0)
|
|
self.assertEqual(Pet.objects.using('other').count(), 0)
|
|
|
|
def test_foreign_key_validation(self):
|
|
"ForeignKey.validate() uses the correct database"
|
|
mickey = Person.objects.using('other').create(name="Mickey")
|
|
pluto = Pet.objects.using('other').create(name="Pluto", owner=mickey)
|
|
self.assertIsNone(pluto.full_clean())
|
|
|
|
# Any router that accesses `model` in db_for_read() works here.
|
|
@override_settings(DATABASE_ROUTERS=[AuthRouter()])
|
|
def test_foreign_key_validation_with_router(self):
|
|
"""
|
|
ForeignKey.validate() passes `model` to db_for_read() even if
|
|
model_instance=None.
|
|
"""
|
|
mickey = Person.objects.create(name="Mickey")
|
|
owner_field = Pet._meta.get_field('owner')
|
|
self.assertEqual(owner_field.clean(mickey.pk, None), mickey.pk)
|
|
|
|
def test_o2o_separation(self):
|
|
"OneToOne fields are constrained to a single database"
|
|
# Create a user and profile on the default database
|
|
alice = User.objects.db_manager('default').create_user('alice', 'alice@example.com')
|
|
alice_profile = UserProfile.objects.using('default').create(user=alice, flavor='chocolate')
|
|
|
|
# Create a user and profile on the other database
|
|
bob = User.objects.db_manager('other').create_user('bob', 'bob@example.com')
|
|
bob_profile = UserProfile.objects.using('other').create(user=bob, flavor='crunchy frog')
|
|
|
|
# Retrieve related objects; queries should be database constrained
|
|
alice = User.objects.using('default').get(username="alice")
|
|
self.assertEqual(alice.userprofile.flavor, "chocolate")
|
|
|
|
bob = User.objects.using('other').get(username="bob")
|
|
self.assertEqual(bob.userprofile.flavor, "crunchy frog")
|
|
|
|
# Queries work across joins
|
|
self.assertEqual(
|
|
list(
|
|
User.objects.using('default')
|
|
.filter(userprofile__flavor='chocolate').values_list('username', flat=True)
|
|
),
|
|
['alice']
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
User.objects.using('other')
|
|
.filter(userprofile__flavor='chocolate').values_list('username', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
|
|
self.assertEqual(
|
|
list(
|
|
User.objects.using('default')
|
|
.filter(userprofile__flavor='crunchy frog').values_list('username', flat=True)
|
|
),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(
|
|
User.objects.using('other')
|
|
.filter(userprofile__flavor='crunchy frog').values_list('username', flat=True)
|
|
),
|
|
['bob']
|
|
)
|
|
|
|
# Reget the objects to clear caches
|
|
alice_profile = UserProfile.objects.using('default').get(flavor='chocolate')
|
|
bob_profile = UserProfile.objects.using('other').get(flavor='crunchy frog')
|
|
|
|
# Retrieve related object by descriptor. Related objects should be database-bound
|
|
self.assertEqual(alice_profile.user.username, 'alice')
|
|
self.assertEqual(bob_profile.user.username, 'bob')
|
|
|
|
def test_o2o_cross_database_protection(self):
|
|
"Operations that involve sharing FK objects across databases raise an error"
|
|
# Create a user and profile on the default database
|
|
alice = User.objects.db_manager('default').create_user('alice', 'alice@example.com')
|
|
|
|
# Create a user and profile on the other database
|
|
bob = User.objects.db_manager('other').create_user('bob', 'bob@example.com')
|
|
|
|
# Set a one-to-one relation with an object from a different database
|
|
alice_profile = UserProfile.objects.using('default').create(user=alice, flavor='chocolate')
|
|
msg = (
|
|
'Cannot assign "%r": the current database router prevents this '
|
|
'relation.' % alice_profile
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
bob.userprofile = alice_profile
|
|
|
|
# BUT! if you assign a FK object when the base object hasn't
|
|
# been saved yet, you implicitly assign the database for the
|
|
# base object.
|
|
bob_profile = UserProfile.objects.using('other').create(user=bob, flavor='crunchy frog')
|
|
|
|
new_bob_profile = UserProfile(flavor="spring surprise")
|
|
|
|
# assigning a profile requires an explicit pk as the object isn't saved
|
|
charlie = User(pk=51, username='charlie', email='charlie@example.com')
|
|
charlie.set_unusable_password()
|
|
|
|
# initially, no db assigned
|
|
self.assertIsNone(new_bob_profile._state.db)
|
|
self.assertIsNone(charlie._state.db)
|
|
|
|
# old object comes from 'other', so the new object is set to use 'other'...
|
|
new_bob_profile.user = bob
|
|
charlie.userprofile = bob_profile
|
|
self.assertEqual(new_bob_profile._state.db, 'other')
|
|
self.assertEqual(charlie._state.db, 'other')
|
|
|
|
# ... but it isn't saved yet
|
|
self.assertEqual(list(User.objects.using('other').values_list('username', flat=True)), ['bob'])
|
|
self.assertEqual(list(UserProfile.objects.using('other').values_list('flavor', flat=True)), ['crunchy frog'])
|
|
|
|
# When saved (no using required), new objects goes to 'other'
|
|
charlie.save()
|
|
bob_profile.save()
|
|
new_bob_profile.save()
|
|
self.assertEqual(list(User.objects.using('default').values_list('username', flat=True)), ['alice'])
|
|
self.assertEqual(list(User.objects.using('other').values_list('username', flat=True)), ['bob', 'charlie'])
|
|
self.assertEqual(list(UserProfile.objects.using('default').values_list('flavor', flat=True)), ['chocolate'])
|
|
self.assertEqual(
|
|
list(UserProfile.objects.using('other').values_list('flavor', flat=True)),
|
|
['crunchy frog', 'spring surprise']
|
|
)
|
|
|
|
# This also works if you assign the O2O relation in the constructor
|
|
denise = User.objects.db_manager('other').create_user('denise', 'denise@example.com')
|
|
denise_profile = UserProfile(flavor="tofu", user=denise)
|
|
|
|
self.assertEqual(denise_profile._state.db, 'other')
|
|
# ... but it isn't saved yet
|
|
self.assertEqual(list(UserProfile.objects.using('default').values_list('flavor', flat=True)), ['chocolate'])
|
|
self.assertEqual(
|
|
list(UserProfile.objects.using('other').values_list('flavor', flat=True)),
|
|
['crunchy frog', 'spring surprise']
|
|
)
|
|
|
|
# When saved, the new profile goes to 'other'
|
|
denise_profile.save()
|
|
self.assertEqual(list(UserProfile.objects.using('default').values_list('flavor', flat=True)), ['chocolate'])
|
|
self.assertEqual(
|
|
list(UserProfile.objects.using('other').values_list('flavor', flat=True)),
|
|
['crunchy frog', 'spring surprise', 'tofu']
|
|
)
|
|
|
|
def test_generic_key_separation(self):
|
|
"Generic fields are constrained to a single database"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
review1 = Review.objects.create(source="Python Monthly", content_object=pro)
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
review2 = Review.objects.using('other').create(source="Python Weekly", content_object=dive)
|
|
|
|
review1 = Review.objects.using('default').get(source="Python Monthly")
|
|
self.assertEqual(review1.content_object.title, "Pro Django")
|
|
|
|
review2 = Review.objects.using('other').get(source="Python Weekly")
|
|
self.assertEqual(review2.content_object.title, "Dive into Python")
|
|
|
|
# Reget the objects to clear caches
|
|
dive = Book.objects.using('other').get(title="Dive into Python")
|
|
|
|
# Retrieve related object by descriptor. Related objects should be database-bound
|
|
self.assertEqual(list(dive.reviews.all().values_list('source', flat=True)), ['Python Weekly'])
|
|
|
|
def test_generic_key_reverse_operations(self):
|
|
"Generic reverse manipulations are all constrained to a single DB"
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
temp = Book.objects.using('other').create(title="Temp", published=datetime.date(2009, 5, 4))
|
|
review1 = Review.objects.using('other').create(source="Python Weekly", content_object=dive)
|
|
review2 = Review.objects.using('other').create(source="Python Monthly", content_object=temp)
|
|
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
['Python Weekly']
|
|
)
|
|
|
|
# Add a second review
|
|
dive.reviews.add(review2)
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
['Python Monthly', 'Python Weekly']
|
|
)
|
|
|
|
# Remove the second author
|
|
dive.reviews.remove(review1)
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
['Python Monthly']
|
|
)
|
|
|
|
# Clear all reviews
|
|
dive.reviews.clear()
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
[]
|
|
)
|
|
|
|
# Create an author through the generic interface
|
|
dive.reviews.create(source='Python Daily')
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
[]
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
['Python Daily']
|
|
)
|
|
|
|
def test_generic_key_cross_database_protection(self):
|
|
"Operations that involve sharing generic key objects across databases raise an error"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
review1 = Review.objects.create(source="Python Monthly", content_object=pro)
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
Review.objects.using('other').create(source="Python Weekly", content_object=dive)
|
|
|
|
# Set a foreign key with an object from a different database
|
|
msg = (
|
|
'Cannot assign "<ContentType: multiple_database | book>": the '
|
|
'current database router prevents this relation.'
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
review1.content_object = dive
|
|
|
|
# Add to a foreign key set with an object from a different database
|
|
msg = (
|
|
"<Review: Python Monthly> instance isn't saved. "
|
|
"Use bulk=False or save the object first."
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
with transaction.atomic(using='other'):
|
|
dive.reviews.add(review1)
|
|
|
|
# BUT! if you assign a FK object when the base object hasn't
|
|
# been saved yet, you implicitly assign the database for the
|
|
# base object.
|
|
review3 = Review(source="Python Daily")
|
|
# initially, no db assigned
|
|
self.assertIsNone(review3._state.db)
|
|
|
|
# Dive comes from 'other', so review3 is set to use 'other'...
|
|
review3.content_object = dive
|
|
self.assertEqual(review3._state.db, 'other')
|
|
# ... but it isn't saved yet
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=pro.pk).values_list('source', flat=True)),
|
|
['Python Monthly']
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
['Python Weekly']
|
|
)
|
|
|
|
# When saved, John goes to 'other'
|
|
review3.save()
|
|
self.assertEqual(
|
|
list(Review.objects.using('default').filter(object_id=pro.pk).values_list('source', flat=True)),
|
|
['Python Monthly']
|
|
)
|
|
self.assertEqual(
|
|
list(Review.objects.using('other').filter(object_id=dive.pk).values_list('source', flat=True)),
|
|
['Python Daily', 'Python Weekly']
|
|
)
|
|
|
|
def test_generic_key_deletion(self):
|
|
"Cascaded deletions of Generic Key relations issue queries on the right database"
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
Review.objects.using('other').create(source="Python Weekly", content_object=dive)
|
|
|
|
# Check the initial state
|
|
self.assertEqual(Book.objects.using('default').count(), 0)
|
|
self.assertEqual(Review.objects.using('default').count(), 0)
|
|
|
|
self.assertEqual(Book.objects.using('other').count(), 1)
|
|
self.assertEqual(Review.objects.using('other').count(), 1)
|
|
|
|
# Delete the Book object, which will cascade onto the pet
|
|
dive.delete(using='other')
|
|
|
|
self.assertEqual(Book.objects.using('default').count(), 0)
|
|
self.assertEqual(Review.objects.using('default').count(), 0)
|
|
|
|
# Both the pet and the person have been deleted from the right database
|
|
self.assertEqual(Book.objects.using('other').count(), 0)
|
|
self.assertEqual(Review.objects.using('other').count(), 0)
|
|
|
|
def test_ordering(self):
|
|
"get_next_by_XXX commands stick to a single database"
|
|
Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
learn = Book.objects.using('other').create(title="Learning Python", published=datetime.date(2008, 7, 16))
|
|
|
|
self.assertEqual(learn.get_next_by_published().title, "Dive into Python")
|
|
self.assertEqual(dive.get_previous_by_published().title, "Learning Python")
|
|
|
|
def test_raw(self):
|
|
"test the raw() method across databases"
|
|
dive = Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
val = Book.objects.db_manager("other").raw('SELECT id FROM multiple_database_book')
|
|
self.assertQuerysetEqual(val, [dive.pk], attrgetter("pk"))
|
|
|
|
val = Book.objects.raw('SELECT id FROM multiple_database_book').using('other')
|
|
self.assertQuerysetEqual(val, [dive.pk], attrgetter("pk"))
|
|
|
|
def test_select_related(self):
|
|
"Database assignment is retained if an object is retrieved with select_related()"
|
|
# Create a book and author on the other database
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
Book.objects.using('other').create(
|
|
title="Dive into Python",
|
|
published=datetime.date(2009, 5, 4),
|
|
editor=mark,
|
|
)
|
|
|
|
# Retrieve the Person using select_related()
|
|
book = Book.objects.using('other').select_related('editor').get(title="Dive into Python")
|
|
|
|
# The editor instance should have a db state
|
|
self.assertEqual(book.editor._state.db, 'other')
|
|
|
|
def test_subquery(self):
|
|
"""Make sure as_sql works with subqueries and primary/replica."""
|
|
sub = Person.objects.using('other').filter(name='fff')
|
|
qs = Book.objects.filter(editor__in=sub)
|
|
|
|
# When you call __str__ on the query object, it doesn't know about using
|
|
# so it falls back to the default. If the subquery explicitly uses a
|
|
# different database, an error should be raised.
|
|
msg = (
|
|
"Subqueries aren't allowed across different databases. Force the "
|
|
"inner query to be evaluated using `list(inner_query)`."
|
|
)
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
str(qs.query)
|
|
|
|
# Evaluating the query shouldn't work, either
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
for obj in qs:
|
|
pass
|
|
|
|
def test_related_manager(self):
|
|
"Related managers return managers, not querysets"
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
|
|
# extra_arg is removed by the BookManager's implementation of
|
|
# create(); but the BookManager's implementation won't get called
|
|
# unless edited returns a Manager, not a queryset
|
|
mark.book_set.create(title="Dive into Python", published=datetime.date(2009, 5, 4), extra_arg=True)
|
|
mark.book_set.get_or_create(title="Dive into Python", published=datetime.date(2009, 5, 4), extra_arg=True)
|
|
mark.edited.create(title="Dive into Water", published=datetime.date(2009, 5, 4), extra_arg=True)
|
|
mark.edited.get_or_create(title="Dive into Water", published=datetime.date(2009, 5, 4), extra_arg=True)
|
|
|
|
|
|
class ConnectionRouterTestCase(SimpleTestCase):
|
|
@override_settings(DATABASE_ROUTERS=[
|
|
'multiple_database.tests.TestRouter',
|
|
'multiple_database.tests.WriteRouter'])
|
|
def test_router_init_default(self):
|
|
connection_router = ConnectionRouter()
|
|
self.assertEqual([r.__class__.__name__ for r in connection_router.routers], ['TestRouter', 'WriteRouter'])
|
|
|
|
def test_router_init_arg(self):
|
|
connection_router = ConnectionRouter([
|
|
'multiple_database.tests.TestRouter',
|
|
'multiple_database.tests.WriteRouter'
|
|
])
|
|
self.assertEqual([r.__class__.__name__ for r in connection_router.routers], ['TestRouter', 'WriteRouter'])
|
|
|
|
# Init with instances instead of strings
|
|
connection_router = ConnectionRouter([TestRouter(), WriteRouter()])
|
|
self.assertEqual([r.__class__.__name__ for r in connection_router.routers], ['TestRouter', 'WriteRouter'])
|
|
|
|
|
|
# Make the 'other' database appear to be a replica of the 'default'
|
|
@override_settings(DATABASE_ROUTERS=[TestRouter()])
|
|
class RouterTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def test_db_selection(self):
|
|
"Querysets obey the router for db suggestions"
|
|
self.assertEqual(Book.objects.db, 'other')
|
|
self.assertEqual(Book.objects.all().db, 'other')
|
|
|
|
self.assertEqual(Book.objects.using('default').db, 'default')
|
|
|
|
self.assertEqual(Book.objects.db_manager('default').db, 'default')
|
|
self.assertEqual(Book.objects.db_manager('default').all().db, 'default')
|
|
|
|
def test_migrate_selection(self):
|
|
"Synchronization behavior is predictable"
|
|
|
|
self.assertTrue(router.allow_migrate_model('default', User))
|
|
self.assertTrue(router.allow_migrate_model('default', Book))
|
|
|
|
self.assertTrue(router.allow_migrate_model('other', User))
|
|
self.assertTrue(router.allow_migrate_model('other', Book))
|
|
|
|
with override_settings(DATABASE_ROUTERS=[TestRouter(), AuthRouter()]):
|
|
# Add the auth router to the chain. TestRouter is a universal
|
|
# synchronizer, so it should have no effect.
|
|
self.assertTrue(router.allow_migrate_model('default', User))
|
|
self.assertTrue(router.allow_migrate_model('default', Book))
|
|
|
|
self.assertTrue(router.allow_migrate_model('other', User))
|
|
self.assertTrue(router.allow_migrate_model('other', Book))
|
|
|
|
with override_settings(DATABASE_ROUTERS=[AuthRouter(), TestRouter()]):
|
|
# Now check what happens if the router order is reversed.
|
|
self.assertFalse(router.allow_migrate_model('default', User))
|
|
self.assertTrue(router.allow_migrate_model('default', Book))
|
|
|
|
self.assertTrue(router.allow_migrate_model('other', User))
|
|
self.assertTrue(router.allow_migrate_model('other', Book))
|
|
|
|
def test_partial_router(self):
|
|
"A router can choose to implement a subset of methods"
|
|
dive = Book.objects.using('other').create(title="Dive into Python",
|
|
published=datetime.date(2009, 5, 4))
|
|
|
|
# First check the baseline behavior.
|
|
|
|
self.assertEqual(router.db_for_read(User), 'other')
|
|
self.assertEqual(router.db_for_read(Book), 'other')
|
|
|
|
self.assertEqual(router.db_for_write(User), 'default')
|
|
self.assertEqual(router.db_for_write(Book), 'default')
|
|
|
|
self.assertTrue(router.allow_relation(dive, dive))
|
|
|
|
self.assertTrue(router.allow_migrate_model('default', User))
|
|
self.assertTrue(router.allow_migrate_model('default', Book))
|
|
|
|
with override_settings(DATABASE_ROUTERS=[WriteRouter(), AuthRouter(), TestRouter()]):
|
|
self.assertEqual(router.db_for_read(User), 'default')
|
|
self.assertEqual(router.db_for_read(Book), 'other')
|
|
|
|
self.assertEqual(router.db_for_write(User), 'writer')
|
|
self.assertEqual(router.db_for_write(Book), 'writer')
|
|
|
|
self.assertTrue(router.allow_relation(dive, dive))
|
|
|
|
self.assertFalse(router.allow_migrate_model('default', User))
|
|
self.assertTrue(router.allow_migrate_model('default', Book))
|
|
|
|
def test_database_routing(self):
|
|
marty = Person.objects.using('default').create(name="Marty Alchin")
|
|
pro = Book.objects.using('default').create(
|
|
title='Pro Django',
|
|
published=datetime.date(2008, 12, 16),
|
|
editor=marty,
|
|
)
|
|
pro.authors.set([marty])
|
|
|
|
# Create a book and author on the other database
|
|
Book.objects.using('other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
# An update query will be routed to the default database
|
|
Book.objects.filter(title='Pro Django').update(pages=200)
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
# By default, the get query will be directed to 'other'
|
|
Book.objects.get(title='Pro Django')
|
|
|
|
# But the same query issued explicitly at a database will work.
|
|
pro = Book.objects.using('default').get(title='Pro Django')
|
|
|
|
# The update worked.
|
|
self.assertEqual(pro.pages, 200)
|
|
|
|
# An update query with an explicit using clause will be routed
|
|
# to the requested database.
|
|
Book.objects.using('other').filter(title='Dive into Python').update(pages=300)
|
|
self.assertEqual(Book.objects.get(title='Dive into Python').pages, 300)
|
|
|
|
# Related object queries stick to the same database
|
|
# as the original object, regardless of the router
|
|
self.assertEqual(list(pro.authors.values_list('name', flat=True)), ['Marty Alchin'])
|
|
self.assertEqual(pro.editor.name, 'Marty Alchin')
|
|
|
|
# get_or_create is a special case. The get needs to be targeted at
|
|
# the write database in order to avoid potential transaction
|
|
# consistency problems
|
|
book, created = Book.objects.get_or_create(title="Pro Django")
|
|
self.assertFalse(created)
|
|
|
|
book, created = Book.objects.get_or_create(title="Dive Into Python",
|
|
defaults={'published': datetime.date(2009, 5, 4)})
|
|
self.assertTrue(created)
|
|
|
|
# Check the head count of objects
|
|
self.assertEqual(Book.objects.using('default').count(), 2)
|
|
self.assertEqual(Book.objects.using('other').count(), 1)
|
|
# If a database isn't specified, the read database is used
|
|
self.assertEqual(Book.objects.count(), 1)
|
|
|
|
# A delete query will also be routed to the default database
|
|
Book.objects.filter(pages__gt=150).delete()
|
|
|
|
# The default database has lost the book.
|
|
self.assertEqual(Book.objects.using('default').count(), 1)
|
|
self.assertEqual(Book.objects.using('other').count(), 1)
|
|
|
|
def test_invalid_set_foreign_key_assignment(self):
|
|
marty = Person.objects.using('default').create(name="Marty Alchin")
|
|
dive = Book.objects.using('other').create(
|
|
title="Dive into Python",
|
|
published=datetime.date(2009, 5, 4),
|
|
)
|
|
# Set a foreign key set with an object from a different database
|
|
msg = "<Book: Dive into Python> instance isn't saved. Use bulk=False or save the object first."
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
marty.edited.set([dive])
|
|
|
|
def test_foreign_key_cross_database_protection(self):
|
|
"Foreign keys can cross databases if they two databases have a common source"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.using('default').create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
marty = Person.objects.using('default').create(name="Marty Alchin")
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using('other').create(title="Dive into Python",
|
|
published=datetime.date(2009, 5, 4))
|
|
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
|
|
# Set a foreign key with an object from a different database
|
|
dive.editor = marty
|
|
|
|
# Database assignments of original objects haven't changed...
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# ... but they will when the affected object is saved.
|
|
dive.save()
|
|
self.assertEqual(dive._state.db, 'default')
|
|
|
|
# ...and the source database now has a copy of any object saved
|
|
Book.objects.using('default').get(title='Dive into Python').delete()
|
|
|
|
# This isn't a real primary/replica database, so restore the original from other
|
|
dive = Book.objects.using('other').get(title='Dive into Python')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
|
|
# Set a foreign key set with an object from a different database
|
|
marty.edited.set([pro, dive], bulk=False)
|
|
|
|
# Assignment implies a save, so database assignments of original objects have changed...
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'default')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# ...and the source database now has a copy of any object saved
|
|
Book.objects.using('default').get(title='Dive into Python').delete()
|
|
|
|
# This isn't a real primary/replica database, so restore the original from other
|
|
dive = Book.objects.using('other').get(title='Dive into Python')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
|
|
# Add to a foreign key set with an object from a different database
|
|
marty.edited.add(dive, bulk=False)
|
|
|
|
# Add implies a save, so database assignments of original objects have changed...
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'default')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# ...and the source database now has a copy of any object saved
|
|
Book.objects.using('default').get(title='Dive into Python').delete()
|
|
|
|
# This isn't a real primary/replica database, so restore the original from other
|
|
dive = Book.objects.using('other').get(title='Dive into Python')
|
|
|
|
# If you assign a FK object when the base object hasn't
|
|
# been saved yet, you implicitly assign the database for the
|
|
# base object.
|
|
chris = Person(name="Chris Mills")
|
|
html5 = Book(title="Dive into HTML5", published=datetime.date(2010, 3, 15))
|
|
# initially, no db assigned
|
|
self.assertIsNone(chris._state.db)
|
|
self.assertIsNone(html5._state.db)
|
|
|
|
# old object comes from 'other', so the new object is set to use the
|
|
# source of 'other'...
|
|
self.assertEqual(dive._state.db, 'other')
|
|
chris.save()
|
|
dive.editor = chris
|
|
html5.editor = mark
|
|
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
self.assertEqual(chris._state.db, 'default')
|
|
self.assertEqual(html5._state.db, 'default')
|
|
|
|
# This also works if you assign the FK in the constructor
|
|
water = Book(title="Dive into Water", published=datetime.date(2001, 1, 1), editor=mark)
|
|
self.assertEqual(water._state.db, 'default')
|
|
|
|
# For the remainder of this test, create a copy of 'mark' in the
|
|
# 'default' database to prevent integrity errors on backends that
|
|
# don't defer constraints checks until the end of the transaction
|
|
mark.save(using='default')
|
|
|
|
# This moved 'mark' in the 'default' database, move it back in 'other'
|
|
mark.save(using='other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# If you create an object through a FK relation, it will be
|
|
# written to the write database, even if the original object
|
|
# was on the read database
|
|
cheesecake = mark.edited.create(title='Dive into Cheesecake', published=datetime.date(2010, 3, 15))
|
|
self.assertEqual(cheesecake._state.db, 'default')
|
|
|
|
# Same goes for get_or_create, regardless of whether getting or creating
|
|
cheesecake, created = mark.edited.get_or_create(
|
|
title='Dive into Cheesecake',
|
|
published=datetime.date(2010, 3, 15),
|
|
)
|
|
self.assertEqual(cheesecake._state.db, 'default')
|
|
|
|
puddles, created = mark.edited.get_or_create(title='Dive into Puddles', published=datetime.date(2010, 3, 15))
|
|
self.assertEqual(puddles._state.db, 'default')
|
|
|
|
def test_m2m_cross_database_protection(self):
|
|
"M2M relations can cross databases if the database share a source"
|
|
# Create books and authors on the inverse to the usual database
|
|
pro = Book.objects.using('other').create(pk=1, title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
marty = Person.objects.using('other').create(pk=1, name="Marty Alchin")
|
|
|
|
dive = Book.objects.using('default').create(pk=2, title="Dive into Python",
|
|
published=datetime.date(2009, 5, 4))
|
|
|
|
mark = Person.objects.using('default').create(pk=2, name="Mark Pilgrim")
|
|
|
|
# Now save back onto the usual database.
|
|
# This simulates primary/replica - the objects exist on both database,
|
|
# but the _state.db is as it is for all other tests.
|
|
pro.save(using='default')
|
|
marty.save(using='default')
|
|
dive.save(using='other')
|
|
mark.save(using='other')
|
|
|
|
# We have 2 of both types of object on both databases
|
|
self.assertEqual(Book.objects.using('default').count(), 2)
|
|
self.assertEqual(Book.objects.using('other').count(), 2)
|
|
self.assertEqual(Person.objects.using('default').count(), 2)
|
|
self.assertEqual(Person.objects.using('other').count(), 2)
|
|
|
|
# Set a m2m set with an object from a different database
|
|
marty.book_set.set([pro, dive])
|
|
|
|
# Database assignments don't change
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# All m2m relations should be saved on the default database
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 2)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
# Reset relations
|
|
Book.authors.through.objects.using('default').delete()
|
|
|
|
# Add to an m2m with an object from a different database
|
|
marty.book_set.add(dive)
|
|
|
|
# Database assignments don't change
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# All m2m relations should be saved on the default database
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 1)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
# Reset relations
|
|
Book.authors.through.objects.using('default').delete()
|
|
|
|
# Set a reverse m2m with an object from a different database
|
|
dive.authors.set([mark, marty])
|
|
|
|
# Database assignments don't change
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# All m2m relations should be saved on the default database
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 2)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
# Reset relations
|
|
Book.authors.through.objects.using('default').delete()
|
|
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 0)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
# Add to a reverse m2m with an object from a different database
|
|
dive.authors.add(marty)
|
|
|
|
# Database assignments don't change
|
|
self.assertEqual(marty._state.db, 'default')
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(mark._state.db, 'other')
|
|
|
|
# All m2m relations should be saved on the default database
|
|
self.assertEqual(Book.authors.through.objects.using('default').count(), 1)
|
|
self.assertEqual(Book.authors.through.objects.using('other').count(), 0)
|
|
|
|
# If you create an object through a M2M relation, it will be
|
|
# written to the write database, even if the original object
|
|
# was on the read database
|
|
alice = dive.authors.create(name='Alice')
|
|
self.assertEqual(alice._state.db, 'default')
|
|
|
|
# Same goes for get_or_create, regardless of whether getting or creating
|
|
alice, created = dive.authors.get_or_create(name='Alice')
|
|
self.assertEqual(alice._state.db, 'default')
|
|
|
|
bob, created = dive.authors.get_or_create(name='Bob')
|
|
self.assertEqual(bob._state.db, 'default')
|
|
|
|
def test_o2o_cross_database_protection(self):
|
|
"Operations that involve sharing FK objects across databases raise an error"
|
|
# Create a user and profile on the default database
|
|
alice = User.objects.db_manager('default').create_user('alice', 'alice@example.com')
|
|
|
|
# Create a user and profile on the other database
|
|
bob = User.objects.db_manager('other').create_user('bob', 'bob@example.com')
|
|
|
|
# Set a one-to-one relation with an object from a different database
|
|
alice_profile = UserProfile.objects.create(user=alice, flavor='chocolate')
|
|
bob.userprofile = alice_profile
|
|
|
|
# Database assignments of original objects haven't changed...
|
|
self.assertEqual(alice._state.db, 'default')
|
|
self.assertEqual(alice_profile._state.db, 'default')
|
|
self.assertEqual(bob._state.db, 'other')
|
|
|
|
# ... but they will when the affected object is saved.
|
|
bob.save()
|
|
self.assertEqual(bob._state.db, 'default')
|
|
|
|
def test_generic_key_cross_database_protection(self):
|
|
"Generic Key operations can span databases if they share a source"
|
|
# Create a book and author on the default database
|
|
pro = Book.objects.using(
|
|
'default').create(title="Pro Django", published=datetime.date(2008, 12, 16))
|
|
|
|
review1 = Review.objects.using(
|
|
'default').create(source="Python Monthly", content_object=pro)
|
|
|
|
# Create a book and author on the other database
|
|
dive = Book.objects.using(
|
|
'other').create(title="Dive into Python", published=datetime.date(2009, 5, 4))
|
|
|
|
review2 = Review.objects.using(
|
|
'other').create(source="Python Weekly", content_object=dive)
|
|
|
|
# Set a generic foreign key with an object from a different database
|
|
review1.content_object = dive
|
|
|
|
# Database assignments of original objects haven't changed...
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(review1._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(review2._state.db, 'other')
|
|
|
|
# ... but they will when the affected object is saved.
|
|
dive.save()
|
|
self.assertEqual(review1._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'default')
|
|
|
|
# ...and the source database now has a copy of any object saved
|
|
Book.objects.using('default').get(title='Dive into Python').delete()
|
|
|
|
# This isn't a real primary/replica database, so restore the original from other
|
|
dive = Book.objects.using('other').get(title='Dive into Python')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
|
|
# Add to a generic foreign key set with an object from a different database
|
|
dive.reviews.add(review1)
|
|
|
|
# Database assignments of original objects haven't changed...
|
|
self.assertEqual(pro._state.db, 'default')
|
|
self.assertEqual(review1._state.db, 'default')
|
|
self.assertEqual(dive._state.db, 'other')
|
|
self.assertEqual(review2._state.db, 'other')
|
|
|
|
# ... but they will when the affected object is saved.
|
|
dive.save()
|
|
self.assertEqual(dive._state.db, 'default')
|
|
|
|
# ...and the source database now has a copy of any object saved
|
|
Book.objects.using('default').get(title='Dive into Python').delete()
|
|
|
|
# BUT! if you assign a FK object when the base object hasn't
|
|
# been saved yet, you implicitly assign the database for the
|
|
# base object.
|
|
review3 = Review(source="Python Daily")
|
|
# initially, no db assigned
|
|
self.assertIsNone(review3._state.db)
|
|
|
|
# Dive comes from 'other', so review3 is set to use the source of 'other'...
|
|
review3.content_object = dive
|
|
self.assertEqual(review3._state.db, 'default')
|
|
|
|
# If you create an object through a M2M relation, it will be
|
|
# written to the write database, even if the original object
|
|
# was on the read database
|
|
dive = Book.objects.using('other').get(title='Dive into Python')
|
|
nyt = dive.reviews.create(source="New York Times", content_object=dive)
|
|
self.assertEqual(nyt._state.db, 'default')
|
|
|
|
def test_m2m_managers(self):
|
|
"M2M relations are represented by managers, and can be controlled like managers"
|
|
pro = Book.objects.using('other').create(pk=1, title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
marty = Person.objects.using('other').create(pk=1, name="Marty Alchin")
|
|
|
|
self.assertEqual(pro.authors.db, 'other')
|
|
self.assertEqual(pro.authors.db_manager('default').db, 'default')
|
|
self.assertEqual(pro.authors.db_manager('default').all().db, 'default')
|
|
|
|
self.assertEqual(marty.book_set.db, 'other')
|
|
self.assertEqual(marty.book_set.db_manager('default').db, 'default')
|
|
self.assertEqual(marty.book_set.db_manager('default').all().db, 'default')
|
|
|
|
def test_foreign_key_managers(self):
|
|
"FK reverse relations are represented by managers, and can be controlled like managers"
|
|
marty = Person.objects.using('other').create(pk=1, name="Marty Alchin")
|
|
Book.objects.using('other').create(
|
|
pk=1,
|
|
title='Pro Django',
|
|
published=datetime.date(2008, 12, 16),
|
|
editor=marty,
|
|
)
|
|
self.assertEqual(marty.edited.db, 'other')
|
|
self.assertEqual(marty.edited.db_manager('default').db, 'default')
|
|
self.assertEqual(marty.edited.db_manager('default').all().db, 'default')
|
|
|
|
def test_generic_key_managers(self):
|
|
"Generic key relations are represented by managers, and can be controlled like managers"
|
|
pro = Book.objects.using('other').create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
Review.objects.using('other').create(source='Python Monthly', content_object=pro)
|
|
|
|
self.assertEqual(pro.reviews.db, 'other')
|
|
self.assertEqual(pro.reviews.db_manager('default').db, 'default')
|
|
self.assertEqual(pro.reviews.db_manager('default').all().db, 'default')
|
|
|
|
def test_subquery(self):
|
|
"""Make sure as_sql works with subqueries and primary/replica."""
|
|
# Create a book and author on the other database
|
|
|
|
mark = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
Book.objects.using('other').create(
|
|
title='Dive into Python',
|
|
published=datetime.date(2009, 5, 4),
|
|
editor=mark,
|
|
)
|
|
|
|
sub = Person.objects.filter(name='Mark Pilgrim')
|
|
qs = Book.objects.filter(editor__in=sub)
|
|
|
|
# When you call __str__ on the query object, it doesn't know about using
|
|
# so it falls back to the default. Don't let routing instructions
|
|
# force the subquery to an incompatible database.
|
|
str(qs.query)
|
|
|
|
# If you evaluate the query, it should work, running on 'other'
|
|
self.assertEqual(list(qs.values_list('title', flat=True)), ['Dive into Python'])
|
|
|
|
def test_deferred_models(self):
|
|
mark_def = Person.objects.using('default').create(name="Mark Pilgrim")
|
|
mark_other = Person.objects.using('other').create(name="Mark Pilgrim")
|
|
orig_b = Book.objects.using('other').create(
|
|
title='Dive into Python',
|
|
published=datetime.date(2009, 5, 4),
|
|
editor=mark_other,
|
|
)
|
|
b = Book.objects.using('other').only('title').get(pk=orig_b.pk)
|
|
self.assertEqual(b.published, datetime.date(2009, 5, 4))
|
|
b = Book.objects.using('other').only('title').get(pk=orig_b.pk)
|
|
b.editor = mark_def
|
|
b.save(using='default')
|
|
self.assertEqual(Book.objects.using('default').get(pk=b.pk).published,
|
|
datetime.date(2009, 5, 4))
|
|
|
|
|
|
@override_settings(DATABASE_ROUTERS=[AuthRouter()])
|
|
class AuthTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def test_auth_manager(self):
|
|
"The methods on the auth manager obey database hints"
|
|
# Create one user using default allocation policy
|
|
User.objects.create_user('alice', 'alice@example.com')
|
|
|
|
# Create another user, explicitly specifying the database
|
|
User.objects.db_manager('default').create_user('bob', 'bob@example.com')
|
|
|
|
# The second user only exists on the other database
|
|
alice = User.objects.using('other').get(username='alice')
|
|
|
|
self.assertEqual(alice.username, 'alice')
|
|
self.assertEqual(alice._state.db, 'other')
|
|
|
|
with self.assertRaises(User.DoesNotExist):
|
|
User.objects.using('default').get(username='alice')
|
|
|
|
# The second user only exists on the default database
|
|
bob = User.objects.using('default').get(username='bob')
|
|
|
|
self.assertEqual(bob.username, 'bob')
|
|
self.assertEqual(bob._state.db, 'default')
|
|
|
|
with self.assertRaises(User.DoesNotExist):
|
|
User.objects.using('other').get(username='bob')
|
|
|
|
# That is... there is one user on each database
|
|
self.assertEqual(User.objects.using('default').count(), 1)
|
|
self.assertEqual(User.objects.using('other').count(), 1)
|
|
|
|
def test_dumpdata(self):
|
|
"dumpdata honors allow_migrate restrictions on the router"
|
|
User.objects.create_user('alice', 'alice@example.com')
|
|
User.objects.db_manager('default').create_user('bob', 'bob@example.com')
|
|
|
|
# dumping the default database doesn't try to include auth because
|
|
# allow_migrate prohibits auth on default
|
|
new_io = StringIO()
|
|
management.call_command('dumpdata', 'auth', format='json', database='default', stdout=new_io)
|
|
command_output = new_io.getvalue().strip()
|
|
self.assertEqual(command_output, '[]')
|
|
|
|
# dumping the other database does include auth
|
|
new_io = StringIO()
|
|
management.call_command('dumpdata', 'auth', format='json', database='other', stdout=new_io)
|
|
command_output = new_io.getvalue().strip()
|
|
self.assertIn('"email": "alice@example.com"', command_output)
|
|
|
|
|
|
class AntiPetRouter:
|
|
# A router that only expresses an opinion on migrate,
|
|
# passing pets to the 'other' database
|
|
|
|
def allow_migrate(self, db, app_label, model_name=None, **hints):
|
|
if db == 'other':
|
|
return model_name == 'pet'
|
|
else:
|
|
return model_name != 'pet'
|
|
|
|
|
|
class FixtureTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
fixtures = ['multidb-common', 'multidb']
|
|
|
|
@override_settings(DATABASE_ROUTERS=[AntiPetRouter()])
|
|
def test_fixture_loading(self):
|
|
"Multi-db fixtures are loaded correctly"
|
|
# "Pro Django" exists on the default database, but not on other database
|
|
Book.objects.get(title="Pro Django")
|
|
Book.objects.using('default').get(title="Pro Django")
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('other').get(title="Pro Django")
|
|
|
|
# "Dive into Python" exists on the default database, but not on other database
|
|
Book.objects.using('other').get(title="Dive into Python")
|
|
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.get(title="Dive into Python")
|
|
with self.assertRaises(Book.DoesNotExist):
|
|
Book.objects.using('default').get(title="Dive into Python")
|
|
|
|
# "Definitive Guide" exists on the both databases
|
|
Book.objects.get(title="The Definitive Guide to Django")
|
|
Book.objects.using('default').get(title="The Definitive Guide to Django")
|
|
Book.objects.using('other').get(title="The Definitive Guide to Django")
|
|
|
|
@override_settings(DATABASE_ROUTERS=[AntiPetRouter()])
|
|
def test_pseudo_empty_fixtures(self):
|
|
"""
|
|
A fixture can contain entries, but lead to nothing in the database;
|
|
this shouldn't raise an error (#14068).
|
|
"""
|
|
new_io = StringIO()
|
|
management.call_command('loaddata', 'pets', stdout=new_io, stderr=new_io)
|
|
command_output = new_io.getvalue().strip()
|
|
# No objects will actually be loaded
|
|
self.assertEqual(command_output, "Installed 0 object(s) (of 2) from 1 fixture(s)")
|
|
|
|
|
|
class PickleQuerySetTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def test_pickling(self):
|
|
for db in connections:
|
|
Book.objects.using(db).create(title='Dive into Python', published=datetime.date(2009, 5, 4))
|
|
qs = Book.objects.all()
|
|
self.assertEqual(qs.db, pickle.loads(pickle.dumps(qs)).db)
|
|
|
|
|
|
class DatabaseReceiver:
|
|
"""
|
|
Used in the tests for the database argument in signals (#13552)
|
|
"""
|
|
def __call__(self, signal, sender, **kwargs):
|
|
self._database = kwargs['using']
|
|
|
|
|
|
class WriteToOtherRouter:
|
|
"""
|
|
A router that sends all writes to the other database.
|
|
"""
|
|
def db_for_write(self, model, **hints):
|
|
return "other"
|
|
|
|
|
|
class SignalTests(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def override_router(self):
|
|
return override_settings(DATABASE_ROUTERS=[WriteToOtherRouter()])
|
|
|
|
def test_database_arg_save_and_delete(self):
|
|
"""
|
|
The pre/post_save signal contains the correct database.
|
|
"""
|
|
# Make some signal receivers
|
|
pre_save_receiver = DatabaseReceiver()
|
|
post_save_receiver = DatabaseReceiver()
|
|
pre_delete_receiver = DatabaseReceiver()
|
|
post_delete_receiver = DatabaseReceiver()
|
|
# Make model and connect receivers
|
|
signals.pre_save.connect(sender=Person, receiver=pre_save_receiver)
|
|
signals.post_save.connect(sender=Person, receiver=post_save_receiver)
|
|
signals.pre_delete.connect(sender=Person, receiver=pre_delete_receiver)
|
|
signals.post_delete.connect(sender=Person, receiver=post_delete_receiver)
|
|
p = Person.objects.create(name='Darth Vader')
|
|
# Save and test receivers got calls
|
|
p.save()
|
|
self.assertEqual(pre_save_receiver._database, DEFAULT_DB_ALIAS)
|
|
self.assertEqual(post_save_receiver._database, DEFAULT_DB_ALIAS)
|
|
# Delete, and test
|
|
p.delete()
|
|
self.assertEqual(pre_delete_receiver._database, DEFAULT_DB_ALIAS)
|
|
self.assertEqual(post_delete_receiver._database, DEFAULT_DB_ALIAS)
|
|
# Save again to a different database
|
|
p.save(using="other")
|
|
self.assertEqual(pre_save_receiver._database, "other")
|
|
self.assertEqual(post_save_receiver._database, "other")
|
|
# Delete, and test
|
|
p.delete(using="other")
|
|
self.assertEqual(pre_delete_receiver._database, "other")
|
|
self.assertEqual(post_delete_receiver._database, "other")
|
|
|
|
signals.pre_save.disconnect(sender=Person, receiver=pre_save_receiver)
|
|
signals.post_save.disconnect(sender=Person, receiver=post_save_receiver)
|
|
signals.pre_delete.disconnect(sender=Person, receiver=pre_delete_receiver)
|
|
signals.post_delete.disconnect(sender=Person, receiver=post_delete_receiver)
|
|
|
|
def test_database_arg_m2m(self):
|
|
"""
|
|
The m2m_changed signal has a correct database arg.
|
|
"""
|
|
# Make a receiver
|
|
receiver = DatabaseReceiver()
|
|
# Connect it
|
|
signals.m2m_changed.connect(receiver=receiver)
|
|
|
|
# Create the models that will be used for the tests
|
|
b = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
p = Person.objects.create(name="Marty Alchin")
|
|
|
|
# Create a copy of the models on the 'other' database to prevent
|
|
# integrity errors on backends that don't defer constraints checks
|
|
Book.objects.using('other').create(pk=b.pk, title=b.title, published=b.published)
|
|
Person.objects.using('other').create(pk=p.pk, name=p.name)
|
|
|
|
# Test addition
|
|
b.authors.add(p)
|
|
self.assertEqual(receiver._database, DEFAULT_DB_ALIAS)
|
|
with self.override_router():
|
|
b.authors.add(p)
|
|
self.assertEqual(receiver._database, "other")
|
|
|
|
# Test removal
|
|
b.authors.remove(p)
|
|
self.assertEqual(receiver._database, DEFAULT_DB_ALIAS)
|
|
with self.override_router():
|
|
b.authors.remove(p)
|
|
self.assertEqual(receiver._database, "other")
|
|
|
|
# Test addition in reverse
|
|
p.book_set.add(b)
|
|
self.assertEqual(receiver._database, DEFAULT_DB_ALIAS)
|
|
with self.override_router():
|
|
p.book_set.add(b)
|
|
self.assertEqual(receiver._database, "other")
|
|
|
|
# Test clearing
|
|
b.authors.clear()
|
|
self.assertEqual(receiver._database, DEFAULT_DB_ALIAS)
|
|
with self.override_router():
|
|
b.authors.clear()
|
|
self.assertEqual(receiver._database, "other")
|
|
|
|
|
|
class AttributeErrorRouter:
|
|
"A router to test the exception handling of ConnectionRouter"
|
|
def db_for_read(self, model, **hints):
|
|
raise AttributeError
|
|
|
|
def db_for_write(self, model, **hints):
|
|
raise AttributeError
|
|
|
|
|
|
class RouterAttributeErrorTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def override_router(self):
|
|
return override_settings(DATABASE_ROUTERS=[AttributeErrorRouter()])
|
|
|
|
def test_attribute_error_read(self):
|
|
"The AttributeError from AttributeErrorRouter bubbles up"
|
|
b = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
with self.override_router():
|
|
with self.assertRaises(AttributeError):
|
|
Book.objects.get(pk=b.pk)
|
|
|
|
def test_attribute_error_save(self):
|
|
"The AttributeError from AttributeErrorRouter bubbles up"
|
|
dive = Book()
|
|
dive.title = "Dive into Python"
|
|
dive.published = datetime.date(2009, 5, 4)
|
|
with self.override_router():
|
|
with self.assertRaises(AttributeError):
|
|
dive.save()
|
|
|
|
def test_attribute_error_delete(self):
|
|
"The AttributeError from AttributeErrorRouter bubbles up"
|
|
b = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
p = Person.objects.create(name="Marty Alchin")
|
|
b.authors.set([p])
|
|
b.editor = p
|
|
with self.override_router():
|
|
with self.assertRaises(AttributeError):
|
|
b.delete()
|
|
|
|
def test_attribute_error_m2m(self):
|
|
"The AttributeError from AttributeErrorRouter bubbles up"
|
|
b = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
p = Person.objects.create(name="Marty Alchin")
|
|
with self.override_router():
|
|
with self.assertRaises(AttributeError):
|
|
b.authors.set([p])
|
|
|
|
|
|
class ModelMetaRouter:
|
|
"A router to ensure model arguments are real model classes"
|
|
def db_for_write(self, model, **hints):
|
|
if not hasattr(model, '_meta'):
|
|
raise ValueError
|
|
|
|
|
|
@override_settings(DATABASE_ROUTERS=[ModelMetaRouter()])
|
|
class RouterModelArgumentTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
def test_m2m_collection(self):
|
|
b = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
|
|
p = Person.objects.create(name="Marty Alchin")
|
|
# test add
|
|
b.authors.add(p)
|
|
# test remove
|
|
b.authors.remove(p)
|
|
# test clear
|
|
b.authors.clear()
|
|
# test setattr
|
|
b.authors.set([p])
|
|
# test M2M collection
|
|
b.delete()
|
|
|
|
def test_foreignkey_collection(self):
|
|
person = Person.objects.create(name='Bob')
|
|
Pet.objects.create(owner=person, name='Wart')
|
|
# test related FK collection
|
|
person.delete()
|
|
|
|
|
|
class SyncOnlyDefaultDatabaseRouter:
|
|
def allow_migrate(self, db, app_label, **hints):
|
|
return db == DEFAULT_DB_ALIAS
|
|
|
|
|
|
class MigrateTestCase(TestCase):
|
|
|
|
# Limit memory usage when calling 'migrate'.
|
|
available_apps = [
|
|
'multiple_database',
|
|
'django.contrib.auth',
|
|
'django.contrib.contenttypes'
|
|
]
|
|
databases = {'default', 'other'}
|
|
|
|
def test_migrate_to_other_database(self):
|
|
"""Regression test for #16039: migrate with --database option."""
|
|
cts = ContentType.objects.using('other').filter(app_label='multiple_database')
|
|
|
|
count = cts.count()
|
|
self.assertGreater(count, 0)
|
|
|
|
cts.delete()
|
|
management.call_command('migrate', verbosity=0, interactive=False, database='other')
|
|
self.assertEqual(cts.count(), count)
|
|
|
|
def test_migrate_to_other_database_with_router(self):
|
|
"""Regression test for #16039: migrate with --database option."""
|
|
cts = ContentType.objects.using('other').filter(app_label='multiple_database')
|
|
|
|
cts.delete()
|
|
with override_settings(DATABASE_ROUTERS=[SyncOnlyDefaultDatabaseRouter()]):
|
|
management.call_command('migrate', verbosity=0, interactive=False, database='other')
|
|
|
|
self.assertEqual(cts.count(), 0)
|
|
|
|
|
|
class RouterUsed(Exception):
|
|
WRITE = 'write'
|
|
|
|
def __init__(self, mode, model, hints):
|
|
self.mode = mode
|
|
self.model = model
|
|
self.hints = hints
|
|
|
|
|
|
class RouteForWriteTestCase(TestCase):
|
|
databases = {'default', 'other'}
|
|
|
|
class WriteCheckRouter:
|
|
def db_for_write(self, model, **hints):
|
|
raise RouterUsed(mode=RouterUsed.WRITE, model=model, hints=hints)
|
|
|
|
def override_router(self):
|
|
return override_settings(DATABASE_ROUTERS=[RouteForWriteTestCase.WriteCheckRouter()])
|
|
|
|
def test_fk_delete(self):
|
|
owner = Person.objects.create(name='Someone')
|
|
pet = Pet.objects.create(name='fido', owner=owner)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
pet.owner.delete()
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Person)
|
|
self.assertEqual(e.hints, {'instance': owner})
|
|
|
|
def test_reverse_fk_delete(self):
|
|
owner = Person.objects.create(name='Someone')
|
|
to_del_qs = owner.pet_set.all()
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
to_del_qs.delete()
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Pet)
|
|
self.assertEqual(e.hints, {'instance': owner})
|
|
|
|
def test_reverse_fk_get_or_create(self):
|
|
owner = Person.objects.create(name='Someone')
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
owner.pet_set.get_or_create(name='fido')
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Pet)
|
|
self.assertEqual(e.hints, {'instance': owner})
|
|
|
|
def test_reverse_fk_update(self):
|
|
owner = Person.objects.create(name='Someone')
|
|
Pet.objects.create(name='fido', owner=owner)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
owner.pet_set.update(name='max')
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Pet)
|
|
self.assertEqual(e.hints, {'instance': owner})
|
|
|
|
def test_m2m_add(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
book.authors.add(auth)
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book.authors.through)
|
|
self.assertEqual(e.hints, {'instance': book})
|
|
|
|
def test_m2m_clear(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
book.authors.clear()
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book.authors.through)
|
|
self.assertEqual(e.hints, {'instance': book})
|
|
|
|
def test_m2m_delete(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
book.authors.all().delete()
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Person)
|
|
self.assertEqual(e.hints, {'instance': book})
|
|
|
|
def test_m2m_get_or_create(self):
|
|
Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
book.authors.get_or_create(name='Someone else')
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book)
|
|
self.assertEqual(e.hints, {'instance': book})
|
|
|
|
def test_m2m_remove(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
book.authors.remove(auth)
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book.authors.through)
|
|
self.assertEqual(e.hints, {'instance': book})
|
|
|
|
def test_m2m_update(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
book.authors.all().update(name='Different')
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Person)
|
|
self.assertEqual(e.hints, {'instance': book})
|
|
|
|
def test_reverse_m2m_add(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
auth.book_set.add(book)
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book.authors.through)
|
|
self.assertEqual(e.hints, {'instance': auth})
|
|
|
|
def test_reverse_m2m_clear(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
auth.book_set.clear()
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book.authors.through)
|
|
self.assertEqual(e.hints, {'instance': auth})
|
|
|
|
def test_reverse_m2m_delete(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
auth.book_set.all().delete()
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book)
|
|
self.assertEqual(e.hints, {'instance': auth})
|
|
|
|
def test_reverse_m2m_get_or_create(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
auth.book_set.get_or_create(title="New Book", published=datetime.datetime.now())
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Person)
|
|
self.assertEqual(e.hints, {'instance': auth})
|
|
|
|
def test_reverse_m2m_remove(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
auth.book_set.remove(book)
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book.authors.through)
|
|
self.assertEqual(e.hints, {'instance': auth})
|
|
|
|
def test_reverse_m2m_update(self):
|
|
auth = Person.objects.create(name='Someone')
|
|
book = Book.objects.create(title="Pro Django",
|
|
published=datetime.date(2008, 12, 16))
|
|
book.authors.add(auth)
|
|
with self.assertRaises(RouterUsed) as cm:
|
|
with self.override_router():
|
|
auth.book_set.all().update(title='Different')
|
|
e = cm.exception
|
|
self.assertEqual(e.mode, RouterUsed.WRITE)
|
|
self.assertEqual(e.model, Book)
|
|
self.assertEqual(e.hints, {'instance': auth})
|
|
|
|
|
|
class NoRelationRouter:
|
|
"""Disallow all relations."""
|
|
def allow_relation(self, obj1, obj2, **hints):
|
|
return False
|
|
|
|
|
|
@override_settings(DATABASE_ROUTERS=[NoRelationRouter()])
|
|
class RelationAssignmentTests(SimpleTestCase):
|
|
"""allow_relation() is called with unsaved model instances."""
|
|
databases = {'default', 'other'}
|
|
router_prevents_msg = 'the current database router prevents this relation'
|
|
|
|
def test_foreign_key_relation(self):
|
|
person = Person(name='Someone')
|
|
pet = Pet()
|
|
with self.assertRaisesMessage(ValueError, self.router_prevents_msg):
|
|
pet.owner = person
|
|
|
|
def test_reverse_one_to_one_relation(self):
|
|
user = User(username='Someone', password='fake_hash')
|
|
profile = UserProfile()
|
|
with self.assertRaisesMessage(ValueError, self.router_prevents_msg):
|
|
user.userprofile = profile
|