mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-06-01 05:18:44 -04:00
Mike Bayer
dbc582a43f
- Usage of version_id_col on a backend that supports cursor.rowcount for execute() but not executemany() now works when a delete is issued (already worked for saves, since those don't use executemany()). For a backend that doesn't support cursor.rowcount at all, a warning is emitted the same as with saves. [ticket:1761]
342 lines
11 KiB
Python
342 lines
11 KiB
Python
import sqlalchemy as sa
|
|
from sqlalchemy.test import engines, testing
|
|
from sqlalchemy import Integer, String, ForeignKey, literal_column, orm, exc
|
|
from sqlalchemy.test.schema import Table, Column
|
|
from sqlalchemy.orm import mapper, relationship, create_session, column_property, sessionmaker
|
|
from sqlalchemy.test.testing import eq_, ne_, assert_raises, assert_raises_message
|
|
from test.orm import _base, _fixtures
|
|
from test.engine import _base as engine_base
|
|
|
|
|
|
_uuids =['1fc614acbb904742a2990f86af6ded95', '23e253786f4d491b9f9d6189dc33de9b', 'fc44910db37e43fd93e9ec8165b885cf',
|
|
'0187a1832b4249e6b48911821d86de58', '778af6ea2fb74a009d8d2f5abe5dc29a', '51a6ce031aff47e4b5f2895c4161f120',
|
|
'7434097cd319401fb9f15fa443ccbbbb', '9bc548a8128e4a85ac18060bc3f4b7d3', '59548715e3c440b7bcb96417d06f7930',
|
|
'd7647c7004734de196885ca2bd73adf8', '70cef121d3ff48d39906b6d1ac77f41a', 'ee37a8a6430c466aa322b8a215a0dd70',
|
|
'782a5f04b4364a53a6fce762f48921c1', 'bef510f2420f4476a7629013ead237f5']
|
|
|
|
def make_uuid():
|
|
"""generate uuids even on Python 2.4 which has no 'uuid'"""
|
|
return _uuids.pop(0)
|
|
|
|
class VersioningTest(_base.MappedTest):
|
|
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table('version_table', metadata,
|
|
Column('id', Integer, primary_key=True,
|
|
test_needs_autoincrement=True),
|
|
Column('version_id', Integer, nullable=False),
|
|
Column('value', String(40), nullable=False))
|
|
|
|
@classmethod
|
|
def setup_classes(cls):
|
|
class Foo(_base.ComparableEntity):
|
|
pass
|
|
|
|
@engines.close_open_connections
|
|
@testing.resolve_artifact_names
|
|
def test_notsane_warning(self):
|
|
# clear the warning module's ignores to force the SAWarning this
|
|
# test relies on to be emitted (it may have already been ignored
|
|
# forever by other VersioningTests)
|
|
try:
|
|
del __warningregistry__
|
|
except NameError:
|
|
pass
|
|
|
|
save = testing.db.dialect.supports_sane_rowcount
|
|
testing.db.dialect.supports_sane_rowcount = False
|
|
try:
|
|
mapper(Foo, version_table,
|
|
version_id_col=version_table.c.version_id)
|
|
|
|
s1 = create_session(autocommit=False)
|
|
f1 = Foo(value='f1')
|
|
f2 = Foo(value='f2')
|
|
s1.add_all((f1, f2))
|
|
s1.commit()
|
|
|
|
f1.value='f1rev2'
|
|
assert_raises(sa.exc.SAWarning, s1.commit)
|
|
finally:
|
|
testing.db.dialect.supports_sane_rowcount = save
|
|
|
|
@testing.emits_warning(r'.*does not support updated rowcount')
|
|
@engines.close_open_connections
|
|
@testing.resolve_artifact_names
|
|
def test_basic(self):
|
|
mapper(Foo, version_table,
|
|
version_id_col=version_table.c.version_id)
|
|
|
|
s1 = create_session(autocommit=False)
|
|
f1 = Foo(value='f1')
|
|
f2 = Foo(value='f2')
|
|
s1.add_all((f1, f2))
|
|
s1.commit()
|
|
|
|
f1.value='f1rev2'
|
|
s1.commit()
|
|
|
|
s2 = create_session(autocommit=False)
|
|
f1_s = s2.query(Foo).get(f1.id)
|
|
f1_s.value='f1rev3'
|
|
s2.commit()
|
|
|
|
f1.value='f1rev3mine'
|
|
|
|
# Only dialects with a sane rowcount can detect the
|
|
# ConcurrentModificationError
|
|
if testing.db.dialect.supports_sane_rowcount:
|
|
assert_raises(sa.orm.exc.ConcurrentModificationError, s1.commit)
|
|
s1.rollback()
|
|
else:
|
|
s1.commit()
|
|
|
|
# new in 0.5 ! dont need to close the session
|
|
f1 = s1.query(Foo).get(f1.id)
|
|
f2 = s1.query(Foo).get(f2.id)
|
|
|
|
f1_s.value='f1rev4'
|
|
s2.commit()
|
|
|
|
s1.delete(f1)
|
|
s1.delete(f2)
|
|
|
|
if testing.db.dialect.supports_sane_rowcount:
|
|
assert_raises(sa.orm.exc.ConcurrentModificationError, s1.commit)
|
|
else:
|
|
s1.commit()
|
|
|
|
@testing.emits_warning(r'.*does not support updated rowcount')
|
|
@engines.close_open_connections
|
|
@testing.resolve_artifact_names
|
|
def test_versioncheck(self):
|
|
"""query.with_lockmode performs a 'version check' on an already loaded instance"""
|
|
|
|
s1 = create_session(autocommit=False)
|
|
|
|
mapper(Foo, version_table, version_id_col=version_table.c.version_id)
|
|
f1s1 = Foo(value='f1 value')
|
|
s1.add(f1s1)
|
|
s1.commit()
|
|
|
|
s2 = create_session(autocommit=False)
|
|
f1s2 = s2.query(Foo).get(f1s1.id)
|
|
f1s2.value='f1 new value'
|
|
s2.commit()
|
|
|
|
# load, version is wrong
|
|
assert_raises(
|
|
sa.orm.exc.ConcurrentModificationError,
|
|
s1.query(Foo).with_lockmode('read').get, f1s1.id
|
|
)
|
|
|
|
# reload it - this expires the old version first
|
|
s1.refresh(f1s1, lockmode='read')
|
|
|
|
# now assert version OK
|
|
s1.query(Foo).with_lockmode('read').get(f1s1.id)
|
|
|
|
# assert brand new load is OK too
|
|
s1.close()
|
|
s1.query(Foo).with_lockmode('read').get(f1s1.id)
|
|
|
|
|
|
@testing.emits_warning(r'.*does not support updated rowcount')
|
|
@engines.close_open_connections
|
|
@testing.requires.update_nowait
|
|
@testing.resolve_artifact_names
|
|
def test_versioncheck_for_update(self):
|
|
"""query.with_lockmode performs a 'version check' on an already loaded instance"""
|
|
|
|
s1 = create_session(autocommit=False)
|
|
|
|
mapper(Foo, version_table, version_id_col=version_table.c.version_id)
|
|
f1s1 = Foo(value='f1 value')
|
|
s1.add(f1s1)
|
|
s1.commit()
|
|
|
|
s2 = create_session(autocommit=False)
|
|
f1s2 = s2.query(Foo).get(f1s1.id)
|
|
s2.refresh(f1s2, lockmode='update')
|
|
f1s2.value='f1 new value'
|
|
|
|
assert_raises(
|
|
exc.DBAPIError,
|
|
s1.refresh, f1s1, lockmode='update_nowait'
|
|
)
|
|
s1.rollback()
|
|
|
|
s2.commit()
|
|
s1.refresh(f1s1, lockmode='update_nowait')
|
|
assert f1s1.version_id == f1s2.version_id
|
|
|
|
@testing.emits_warning(r'.*does not support updated rowcount')
|
|
@engines.close_open_connections
|
|
@testing.resolve_artifact_names
|
|
def test_noversioncheck(self):
|
|
"""test query.with_lockmode works when the mapper has no version id col"""
|
|
s1 = create_session(autocommit=False)
|
|
mapper(Foo, version_table)
|
|
f1s1 = Foo(value="foo", version_id=0)
|
|
s1.add(f1s1)
|
|
s1.commit()
|
|
|
|
s2 = create_session(autocommit=False)
|
|
f1s2 = s2.query(Foo).with_lockmode('read').get(f1s1.id)
|
|
assert f1s2.id == f1s1.id
|
|
assert f1s2.value == f1s1.value
|
|
|
|
|
|
|
|
class RowSwitchTest(_base.MappedTest):
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table('p', metadata,
|
|
Column('id', String(10), primary_key=True),
|
|
Column('version_id', Integer, default=1, nullable=False),
|
|
Column('data', String(50))
|
|
)
|
|
Table('c', metadata,
|
|
Column('id', String(10), ForeignKey('p.id'), primary_key=True),
|
|
Column('version_id', Integer, default=1, nullable=False),
|
|
Column('data', String(50))
|
|
)
|
|
|
|
@classmethod
|
|
def setup_classes(cls):
|
|
class P(_base.ComparableEntity):
|
|
pass
|
|
class C(_base.ComparableEntity):
|
|
pass
|
|
|
|
@classmethod
|
|
@testing.resolve_artifact_names
|
|
def setup_mappers(cls):
|
|
mapper(P, p, version_id_col=p.c.version_id,
|
|
properties={
|
|
'c':relationship(C, uselist=False, cascade='all, delete-orphan')
|
|
})
|
|
mapper(C, c, version_id_col=c.c.version_id)
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_row_switch(self):
|
|
session = sessionmaker()()
|
|
session.add(P(id='P1', data='P version 1'))
|
|
session.commit()
|
|
session.close()
|
|
|
|
p = session.query(P).first()
|
|
session.delete(p)
|
|
session.add(P(id='P1', data="really a row-switch"))
|
|
session.commit()
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_child_row_switch(self):
|
|
assert P.c.property.strategy.use_get
|
|
|
|
session = sessionmaker()()
|
|
session.add(P(id=1, data='P version 1'))
|
|
session.commit()
|
|
session.close()
|
|
|
|
p = session.query(P).first()
|
|
p.c = C(data='child version 1')
|
|
session.commit()
|
|
|
|
p = session.query(P).first()
|
|
p.c = C(data='child row-switch')
|
|
session.commit()
|
|
|
|
class AlternateGeneratorTest(_base.MappedTest):
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table('p', metadata,
|
|
Column('id', String(10), primary_key=True),
|
|
Column('version_id', String(32), nullable=False),
|
|
Column('data', String(50))
|
|
)
|
|
Table('c', metadata,
|
|
Column('id', String(10), ForeignKey('p.id'), primary_key=True),
|
|
Column('version_id', String(32), nullable=False),
|
|
Column('data', String(50))
|
|
)
|
|
|
|
@classmethod
|
|
def setup_classes(cls):
|
|
class P(_base.ComparableEntity):
|
|
pass
|
|
class C(_base.ComparableEntity):
|
|
pass
|
|
|
|
@classmethod
|
|
@testing.resolve_artifact_names
|
|
def setup_mappers(cls):
|
|
mapper(P, p, version_id_col=p.c.version_id,
|
|
version_id_generator=lambda x:make_uuid(),
|
|
properties={
|
|
'c':relationship(C, uselist=False, cascade='all, delete-orphan')
|
|
})
|
|
mapper(C, c, version_id_col=c.c.version_id,
|
|
version_id_generator=lambda x:make_uuid(),
|
|
)
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_row_switch(self):
|
|
session = sessionmaker()()
|
|
session.add(P(id='P1', data='P version 1'))
|
|
session.commit()
|
|
session.close()
|
|
|
|
p = session.query(P).first()
|
|
session.delete(p)
|
|
session.add(P(id='P1', data="really a row-switch"))
|
|
session.commit()
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_child_row_switch_one(self):
|
|
assert P.c.property.strategy.use_get
|
|
|
|
session = sessionmaker()()
|
|
session.add(P(id=1, data='P version 1'))
|
|
session.commit()
|
|
session.close()
|
|
|
|
p = session.query(P).first()
|
|
p.c = C(data='child version 1')
|
|
session.commit()
|
|
|
|
p = session.query(P).first()
|
|
p.c = C(data='child row-switch')
|
|
session.commit()
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_child_row_switch_two(self):
|
|
Session = sessionmaker()
|
|
|
|
sess1 = Session()
|
|
sess1.add(P(id=1, data='P version 1'))
|
|
sess1.commit()
|
|
sess1.close()
|
|
|
|
p1 = sess1.query(P).first()
|
|
|
|
sess2 = Session()
|
|
p2 = sess2.query(P).first()
|
|
|
|
sess1.delete(p1)
|
|
sess1.commit()
|
|
|
|
sess1.add(P(id='P1', data='P version 2'))
|
|
sess1.commit()
|
|
|
|
p2.data = 'P overwritten by concurrent tx'
|
|
assert_raises(
|
|
orm.exc.ConcurrentModificationError,
|
|
sess2.commit
|
|
)
|
|
|
|
|
|
|
|
|
|
|