mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-28 11:35:19 -04:00
Mike Bayer
1a08c41048
Removed a very old keyword argument from :class:`.scoped_session` called ``scope``. This keyword was never documented and was an early attempt at allowing for variable scopes. Change-Id: I9a26498a1a66d1eafb0723e98b527688a60f9d9b Fixes: #3796
119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
from sqlalchemy.testing import assert_raises, assert_raises_message
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import testing
|
|
from sqlalchemy.orm import scoped_session
|
|
from sqlalchemy import Integer, String, ForeignKey
|
|
from sqlalchemy.testing.schema import Table, Column
|
|
from sqlalchemy.orm import mapper, relationship, query
|
|
from sqlalchemy.testing import eq_
|
|
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy.testing.mock import Mock
|
|
|
|
|
|
class _ScopedTest(fixtures.MappedTest):
|
|
"""Adds another lookup bucket to emulate Session globals."""
|
|
|
|
run_setup_mappers = 'once'
|
|
|
|
@classmethod
|
|
def setup_class(cls):
|
|
cls.scoping = _base.adict()
|
|
super(_ScopedTest, cls).setup_class()
|
|
|
|
@classmethod
|
|
def teardown_class(cls):
|
|
cls.scoping.clear()
|
|
super(_ScopedTest, cls).teardown_class()
|
|
|
|
|
|
class ScopedSessionTest(fixtures.MappedTest):
|
|
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table('table1', metadata,
|
|
Column('id', Integer, primary_key=True,
|
|
test_needs_autoincrement=True),
|
|
Column('data', String(30)))
|
|
Table('table2', metadata,
|
|
Column('id', Integer, primary_key=True,
|
|
test_needs_autoincrement=True),
|
|
Column('someid', None, ForeignKey('table1.id')))
|
|
|
|
def test_basic(self):
|
|
table2, table1 = self.tables.table2, self.tables.table1
|
|
|
|
Session = scoped_session(sa.orm.sessionmaker())
|
|
|
|
class CustomQuery(query.Query):
|
|
pass
|
|
|
|
class SomeObject(fixtures.ComparableEntity):
|
|
query = Session.query_property()
|
|
|
|
class SomeOtherObject(fixtures.ComparableEntity):
|
|
query = Session.query_property()
|
|
custom_query = Session.query_property(query_cls=CustomQuery)
|
|
|
|
mapper(SomeObject, table1, properties={
|
|
'options': relationship(SomeOtherObject)})
|
|
mapper(SomeOtherObject, table2)
|
|
|
|
s = SomeObject(id=1, data="hello")
|
|
sso = SomeOtherObject()
|
|
s.options.append(sso)
|
|
Session.add(s)
|
|
Session.commit()
|
|
Session.refresh(sso)
|
|
Session.remove()
|
|
|
|
eq_(SomeObject(id=1, data="hello",
|
|
options=[SomeOtherObject(someid=1)]),
|
|
Session.query(SomeObject).one())
|
|
eq_(SomeObject(id=1, data="hello",
|
|
options=[SomeOtherObject(someid=1)]),
|
|
SomeObject.query.one())
|
|
eq_(SomeOtherObject(someid=1),
|
|
SomeOtherObject.query.filter(
|
|
SomeOtherObject.someid == sso.someid).one())
|
|
assert isinstance(SomeOtherObject.query, query.Query)
|
|
assert not isinstance(SomeOtherObject.query, CustomQuery)
|
|
assert isinstance(SomeOtherObject.custom_query, query.Query)
|
|
|
|
def test_config_errors(self):
|
|
Session = scoped_session(sa.orm.sessionmaker())
|
|
|
|
s = Session()
|
|
assert_raises_message(
|
|
sa.exc.InvalidRequestError,
|
|
"Scoped session is already present",
|
|
Session, bind=testing.db
|
|
)
|
|
|
|
assert_raises_message(
|
|
sa.exc.SAWarning,
|
|
"At least one scoped session is already present. ",
|
|
Session.configure, bind=testing.db
|
|
)
|
|
|
|
def test_call_with_kwargs(self):
|
|
mock_scope_func = Mock()
|
|
SessionMaker = sa.orm.sessionmaker()
|
|
Session = scoped_session(sa.orm.sessionmaker(), mock_scope_func)
|
|
|
|
s0 = SessionMaker()
|
|
assert s0.autocommit == False
|
|
|
|
mock_scope_func.return_value = 0
|
|
s1 = Session()
|
|
assert s1.autocommit == False
|
|
|
|
assert_raises_message(
|
|
sa.exc.InvalidRequestError,
|
|
"Scoped session is already present",
|
|
Session, autocommit=True
|
|
)
|
|
|
|
mock_scope_func.return_value = 1
|
|
s2 = Session(autocommit=True)
|
|
assert s2.autocommit == True
|