mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-06-01 05:18:44 -04:00
Mike Bayer
065fcbd9d2
relationship(), to eliminate confusion over the relational algebra term. relation() however will remain available in equal capacity for the foreseeable future. [ticket:1740]
144 lines
4.9 KiB
Python
144 lines
4.9 KiB
Python
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.test import testing
|
|
from sqlalchemy import Integer, String, ForeignKey
|
|
from sqlalchemy.test.schema import Table, Column
|
|
from sqlalchemy.orm import mapper, relationship, create_session
|
|
from test.orm import _base
|
|
from sqlalchemy.test.testing import eq_
|
|
|
|
|
|
class TriggerDefaultsTest(_base.MappedTest):
|
|
__requires__ = ('row_triggers',)
|
|
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
dt = Table('dt', metadata,
|
|
Column('id', Integer, primary_key=True, test_needs_autoincrement=True),
|
|
Column('col1', String(20)),
|
|
Column('col2', String(20),
|
|
server_default=sa.schema.FetchedValue()),
|
|
Column('col3', String(20),
|
|
sa.schema.FetchedValue(for_update=True)),
|
|
Column('col4', String(20),
|
|
sa.schema.FetchedValue(),
|
|
sa.schema.FetchedValue(for_update=True)))
|
|
for ins in (
|
|
sa.DDL("CREATE TRIGGER dt_ins AFTER INSERT ON dt "
|
|
"FOR EACH ROW BEGIN "
|
|
"UPDATE dt SET col2='ins', col4='ins' "
|
|
"WHERE dt.id = NEW.id; END",
|
|
on='sqlite'),
|
|
sa.DDL("CREATE TRIGGER dt_ins ON dt AFTER INSERT AS "
|
|
"UPDATE dt SET col2='ins', col4='ins' "
|
|
"WHERE dt.id IN (SELECT id FROM inserted);",
|
|
on='mssql'),
|
|
sa.DDL("CREATE TRIGGER dt_ins BEFORE INSERT "
|
|
"ON dt "
|
|
"FOR EACH ROW "
|
|
"BEGIN "
|
|
":NEW.col2 := 'ins'; :NEW.col4 := 'ins'; END;",
|
|
on='oracle'),
|
|
sa.DDL("CREATE TRIGGER dt_ins BEFORE INSERT ON dt "
|
|
"FOR EACH ROW BEGIN "
|
|
"SET NEW.col2='ins'; SET NEW.col4='ins'; END",
|
|
on=lambda ddl, event, target, bind, **kw:
|
|
bind.engine.name not in ('oracle', 'mssql', 'sqlite')
|
|
),
|
|
):
|
|
ins.execute_at('after-create', dt)
|
|
|
|
sa.DDL("DROP TRIGGER dt_ins").execute_at('before-drop', dt)
|
|
|
|
for up in (
|
|
sa.DDL("CREATE TRIGGER dt_up AFTER UPDATE ON dt "
|
|
"FOR EACH ROW BEGIN "
|
|
"UPDATE dt SET col3='up', col4='up' "
|
|
"WHERE dt.id = OLD.id; END",
|
|
on='sqlite'),
|
|
sa.DDL("CREATE TRIGGER dt_up ON dt AFTER UPDATE AS "
|
|
"UPDATE dt SET col3='up', col4='up' "
|
|
"WHERE dt.id IN (SELECT id FROM deleted);",
|
|
on='mssql'),
|
|
sa.DDL("CREATE TRIGGER dt_up BEFORE UPDATE ON dt "
|
|
"FOR EACH ROW BEGIN "
|
|
":NEW.col3 := 'up'; :NEW.col4 := 'up'; END;",
|
|
on='oracle'),
|
|
sa.DDL("CREATE TRIGGER dt_up BEFORE UPDATE ON dt "
|
|
"FOR EACH ROW BEGIN "
|
|
"SET NEW.col3='up'; SET NEW.col4='up'; END",
|
|
on=lambda ddl, event, target, bind, **kw:
|
|
bind.engine.name not in ('oracle', 'mssql', 'sqlite')
|
|
),
|
|
):
|
|
up.execute_at('after-create', dt)
|
|
|
|
sa.DDL("DROP TRIGGER dt_up").execute_at('before-drop', dt)
|
|
|
|
|
|
@classmethod
|
|
def setup_classes(cls):
|
|
class Default(_base.BasicEntity):
|
|
pass
|
|
|
|
@classmethod
|
|
@testing.resolve_artifact_names
|
|
def setup_mappers(cls):
|
|
mapper(Default, dt)
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_insert(self):
|
|
|
|
d1 = Default(id=1)
|
|
|
|
eq_(d1.col1, None)
|
|
eq_(d1.col2, None)
|
|
eq_(d1.col3, None)
|
|
eq_(d1.col4, None)
|
|
|
|
session = create_session()
|
|
session.add(d1)
|
|
session.flush()
|
|
|
|
eq_(d1.col1, None)
|
|
eq_(d1.col2, 'ins')
|
|
eq_(d1.col3, None)
|
|
# don't care which trigger fired
|
|
assert d1.col4 in ('ins', 'up')
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_update(self):
|
|
d1 = Default(id=1)
|
|
|
|
session = create_session()
|
|
session.add(d1)
|
|
session.flush()
|
|
d1.col1 = 'set'
|
|
session.flush()
|
|
|
|
eq_(d1.col1, 'set')
|
|
eq_(d1.col2, 'ins')
|
|
eq_(d1.col3, 'up')
|
|
eq_(d1.col4, 'up')
|
|
|
|
class ExcludedDefaultsTest(_base.MappedTest):
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
dt = Table('dt', metadata,
|
|
Column('id', Integer, primary_key=True, test_needs_autoincrement=True),
|
|
Column('col1', String(20), default="hello"),
|
|
)
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_exclude(self):
|
|
class Foo(_base.ComparableEntity):
|
|
pass
|
|
mapper(Foo, dt, exclude_properties=('col1',))
|
|
|
|
f1 = Foo()
|
|
sess = create_session()
|
|
sess.add(f1)
|
|
sess.flush()
|
|
eq_(dt.select().execute().fetchall(), [(1, "hello")])
|
|
|