96 lines
2.3 KiB
Python
96 lines
2.3 KiB
Python
import pytest
|
|
import sqlalchemy as sa
|
|
from flexmock import flexmock
|
|
|
|
from sqlalchemy_utils import create_database, database_exists, drop_database
|
|
|
|
pymysql = None
|
|
try:
|
|
import pymysql # noqa
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
class DatabaseTest(object):
|
|
def test_create_and_drop(self, dsn):
|
|
assert not database_exists(dsn)
|
|
create_database(dsn)
|
|
assert database_exists(dsn)
|
|
drop_database(dsn)
|
|
assert not database_exists(dsn)
|
|
|
|
|
|
@pytest.mark.usefixtures('sqlite_memory_dsn')
|
|
class TestDatabaseSQLiteMemory(object):
|
|
|
|
def test_exists_memory(self, dsn):
|
|
assert database_exists(dsn)
|
|
|
|
|
|
@pytest.mark.usefixtures('sqlite_file_dsn')
|
|
class TestDatabaseSQLiteFile(DatabaseTest):
|
|
pass
|
|
|
|
|
|
@pytest.mark.skipif('pymysql is None')
|
|
@pytest.mark.usefixtures('mysql_dsn')
|
|
class TestDatabaseMySQL(DatabaseTest):
|
|
|
|
@pytest.fixture
|
|
def db_name(self):
|
|
return 'db_test_sqlalchemy_util'
|
|
|
|
|
|
@pytest.mark.skipif('pymysql is None')
|
|
@pytest.mark.usefixtures('mysql_dsn')
|
|
class TestDatabaseMySQLWithQuotedName(DatabaseTest):
|
|
|
|
@pytest.fixture
|
|
def db_name(self):
|
|
return 'db_test_sqlalchemy-util'
|
|
|
|
|
|
@pytest.mark.usefixtures('postgresql_dsn')
|
|
class TestDatabasePostgres(DatabaseTest):
|
|
|
|
@pytest.fixture
|
|
def db_name(self):
|
|
return 'db_test_sqlalchemy_util'
|
|
|
|
def test_template(self):
|
|
(
|
|
flexmock(sa.engine.Engine)
|
|
.should_receive('execute')
|
|
.with_args(
|
|
"CREATE DATABASE db_test_sqlalchemy_util ENCODING 'utf8' "
|
|
"TEMPLATE my_template"
|
|
)
|
|
)
|
|
create_database(
|
|
'postgres://postgres@localhost/db_test_sqlalchemy_util',
|
|
template='my_template'
|
|
)
|
|
|
|
|
|
@pytest.mark.usefixtures('postgresql_dsn')
|
|
class TestDatabasePostgresWithQuotedName(DatabaseTest):
|
|
|
|
@pytest.fixture
|
|
def db_name(self):
|
|
return 'db_test_sqlalchemy-util'
|
|
|
|
def test_template(self):
|
|
(
|
|
flexmock(sa.engine.Engine)
|
|
.should_receive('execute')
|
|
.with_args(
|
|
'''CREATE DATABASE "db_test_sqlalchemy-util"'''
|
|
" ENCODING 'utf8' "
|
|
'TEMPLATE "my-template"'
|
|
)
|
|
)
|
|
create_database(
|
|
'postgres://postgres@localhost/db_test_sqlalchemy-util',
|
|
template='my-template'
|
|
)
|