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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
# coding: utf-8
"""verrrrry basic unicode column name testing"""
import testbase
from sqlalchemy import *
from sqlalchemy.orm import mapper, relation, create_session, eagerload
from testlib import *
from testlib.engines import utf8_engine
class UnicodeSchemaTest(PersistTest):
def setUpAll(self):
global unicode_bind, metadata, t1, t2
unicode_bind = utf8_engine()
metadata = MetaData(unicode_bind)
t1 = Table('unitable1', metadata,
Column(u'méil', Integer, primary_key=True),
Column(u'\u6e2c\u8a66', Integer),
)
t2 = Table(u'Unitéble2', metadata,
Column(u'méil', Integer, primary_key=True, key="a"),
Column(u'\u6e2c\u8a66', Integer, ForeignKey(u'unitable1.méil'), key="b"),
)
metadata.create_all()
def tearDown(self):
if metadata.tables:
t2.delete().execute()
t1.delete().execute()
def tearDownAll(self):
global unicode_bind
metadata.drop_all()
del unicode_bind
def test_insert(self):
t1.insert().execute({u'méil':1, u'\u6e2c\u8a66':5})
t2.insert().execute({'a':1, 'b':1})
assert t1.select().execute().fetchall() == [(1, 5)]
assert t2.select().execute().fetchall() == [(1, 1)]
def test_reflect(self):
t1.insert().execute({u'méil':2, u'\u6e2c\u8a66':7})
t2.insert().execute({'a':2, 'b':2})
meta = MetaData(unicode_bind)
tt1 = Table(t1.name, meta, autoload=True)
tt2 = Table(t2.name, meta, autoload=True)
tt1.insert().execute({u'méil':1, u'\u6e2c\u8a66':5})
tt2.insert().execute({u'méil':1, u'\u6e2c\u8a66':1})
self.assert_(tt1.select(order_by=desc(u'méil')).execute().fetchall() ==
[(2, 7), (1, 5)])
self.assert_(tt2.select(order_by=desc(u'méil')).execute().fetchall() ==
[(2, 2), (1, 1)])
meta.drop_all()
metadata.create_all()
def test_mapping(self):
# TODO: this test should be moved to the ORM tests, tests should be
# added to this module testing SQL syntax and joins, etc.
class A(object):pass
class B(object):pass
mapper(A, t1, properties={
't2s':relation(B),
'a':t1.c[u'méil'],
'b':t1.c[u'\u6e2c\u8a66']
})
mapper(B, t2)
sess = create_session()
a1 = A()
b1 = B()
a1.t2s.append(b1)
sess.save(a1)
sess.flush()
sess.clear()
new_a1 = sess.query(A).filter(t1.c[u'méil'] == a1.a).one()
assert new_a1.a == a1.a
assert new_a1.t2s[0].a == b1.a
sess.clear()
new_a1 = sess.query(A).options(eagerload('t2s')).filter(t1.c[u'méil'] == a1.a).one()
assert new_a1.a == a1.a
assert new_a1.t2s[0].a == b1.a
sess.clear()
new_a1 = sess.query(A).filter(A.a == a1.a).one()
assert new_a1.a == a1.a
assert new_a1.t2s[0].a == b1.a
sess.clear()
if __name__ == '__main__':
testbase.main()
|
