mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-06 08:56:51 -04:00
91 lines
1.5 KiB
Python
91 lines
1.5 KiB
Python
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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|