def test_new_aborted_creation(self):
"Tests recovery on automatic creation for new tenant's save"
def signal_receiver(*args, **kwargs):
raise Exception
self.assertFalse(schema_exists("tenant1"))
tenant = TenantModel(schema_name="tenant1")
schema_post_sync.connect(signal_receiver)
with self.assertRaises(Exception):
tenant.save(verbosity=0)
self.assertFalse(schema_exists("tenant1"))
self.assertEqual(0, TenantModel.objects.count())
schema_post_sync.disconnect(signal_receiver)
def test_existing_aborted_creation(self):
"Tests recovery on automatic creation for new tenant's save"
def signal_receiver(*args, **kwargs):
raise Exception
self.assertFalse(schema_exists("tenant1"))
tenant = TenantModel(schema_name="tenant1")
tenant.auto_create_schema = False
tenant.save(verbosity=0)
tenant.auto_create_schema = True
schema_post_sync.connect(signal_receiver)
with self.assertRaises(Exception):
tenant.save(verbosity=0)
self.assertFalse(schema_exists("tenant1"))
self.assertEqual(1, TenantModel.objects.count())
schema_post_sync.disconnect(signal_receiver)
# Self-cleanup
tenant.delete(force_drop=True)
self.assertEqual(0, TenantModel.objects.count())