import unittest

from contextlib import contextmanager

from io import StringIO

from unittest import mock

from django.core.exceptions import ImproperlyConfigured

from django.db import DatabaseError, connection

from django.db.backends.base.creation import BaseDatabaseCreation

from django.test import SimpleTestCase

try:

    import psycopg2  # NOQA

except ImportError:

    pass

else:

    from psycopg2 import errorcodes

    from django.db.backends.postgresql.creation import DatabaseCreation

@unittest.skipUnless(connection.vendor == "postgresql", "PostgreSQL tests")

class DatabaseCreationTests(SimpleTestCase):

    @contextmanager

    def changed_test_settings(self, **kwargs):

        settings = connection.settings_dict["TEST"]

        saved_values = {}

        for name in kwargs:

            if name in settings:

                saved_values[name] = settings[name]

        for name, value in kwargs.items():

            settings[name] = value

        try:

            yield

        finally:

            for name in kwargs:

                if name in saved_values:

                    settings[name] = saved_values[name]

                else:

                    del settings[name]

    def check_sql_table_creation_suffix(self, settings, expected):

        with self.changed_test_settings(**settings):

            creation = DatabaseCreation(connection)

            suffix = creation.sql_table_creation_suffix()

            self.assertEqual(suffix, expected)

    def test_sql_table_creation_suffix_with_none_settings(self):

        settings = {"CHARSET": None, "TEMPLATE": None}

        self.check_sql_table_creation_suffix(settings, "")

    def test_sql_table_creation_suffix_with_encoding(self):

        settings = {"CHARSET": "UTF8"}

        self.check_sql_table_creation_suffix(settings, "WITH ENCODING 'UTF8'")

    def test_sql_table_creation_suffix_with_template(self):

        settings = {"TEMPLATE": "template0"}

        self.check_sql_table_creation_suffix(settings, 'WITH TEMPLATE "template0"')

    def test_sql_table_creation_suffix_with_encoding_and_template(self):

        settings = {"CHARSET": "UTF8", "TEMPLATE": "template0"}

        self.check_sql_table_creation_suffix(

            settings, '''WITH ENCODING 'UTF8' TEMPLATE "template0"'''

        )

    def test_sql_table_creation_raises_with_collation(self):

        settings = {"COLLATION": "test"}

        msg = (

            "PostgreSQL does not support collation setting at database "

            "creation time."

        )

        with self.assertRaisesMessage(ImproperlyConfigured, msg):

            self.check_sql_table_creation_suffix(settings, None)

    def _execute_raise_database_already_exists(self, cursor, parameters, keepdb=False):

        error = DatabaseError("database %s already exists" % parameters["dbname"])

        error.pgcode = errorcodes.DUPLICATE_DATABASE

        raise DatabaseError() from error

    def _execute_raise_permission_denied(self, cursor, parameters, keepdb=False):

        error = DatabaseError("permission denied to create database")

        error.pgcode = errorcodes.INSUFFICIENT_PRIVILEGE

        raise DatabaseError() from error

    def patch_test_db_creation(self, execute_create_test_db):

        return mock.patch.object(

            BaseDatabaseCreation, "_execute_create_test_db", execute_create_test_db

        )

    @mock.patch("sys.stdout", new_callable=StringIO)

    @mock.patch("sys.stderr", new_callable=StringIO)

    def test_create_test_db(self, *mocked_objects):

        creation = DatabaseCreation(connection)

        # Simulate test database creation raising "database already exists"

        with self.patch_test_db_creation(self._execute_raise_database_already_exists):

            with mock.patch("builtins.input", return_value="no"):

                with self.assertRaises(SystemExit):

                    # SystemExit is raised if the user answers "no" to the

                    # prompt asking if it's okay to delete the test database.

                    creation._create_test_db(

                        verbosity=0, autoclobber=False, keepdb=False

                    )

            # "Database already exists" error is ignored when keepdb is on

            creation._create_test_db(verbosity=0, autoclobber=False, keepdb=True)

        # Simulate test database creation raising unexpected error

        with self.patch_test_db_creation(self._execute_raise_permission_denied):

            with mock.patch.object(

                DatabaseCreation, "_database_exists", return_value=False

            ):

                with self.assertRaises(SystemExit):

                    creation._create_test_db(

                        verbosity=0, autoclobber=False, keepdb=False

                    )

                with self.assertRaises(SystemExit):

                    creation._create_test_db(

                        verbosity=0, autoclobber=False, keepdb=True

                    )

        # Simulate test database creation raising "insufficient privileges".

        # An error shouldn't appear when keepdb is on and the database already

        # exists.

        with self.patch_test_db_creation(self._execute_raise_permission_denied):

            with mock.patch.object(

                DatabaseCreation, "_database_exists", return_value=True

            ):

                creation._create_test_db(verbosity=0, autoclobber=False, keepdb=True)