diff options
Diffstat (limited to 'parts/django/tests/modeltests/update/tests.py')
-rw-r--r-- | parts/django/tests/modeltests/update/tests.py | 116 |
1 files changed, 0 insertions, 116 deletions
diff --git a/parts/django/tests/modeltests/update/tests.py b/parts/django/tests/modeltests/update/tests.py deleted file mode 100644 index d0b6ea3..0000000 --- a/parts/django/tests/modeltests/update/tests.py +++ /dev/null @@ -1,116 +0,0 @@ -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') |