def test_bulk_insert_returning_ids(self):
m1 = Medium(name="test_bi_1")
m2 = Medium(name="test_bi_2")
self.assertIsNone(m1.id)
self.assertIsNone(m2.id)
new_objects = bulk_insert_returning_ids([m1, m2])
self.assertIsNone(m1.id)
self.assertIsNone(m2.id)
self.assertIsNotNone(new_objects[0].id)
self.assertIsNotNone(new_objects[1].id)
self.assertEqual("test_bi_1",
Medium.objects.get(id=new_objects[0].id).name)
self.assertEqual("test_bi_2",
Medium.objects.get(id=new_objects[1].id).name)
def _create_medium(self, source):
from amcat.models import Language
try:
Medium.objects.get(name__iexact=source)
except Medium.DoesNotExist:
l = Language.objects.get(id=1)
Medium(name=source, abbrev=source[0:5], circulation=1,
language=l).save()
