From 0abf641001b9c99addd1c3c950449d559e133557 Mon Sep 17 00:00:00 2001
From: Benjamin Dauvergne <bdauvergne@entrouvert.com>
Date: Thu, 12 Apr 2018 11:10:21 +0200
Subject: [PATCH] multitenant: wrap schema creation in atomic() (#23119)

---
 .../management/commands/create_tenant.py      |  5 +-
 tests_multitenant/test_create_tenant.py       | 65 +++++++++++++++++++
 2 files changed, 68 insertions(+), 2 deletions(-)
 create mode 100644 tests_multitenant/test_create_tenant.py

diff --git a/hobo/multitenant/management/commands/create_tenant.py b/hobo/multitenant/management/commands/create_tenant.py
index d25f7f4..cda29fc 100644
--- a/hobo/multitenant/management/commands/create_tenant.py
+++ b/hobo/multitenant/management/commands/create_tenant.py
@@ -1,7 +1,7 @@
 import os
 import sys
 
-from django.db import connection
+from django.db import connection, transaction
 from django.core.management.base import CommandError, BaseCommand
 
 from hobo.multitenant.middleware import TenantMiddleware, get_tenant_model
@@ -47,7 +47,8 @@ class Command(BaseCommand):
                     print
                     print self.style.NOTICE("=== Creating schema ") \
                         + self.style.SQL_TABLE(tenant.schema_name)
-                tenant.create_schema(check_if_exists=True)
+                with transaction.atomic():
+                    tenant.create_schema(check_if_exists=True)
             except Exception as e:
                 os.rmdir(tenant_dir_tmp)
                 raise CommandError('tenant creation failed (%s)' % str(e))
diff --git a/tests_multitenant/test_create_tenant.py b/tests_multitenant/test_create_tenant.py
new file mode 100644
index 0000000..b7aafcf
--- /dev/null
+++ b/tests_multitenant/test_create_tenant.py
@@ -0,0 +1,65 @@
+# hobo - portal to configure and deploy applications
+# Copyright (C) 2019  Entr'ouvert
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from django.db import connection
+from django.core.management import call_command
+from django.core.management.base import CommandError
+from django.contrib.auth.models import User
+
+from tenant_schemas.utils import tenant_context
+
+from hobo.multitenant.middleware import TenantMiddleware
+
+import mock
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def configuration(settings, tmpdir):
+    settings.TENANT_BASE = str(tmpdir.mkdir('tenants'))
+
+
+def schema_exists(schema_name):
+    with connection.cursor() as cursor:
+        cursor.execute('select schema_name from information_schema.schemata')
+        return 'www_example_com' in [row[0] for row in cursor.fetchall()]
+
+
+def test_create_tenant(db):
+    assert not schema_exists('www_example_com')
+    call_command('create_tenant', 'www.example.com')
+    assert schema_exists('www_example_com')
+    tenants = list(TenantMiddleware.get_tenants())
+    assert len(tenants) == 1
+    tenant = tenants[0]
+    assert tenant.domain_url == 'www.example.com'
+    assert tenant.schema_name == 'www_example_com'
+    with tenant_context(tenant):
+        User.objects.create(username='admin')
+
+
+def test_create_tenant_failure(db, caplog):
+    with mock.patch('hobo.multitenant.management.commands.migrate_schemas.MigrateSchemasCommand.handle') as handle:
+        handle.side_effect = CommandError('unable to migrate')
+        assert not schema_exists('www_example_com')
+        with pytest.raises(CommandError) as exc_info:
+            call_command('create_tenant', 'www.example.com')
+        assert str(exc_info.value) == 'tenant creation failed (unable to migrate)'
+        assert not schema_exists('www_example_com')
+
+    with connection.cursor() as cursor:
+        cursor.execute('select schema_name from information_schema.schemata')
+        assert 'www_example_com' not in [row[0] for row in cursor.fetchall()]
-- 
2.20.1