mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-28 03:26:01 -04:00
Mike Bayer
8b2ad56981
This is a very useful assertion which prevents unused variables
from being set up allows code to be more readable and sometimes
even more efficient. test suites seem to be where the most
problems are and there do not seem to be documentation examples
that are using this, or at least the linter is not taking effect
within rst blocks.
Change-Id: I2b3341d8dd14da34879d8425838e66a4b9f8e27d
(cherry picked from commit 190e0139e8)
130 lines
3.8 KiB
Python
130 lines
3.8 KiB
Python
import sqlalchemy as sa
|
|
from sqlalchemy import ForeignKey
|
|
from sqlalchemy import Integer
|
|
from sqlalchemy import String
|
|
from sqlalchemy import testing
|
|
from sqlalchemy.orm import mapper
|
|
from sqlalchemy.orm import query
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.orm import scoped_session
|
|
from sqlalchemy.testing import assert_raises_message
|
|
from sqlalchemy.testing import eq_
|
|
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy.testing.mock import Mock
|
|
from sqlalchemy.testing.schema import Column
|
|
from sqlalchemy.testing.schema import Table
|
|
|
|
|
|
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() # noqa
|
|
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
|