6b95fc966d
Before it was hard-coded to the `postgres` user. I’m not quite sure why these tests failed for me, considering that the calls to the engine should be mocked. Perhaps a new version of `flexmock` has changed something. Downgrading SQLAlchemy did not help.
102 lines
2.6 KiB
Python
102 lines
2.6 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.usefixture('sqlite_none_database_dsn')
|
|
class TestDatabaseSQLiteMemoryNoDatabaseString(object):
|
|
def test_exists_memory_none_database(self, sqlite_none_database_dsn):
|
|
assert database_exists(sqlite_none_database_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, postgresql_db_user):
|
|
(
|
|
flexmock(sa.engine.Engine)
|
|
.should_receive('execute')
|
|
.with_args(
|
|
"CREATE DATABASE db_test_sqlalchemy_util ENCODING 'utf8' "
|
|
"TEMPLATE my_template"
|
|
)
|
|
)
|
|
dsn = 'postgres://{0}@localhost/db_test_sqlalchemy_util'.format(
|
|
postgresql_db_user
|
|
)
|
|
create_database(dsn, 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, postgresql_db_user):
|
|
(
|
|
flexmock(sa.engine.Engine)
|
|
.should_receive('execute')
|
|
.with_args(
|
|
'''CREATE DATABASE "db_test_sqlalchemy-util"'''
|
|
" ENCODING 'utf8' "
|
|
'TEMPLATE "my-template"'
|
|
)
|
|
)
|
|
dsn = 'postgres://{0}@localhost/db_test_sqlalchemy-util'.format(
|
|
postgresql_db_user
|
|
)
|
|
create_database(dsn, template='my-template')
|