1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
"""Generic mapping to Select statements"""
import testenv; testenv.configure_for_tests()
from testlib import sa, testing
from testlib.sa import Table, Column, String, Integer, select
from testlib.sa.orm import mapper, create_session
from testlib.testing import eq_
from orm import _base
# TODO: more tests mapping to selects
class SelectableNoFromsTest(_base.MappedTest):
def define_tables(self, metadata):
Table('common', metadata,
Column('id', Integer, primary_key=True),
Column('data', Integer),
Column('extra', String(45)))
def setup_classes(self):
class Subset(_base.ComparableEntity):
pass
@testing.resolve_artifact_names
def test_no_tables(self):
selectable = select(["x", "y", "z"])
self.assertRaisesMessage(sa.exc.InvalidRequestError,
"Could not find any Table objects",
mapper, Subset, selectable)
@testing.emits_warning('.*creating an Alias.*')
@testing.resolve_artifact_names
def test_basic(self):
subset_select = select([common.c.id, common.c.data])
subset_mapper = mapper(Subset, subset_select)
sess = create_session(bind=testing.db)
sess.add(Subset(data=1))
sess.flush()
sess.clear()
eq_(sess.query(Subset).all(), [Subset(data=1)])
eq_(sess.query(Subset).filter(Subset.data==1).one(), Subset(data=1))
eq_(sess.query(Subset).filter(Subset.data!=1).first(), None)
subset_select = sa.orm.class_mapper(Subset).mapped_table
eq_(sess.query(Subset).filter(subset_select.c.data==1).one(),
Subset(data=1))
if __name__ == '__main__':
testenv.main()
|
