mirror of
https://github.com/janeczku/calibre-web
synced 2024-12-29 11:30:31 +00:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
|
from __future__ import with_statement
|
||
|
|
|
||
|
|
from .. import fixtures, config, util
|
||
|
|
from ..config import requirements
|
||
|
|
from ..assertions import eq_
|
||
|
|
|
||
|
|
from sqlalchemy import Table, Column, Integer, String
|
||
|
|
|
||
|
|
|
||
|
|
class TableDDLTest(fixtures.TestBase):
|
||
|
|
|
||
|
|
def _simple_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
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ('TableDDLTest', )
|