mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-13 19:42:20 -05:00
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
|
from .. import fixtures, config
|
||
|
from ..assertions import eq_
|
||
|
|
||
|
from sqlalchemy import Integer, String
|
||
|
from ..schema import Table, Column
|
||
|
|
||
|
|
||
|
class SimpleUpdateDeleteTest(fixtures.TablesTest):
|
||
|
run_deletes = 'each'
|
||
|
__backend__ = True
|
||
|
|
||
|
@classmethod
|
||
|
def define_tables(cls, metadata):
|
||
|
Table('plain_pk', metadata,
|
||
|
Column('id', Integer, primary_key=True),
|
||
|
Column('data', String(50))
|
||
|
)
|
||
|
|
||
|
@classmethod
|
||
|
def insert_data(cls):
|
||
|
config.db.execute(
|
||
|
cls.tables.plain_pk.insert(),
|
||
|
[
|
||
|
{"id":1, "data":"d1"},
|
||
|
{"id":2, "data":"d2"},
|
||
|
{"id":3, "data":"d3"},
|
||
|
]
|
||
|
)
|
||
|
|
||
|
def test_update(self):
|
||
|
t = self.tables.plain_pk
|
||
|
r = config.db.execute(
|
||
|
t.update().where(t.c.id == 2),
|
||
|
data="d2_new"
|
||
|
)
|
||
|
assert not r.is_insert
|
||
|
assert not r.returns_rows
|
||
|
|
||
|
eq_(
|
||
|
config.db.execute(t.select().order_by(t.c.id)).fetchall(),
|
||
|
[
|
||
|
(1, "d1"),
|
||
|
(2, "d2_new"),
|
||
|
(3, "d3")
|
||
|
]
|
||
|
)
|
||
|
|
||
|
def test_delete(self):
|
||
|
t = self.tables.plain_pk
|
||
|
r = config.db.execute(
|
||
|
t.delete().where(t.c.id == 2)
|
||
|
)
|
||
|
assert not r.is_insert
|
||
|
assert not r.returns_rows
|
||
|
eq_(
|
||
|
config.db.execute(t.select().order_by(t.c.id)).fetchall(),
|
||
|
[
|
||
|
(1, "d1"),
|
||
|
(3, "d3")
|
||
|
]
|
||
|
)
|
||
|
|
||
|
__all__ = ('SimpleUpdateDeleteTest', )
|