mirror of
https://github.com/django/django.git
synced 2024-12-27 11:35:53 +00:00
99321e30ce
Deferred models now automatically update only the fields which are loaded from the db (with .only() or .defer()). In addition, any field set manually after the load is updated on save.
256 lines
9.4 KiB
Python
256 lines
9.4 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.test import TestCase
|
|
from django.db.models.signals import pre_save, post_save
|
|
from .models import Person, Employee, ProxyEmployee, Profile, Account
|
|
|
|
|
|
class UpdateOnlyFieldsTests(TestCase):
|
|
def test_update_fields_basic(self):
|
|
s = Person.objects.create(name='Sara', gender='F')
|
|
self.assertEqual(s.gender, 'F')
|
|
|
|
s.gender = 'M'
|
|
s.name = 'Ian'
|
|
s.save(update_fields=['name'])
|
|
|
|
s = Person.objects.get(pk=s.pk)
|
|
self.assertEqual(s.gender, 'F')
|
|
self.assertEqual(s.name, 'Ian')
|
|
|
|
def test_update_fields_deferred(self):
|
|
s = Person.objects.create(name='Sara', gender='F', pid=22)
|
|
self.assertEqual(s.gender, 'F')
|
|
|
|
s1 = Person.objects.defer("gender", "pid").get(pk=s.pk)
|
|
s1.name = "Emily"
|
|
s1.gender = "M"
|
|
|
|
with self.assertNumQueries(1):
|
|
s1.save()
|
|
|
|
s2 = Person.objects.get(pk=s1.pk)
|
|
self.assertEqual(s2.name, "Emily")
|
|
self.assertEqual(s2.gender, "M")
|
|
|
|
def test_update_fields_only_1(self):
|
|
s = Person.objects.create(name='Sara', gender='F')
|
|
self.assertEqual(s.gender, 'F')
|
|
|
|
s1 = Person.objects.only('name').get(pk=s.pk)
|
|
s1.name = "Emily"
|
|
s1.gender = "M"
|
|
|
|
with self.assertNumQueries(1):
|
|
s1.save()
|
|
|
|
s2 = Person.objects.get(pk=s1.pk)
|
|
self.assertEqual(s2.name, "Emily")
|
|
self.assertEqual(s2.gender, "M")
|
|
|
|
def test_update_fields_only_2(self):
|
|
s = Person.objects.create(name='Sara', gender='F', pid=22)
|
|
self.assertEqual(s.gender, 'F')
|
|
|
|
s1 = Person.objects.only('name').get(pk=s.pk)
|
|
s1.name = "Emily"
|
|
s1.gender = "M"
|
|
|
|
with self.assertNumQueries(2):
|
|
s1.save(update_fields=['pid'])
|
|
|
|
s2 = Person.objects.get(pk=s1.pk)
|
|
self.assertEqual(s2.name, "Sara")
|
|
self.assertEqual(s2.gender, "F")
|
|
|
|
def test_update_fields_only_repeated(self):
|
|
s = Person.objects.create(name='Sara', gender='F')
|
|
self.assertEqual(s.gender, 'F')
|
|
|
|
s1 = Person.objects.only('name').get(pk=s.pk)
|
|
s1.gender = 'M'
|
|
with self.assertNumQueries(1):
|
|
s1.save()
|
|
# Test that the deferred class does not remember that gender was
|
|
# set, instead the instace should remember this.
|
|
s1 = Person.objects.only('name').get(pk=s.pk)
|
|
with self.assertNumQueries(1):
|
|
s1.save()
|
|
|
|
def test_update_fields_inheritance_defer(self):
|
|
profile_boss = Profile.objects.create(name='Boss', salary=3000)
|
|
e1 = Employee.objects.create(name='Sara', gender='F',
|
|
employee_num=1, profile=profile_boss)
|
|
e1 = Employee.objects.only('name').get(pk=e1.pk)
|
|
e1.name = 'Linda'
|
|
with self.assertNumQueries(1):
|
|
e1.save()
|
|
self.assertEqual(Employee.objects.get(pk=e1.pk).name,
|
|
'Linda')
|
|
|
|
def test_update_fields_fk_defer(self):
|
|
profile_boss = Profile.objects.create(name='Boss', salary=3000)
|
|
profile_receptionist = Profile.objects.create(name='Receptionist', salary=1000)
|
|
e1 = Employee.objects.create(name='Sara', gender='F',
|
|
employee_num=1, profile=profile_boss)
|
|
e1 = Employee.objects.only('profile').get(pk=e1.pk)
|
|
e1.profile = profile_receptionist
|
|
with self.assertNumQueries(1):
|
|
e1.save()
|
|
self.assertEqual(Employee.objects.get(pk=e1.pk).profile, profile_receptionist)
|
|
e1.profile_id = profile_boss.pk
|
|
with self.assertNumQueries(1):
|
|
e1.save()
|
|
self.assertEqual(Employee.objects.get(pk=e1.pk).profile, profile_boss)
|
|
|
|
def test_select_related_only_interaction(self):
|
|
profile_boss = Profile.objects.create(name='Boss', salary=3000)
|
|
e1 = Employee.objects.create(name='Sara', gender='F',
|
|
employee_num=1, profile=profile_boss)
|
|
e1 = Employee.objects.only('profile__salary').select_related('profile').get(pk=e1.pk)
|
|
profile_boss.name = 'Clerk'
|
|
profile_boss.salary = 1000
|
|
profile_boss.save()
|
|
# The loaded salary of 3000 gets saved, the name of 'Clerk' isn't
|
|
# overwritten.
|
|
with self.assertNumQueries(1):
|
|
e1.profile.save()
|
|
reloaded_profile = Profile.objects.get(pk=profile_boss.pk)
|
|
self.assertEqual(reloaded_profile.name, profile_boss.name)
|
|
self.assertEqual(reloaded_profile.salary, 3000)
|
|
|
|
def test_update_fields_m2m(self):
|
|
profile_boss = Profile.objects.create(name='Boss', salary=3000)
|
|
e1 = Employee.objects.create(name='Sara', gender='F',
|
|
employee_num=1, profile=profile_boss)
|
|
|
|
a1 = Account.objects.create(num=1)
|
|
a2 = Account.objects.create(num=2)
|
|
|
|
e1.accounts = [a1,a2]
|
|
|
|
with self.assertRaises(ValueError):
|
|
e1.save(update_fields=['accounts'])
|
|
|
|
def test_update_fields_inheritance(self):
|
|
profile_boss = Profile.objects.create(name='Boss', salary=3000)
|
|
profile_receptionist = Profile.objects.create(name='Receptionist', salary=1000)
|
|
|
|
e1 = Employee.objects.create(name='Sara', gender='F',
|
|
employee_num=1, profile=profile_boss)
|
|
|
|
e1.name = 'Ian'
|
|
e1.gender = 'M'
|
|
e1.save(update_fields=['name'])
|
|
|
|
e2 = Employee.objects.get(pk=e1.pk)
|
|
self.assertEqual(e2.name, 'Ian')
|
|
self.assertEqual(e2.gender, 'F')
|
|
self.assertEqual(e2.profile, profile_boss)
|
|
|
|
e2.profile = profile_receptionist
|
|
e2.name = 'Sara'
|
|
e2.save(update_fields=['profile'])
|
|
|
|
e3 = Employee.objects.get(pk=e1.pk)
|
|
self.assertEqual(e3.name, 'Ian')
|
|
self.assertEqual(e3.profile, profile_receptionist)
|
|
|
|
with self.assertNumQueries(1):
|
|
e3.profile = profile_boss
|
|
e3.save(update_fields=['profile_id'])
|
|
|
|
e4 = Employee.objects.get(pk=e3.pk)
|
|
self.assertEqual(e4.profile, profile_boss)
|
|
self.assertEqual(e4.profile_id, profile_boss.pk)
|
|
|
|
def test_update_fields_inheritance_with_proxy_model(self):
|
|
profile_boss = Profile.objects.create(name='Boss', salary=3000)
|
|
profile_receptionist = Profile.objects.create(name='Receptionist', salary=1000)
|
|
|
|
e1 = ProxyEmployee.objects.create(name='Sara', gender='F',
|
|
employee_num=1, profile=profile_boss)
|
|
|
|
e1.name = 'Ian'
|
|
e1.gender = 'M'
|
|
e1.save(update_fields=['name'])
|
|
|
|
e2 = ProxyEmployee.objects.get(pk=e1.pk)
|
|
self.assertEqual(e2.name, 'Ian')
|
|
self.assertEqual(e2.gender, 'F')
|
|
self.assertEqual(e2.profile, profile_boss)
|
|
|
|
e2.profile = profile_receptionist
|
|
e2.name = 'Sara'
|
|
e2.save(update_fields=['profile'])
|
|
|
|
e3 = ProxyEmployee.objects.get(pk=e1.pk)
|
|
self.assertEqual(e3.name, 'Ian')
|
|
self.assertEqual(e3.profile, profile_receptionist)
|
|
|
|
def test_update_fields_signals(self):
|
|
p = Person.objects.create(name='Sara', gender='F')
|
|
pre_save_data = []
|
|
def pre_save_receiver(**kwargs):
|
|
pre_save_data.append(kwargs['update_fields'])
|
|
pre_save.connect(pre_save_receiver)
|
|
post_save_data = []
|
|
def post_save_receiver(**kwargs):
|
|
post_save_data.append(kwargs['update_fields'])
|
|
post_save.connect(post_save_receiver)
|
|
p.save(update_fields=['name'])
|
|
self.assertEqual(len(pre_save_data), 1)
|
|
self.assertEqual(len(pre_save_data[0]), 1)
|
|
self.assertTrue('name' in pre_save_data[0])
|
|
self.assertEqual(len(post_save_data), 1)
|
|
self.assertEqual(len(post_save_data[0]), 1)
|
|
self.assertTrue('name' in post_save_data[0])
|
|
|
|
def test_update_fields_incorrect_params(self):
|
|
s = Person.objects.create(name='Sara', gender='F')
|
|
|
|
with self.assertRaises(ValueError):
|
|
s.save(update_fields=['first_name'])
|
|
|
|
with self.assertRaises(ValueError):
|
|
s.save(update_fields="name")
|
|
|
|
def test_empty_update_fields(self):
|
|
s = Person.objects.create(name='Sara', gender='F')
|
|
pre_save_data = []
|
|
def pre_save_receiver(**kwargs):
|
|
pre_save_data.append(kwargs['update_fields'])
|
|
pre_save.connect(pre_save_receiver)
|
|
post_save_data = []
|
|
def post_save_receiver(**kwargs):
|
|
post_save_data.append(kwargs['update_fields'])
|
|
post_save.connect(post_save_receiver)
|
|
# Save is skipped.
|
|
with self.assertNumQueries(0):
|
|
s.save(update_fields=[])
|
|
# Signals were skipped, too...
|
|
self.assertEqual(len(pre_save_data), 0)
|
|
self.assertEqual(len(post_save_data), 0)
|
|
|
|
def test_num_queries_inheritance(self):
|
|
s = Employee.objects.create(name='Sara', gender='F')
|
|
s.employee_num = 1
|
|
s.name = 'Emily'
|
|
with self.assertNumQueries(1):
|
|
s.save(update_fields=['employee_num'])
|
|
s = Employee.objects.get(pk=s.pk)
|
|
self.assertEqual(s.employee_num, 1)
|
|
self.assertEqual(s.name, 'Sara')
|
|
s.employee_num = 2
|
|
s.name = 'Emily'
|
|
with self.assertNumQueries(1):
|
|
s.save(update_fields=['name'])
|
|
s = Employee.objects.get(pk=s.pk)
|
|
self.assertEqual(s.name, 'Emily')
|
|
self.assertEqual(s.employee_num, 1)
|
|
# A little sanity check that we actually did updates...
|
|
self.assertEqual(Employee.objects.count(), 1)
|
|
self.assertEqual(Person.objects.count(), 1)
|
|
with self.assertNumQueries(2):
|
|
s.save(update_fields=['name', 'employee_num'])
|