Merge pull request #239 from browniebroke/bugfix/softdelete-multi-db

SoftDeletableModel: use correct DB connection
This commit is contained in:
Carl Meyer 2016-11-09 13:00:24 -07:00 committed by GitHub
commit 59b5fa24f7
2 changed files with 8 additions and 1 deletions

View file

@ -120,4 +120,4 @@ class SoftDeletableModel(models.Model):
Soft delete object (set its ``is_removed`` field to True)
"""
self.is_removed = True
self.save()
self.save(using=using)

View file

@ -1,6 +1,7 @@
from __future__ import unicode_literals
from datetime import datetime, timedelta
try:
from unittest import skipUnless
except ImportError: # Python 2.6
@ -9,6 +10,7 @@ except ImportError: # Python 2.6
import django
from django.db import models
from django.db.models.fields import FieldDoesNotExist
from django.db.utils import ConnectionDoesNotExist
from django.utils.six import text_type
from django.core.exceptions import ImproperlyConfigured, FieldError
from django.core.management import call_command
@ -2004,3 +2006,8 @@ class SoftDeletableModelTests(TestCase):
self.assertEqual(SoftDeletable.objects.count(), 0)
self.assertEqual(SoftDeletable.all_objects.count(), 1)
def test_delete_instance_no_connection(self):
obj = SoftDeletable.objects.create(name='a')
self.assertRaises(ConnectionDoesNotExist, obj.delete, using='other')