/tests/modeltests/update/tests.py
Python | 116 lines | 65 code | 15 blank | 36 comment | 1 complexity | 6502e606efe91796c30b8db3da3393af MD5 | raw file
Possible License(s): BSD-3-Clause
- from django.test import TestCase
- from models import A, B, C, D, DataPoint, RelatedPoint
- class SimpleTest(TestCase):
- def setUp(self):
- self.a1 = A.objects.create()
- self.a2 = A.objects.create()
- for x in range(20):
- B.objects.create(a=self.a1)
- D.objects.create(a=self.a1)
- def test_nonempty_update(self):
- """
- Test that update changes the right number of rows for a nonempty queryset
- """
- num_updated = self.a1.b_set.update(y=100)
- self.assertEqual(num_updated, 20)
- cnt = B.objects.filter(y=100).count()
- self.assertEqual(cnt, 20)
- def test_empty_update(self):
- """
- Test that update changes the right number of rows for an empty queryset
- """
- num_updated = self.a2.b_set.update(y=100)
- self.assertEqual(num_updated, 0)
- cnt = B.objects.filter(y=100).count()
- self.assertEqual(cnt, 0)
- def test_nonempty_update_with_inheritance(self):
- """
- Test that update changes the right number of rows for an empty queryset
- when the update affects only a base table
- """
- num_updated = self.a1.d_set.update(y=100)
- self.assertEqual(num_updated, 20)
- cnt = D.objects.filter(y=100).count()
- self.assertEqual(cnt, 20)
- def test_empty_update_with_inheritance(self):
- """
- Test that update changes the right number of rows for an empty queryset
- when the update affects only a base table
- """
- num_updated = self.a2.d_set.update(y=100)
- self.assertEqual(num_updated, 0)
- cnt = D.objects.filter(y=100).count()
- self.assertEqual(cnt, 0)
- class AdvancedTests(TestCase):
- def setUp(self):
- self.d0 = DataPoint.objects.create(name="d0", value="apple")
- self.d2 = DataPoint.objects.create(name="d2", value="banana")
- self.d3 = DataPoint.objects.create(name="d3", value="banana")
- self.r1 = RelatedPoint.objects.create(name="r1", data=self.d3)
- def test_update(self):
- """
- Objects are updated by first filtering the candidates into a queryset
- and then calling the update() method. It executes immediately and
- returns nothing.
- """
- resp = DataPoint.objects.filter(value="apple").update(name="d1")
- self.assertEqual(resp, 1)
- resp = DataPoint.objects.filter(value="apple")
- self.assertEqual(list(resp), [self.d0])
- def test_update_multiple_objects(self):
- """
- We can update multiple objects at once.
- """
- resp = DataPoint.objects.filter(value="banana").update(
- value="pineapple")
- self.assertEqual(resp, 2)
- self.assertEqual(DataPoint.objects.get(name="d2").value, u'pineapple')
- def test_update_fk(self):
- """
- Foreign key fields can also be updated, although you can only update
- the object referred to, not anything inside the related object.
- """
- resp = RelatedPoint.objects.filter(name="r1").update(data=self.d0)
- self.assertEqual(resp, 1)
- resp = RelatedPoint.objects.filter(data__name="d0")
- self.assertEqual(list(resp), [self.r1])
- def test_update_multiple_fields(self):
- """
- Multiple fields can be updated at once
- """
- resp = DataPoint.objects.filter(value="apple").update(
- value="fruit", another_value="peach")
- self.assertEqual(resp, 1)
- d = DataPoint.objects.get(name="d0")
- self.assertEqual(d.value, u'fruit')
- self.assertEqual(d.another_value, u'peach')
- def test_update_all(self):
- """
- In the rare case you want to update every instance of a model, update()
- is also a manager method.
- """
- self.assertEqual(DataPoint.objects.update(value='thing'), 3)
- resp = DataPoint.objects.values('value').distinct()
- self.assertEqual(list(resp), [{'value': u'thing'}])
- def test_update_slice_fail(self):
- """
- We do not support update on already sliced query sets.
- """
- method = DataPoint.objects.all()[:2].update
- self.assertRaises(AssertionError, method,
- another_value='another thing')