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
|
import testbase
from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.sessioncontext import SessionContext
from testlib import *
class Jack(object):
def __repr__(self):
return "{Jack %s - %s}" % (self.id, self.number)
def __init__(self, room=None, subroom=None, number=None, status=None):
self.id = None
self.room = room
self.subroom = subroom
self.number = number
self.status = status
class Port(object):
def __repr__(self):
return "{Port %s - %s}" % (self.id, self.name)
def __init__(self, name=None, description=None):
self.id=None
self.name=name
self.description = description
class O2OTest(AssertMixin):
def setUpAll(self):
global jack, port, metadata, ctx
metadata = MetaData(testbase.db)
ctx = SessionContext(create_session)
jack = Table('jack', metadata,
Column('id', Integer, primary_key=True),
#Column('room_id', Integer, ForeignKey("room.id")),
Column('number', String(50)),
Column('status', String(20)),
Column('subroom', String(5)),
)
port = Table('port', metadata,
Column('id', Integer, primary_key=True),
#Column('device_id', Integer, ForeignKey("device.id")),
Column('name', String(30)),
Column('description', String(100)),
Column('jack_id', Integer, ForeignKey("jack.id")),
)
metadata.create_all()
def setUp(self):
pass
def tearDown(self):
clear_mappers()
def tearDownAll(self):
metadata.drop_all()
def test1(self):
mapper(Port, port, extension=ctx.mapper_extension)
mapper(Jack, jack, order_by=[jack.c.number],properties = {
'port': relation(Port, backref='jack', uselist=False, lazy=True),
}, extension=ctx.mapper_extension)
j=Jack(number='101')
p=Port(name='fa0/1')
j.port=p
ctx.current.flush()
jid = j.id
pid = p.id
j=ctx.current.query(Jack).get(jid)
p=ctx.current.query(Port).get(pid)
print p.jack
assert p.jack is not None
assert p.jack is j
assert j.port is not None
p.jack=None
assert j.port is None #works
ctx.current.clear()
j=ctx.current.query(Jack).get(jid)
p=ctx.current.query(Port).get(pid)
j.port=None
self.assert_(p.jack is None)
ctx.current.flush()
ctx.current.delete(j)
ctx.current.flush()
if __name__ == "__main__":
testbase.main()
|