2380151db9
SQLAlchemy's documentation states if you want to create an in-memory database you can provide a URL of `sqlite://` and that `:memory` is implied. The `database_exists` and `create_database` function did not handle this use-case.
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):
|
|
(
|
|
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'
|
|
)
|