blob: 39bcf3e536ea829410cd3add0262b48f258368e9 (
plain)
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
|
from sqlalchemy import *
from sqlalchemy.ext.proxy import AutoConnectEngine
from testbase import PersistTest
import testbase
import os
#
# Define an engine, table and mapper at the module level, to show that the
# table and mapper can be used with different real engines in multiple threads
#
module_engine = AutoConnectEngine( testbase.db_uri )
users = Table('users', module_engine,
Column('user_id', Integer, primary_key=True),
Column('user_name', String(16)),
Column('password', String(20))
)
class User(object):
pass
class AutoConnectEngineTest1(PersistTest):
def setUp(self):
clear_mappers()
objectstore.clear()
def test_engine_connect(self):
users.create()
assign_mapper(User, users)
try:
trans = objectstore.begin()
user = User()
user.user_name='fred'
user.password='*'
trans.commit()
# select
sqluser = User.select_by(user_name='fred')[0]
assert sqluser.user_name == 'fred'
# modify
sqluser.user_name = 'fred jones'
# commit - saves everything that changed
objectstore.commit()
allusers = [ user.user_name for user in User.select() ]
assert allusers == [ 'fred jones' ]
finally:
users.drop()
if __name__ == "__main__":
testbase.main()
|