91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
from __future__ import absolute_import
|
|
|
|
from datetime import date
|
|
import traceback
|
|
|
|
from django.db import IntegrityError
|
|
from django.test import TestCase
|
|
|
|
from .models import Person, ManualPrimaryKeyTest
|
|
|
|
|
|
class GetOrCreateTests(TestCase):
|
|
def test_get_or_create(self):
|
|
p = Person.objects.create(
|
|
first_name='John', last_name='Lennon', birthday=date(1940, 10, 9)
|
|
)
|
|
|
|
p, created = Person.objects.get_or_create(
|
|
first_name="John", last_name="Lennon", defaults={
|
|
"birthday": date(1940, 10, 9)
|
|
}
|
|
)
|
|
self.assertFalse(created)
|
|
self.assertEqual(Person.objects.count(), 1)
|
|
|
|
p, created = Person.objects.get_or_create(
|
|
first_name='George', last_name='Harrison', defaults={
|
|
'birthday': date(1943, 2, 25)
|
|
}
|
|
)
|
|
self.assertTrue(created)
|
|
self.assertEqual(Person.objects.count(), 2)
|
|
|
|
# If we execute the exact same statement, it won't create a Person.
|
|
p, created = Person.objects.get_or_create(
|
|
first_name='George', last_name='Harrison', defaults={
|
|
'birthday': date(1943, 2, 25)
|
|
}
|
|
)
|
|
self.assertFalse(created)
|
|
self.assertEqual(Person.objects.count(), 2)
|
|
|
|
# If you don't specify a value or default value for all required
|
|
# fields, you will get an error.
|
|
self.assertRaises(IntegrityError,
|
|
Person.objects.get_or_create, first_name="Tom", last_name="Smith"
|
|
)
|
|
|
|
# If you specify an existing primary key, but different other fields,
|
|
# then you will get an error and data will not be updated.
|
|
m = ManualPrimaryKeyTest.objects.create(id=1, data="Original")
|
|
self.assertRaises(IntegrityError,
|
|
ManualPrimaryKeyTest.objects.get_or_create, id=1, data="Different"
|
|
)
|
|
self.assertEqual(ManualPrimaryKeyTest.objects.get(id=1).data, "Original")
|
|
|
|
# get_or_create should raise IntegrityErrors with the full traceback.
|
|
# This is tested by checking that a known method call is in the traceback.
|
|
# We cannot use assertRaises/assertRaises here because we need to inspect
|
|
# the actual traceback. Refs #16340.
|
|
try:
|
|
ManualPrimaryKeyTest.objects.get_or_create(id=1, data="Different")
|
|
except IntegrityError as e:
|
|
formatted_traceback = traceback.format_exc()
|
|
self.assertIn('obj.save', formatted_traceback)
|
|
|
|
|
|
def test_initial_get_on_read_db(self):
|
|
"""
|
|
get_or_create should only set _for_write when it's actually doing a
|
|
create action. This makes sure that the initial .get() will be able to
|
|
use a slave database. Specially when some form of database pinning is
|
|
in place this will help to not put all the SELECT queries on the
|
|
master. Refs #16865.
|
|
|
|
"""
|
|
qs = Person.objects.get_query_set()
|
|
p, created = qs.get_or_create(
|
|
first_name="Stuart", last_name="Sutcliffe", defaults={
|
|
"birthday": date(1940, 6, 23),
|
|
}
|
|
)
|
|
self.assertTrue(created)
|
|
self.assertTrue(qs._for_write)
|
|
|
|
qs = Person.objects.get_query_set()
|
|
p, created = qs.get_or_create(
|
|
first_name="Stuart", last_name="Sutcliffe")
|
|
self.assertFalse(created)
|
|
self.assertFalse(qs._for_write)
|