mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-06 01:15:05 -05:00
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
|
|
|
|
from .. import fixtures, config, util
|
|
from ..config import requirements
|
|
from ..assertions import eq_
|
|
|
|
from sqlalchemy import Table, Column, Integer, String
|
|
|
|
|
|
class TableDDLTest(fixtures.TestBase):
|
|
__backend__ = True
|
|
|
|
def _simple_fixture(self):
|
|
return Table('test_table', self.metadata,
|
|
Column('id', Integer, primary_key=True, autoincrement=False),
|
|
Column('data', String(50))
|
|
)
|
|
|
|
def _underscore_fixture(self):
|
|
return Table('_test_table', self.metadata,
|
|
Column('id', Integer, primary_key=True, autoincrement=False),
|
|
Column('_data', String(50))
|
|
)
|
|
|
|
def _simple_roundtrip(self, table):
|
|
with config.db.begin() as conn:
|
|
conn.execute(table.insert().values((1, 'some data')))
|
|
result = conn.execute(table.select())
|
|
eq_(
|
|
result.first(),
|
|
(1, 'some data')
|
|
)
|
|
|
|
@requirements.create_table
|
|
@util.provide_metadata
|
|
def test_create_table(self):
|
|
table = self._simple_fixture()
|
|
table.create(
|
|
config.db, checkfirst=False
|
|
)
|
|
self._simple_roundtrip(table)
|
|
|
|
@requirements.drop_table
|
|
@util.provide_metadata
|
|
def test_drop_table(self):
|
|
table = self._simple_fixture()
|
|
table.create(
|
|
config.db, checkfirst=False
|
|
)
|
|
table.drop(
|
|
config.db, checkfirst=False
|
|
)
|
|
|
|
@requirements.create_table
|
|
@util.provide_metadata
|
|
def test_underscore_names(self):
|
|
table = self._underscore_fixture()
|
|
table.create(
|
|
config.db, checkfirst=False
|
|
)
|
|
self._simple_roundtrip(table)
|
|
|
|
__all__ = ('TableDDLTest', )
|