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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
# vim:sw=4:et:ai
import unittest
from weakref import ref as wref
from sys import getrefcount
from gaphor import UML
from gaphor.diagram import items
from gaphor.application import Application
Application.shutdown()
factory = UML.ElementFactory()
factory.init(Application)
class TestDiagramItems(unittest.TestCase):
def gc_collect(self):
import gc
for x in range(6): gc.collect()
def _testTemplate1(self, uml_class, item_class):
m = factory.create(UML.Package)
d = factory.create(UML.Diagram)
d.package = m
p1 = factory.create(uml_class)
p1.package = m
pi1 = d.create(item_class)
pi1.subject = p1
w_m = wref(m)
w_d = wref(d)
w_p1 = wref(p1)
w_pi1 = wref(pi1)
del m, d, p1, pi1
factory.flush()
self.gc_collect()
self.assertEquals(w_m(), None)
self.failUnless(w_d() is None)
self.failUnless(w_p1() is None)
self.failUnless(w_pi1() is None)
def _testTemplate2(self, uml_class, item_class):
m = factory.create(UML.Package)
d = factory.create(UML.Diagram)
d.package = m
p1 = factory.create(uml_class)
p1.package = m
pi1 = d.create(item_class)
pi1.subject = p1
rc_p1 = getrefcount(p1)
rc_pi1 = getrefcount(pi1)
p2 = factory.create(uml_class)
p2.package = m
pi2 = d.create(item_class)
pi2.subject = p2
self.assertEquals(getrefcount(p1), rc_p1)
self.assertEquals(getrefcount(pi1), rc_pi1)
d.canvas.update_now()
self.assertEquals(getrefcount(p1), rc_p1)
self.assertEquals(getrefcount(pi1), rc_pi1)
w_m = wref(m)
w_d = wref(d)
w_p1 = wref(p1)
w_pi1 = wref(pi1)
w_p2 = wref(p2)
w_pi2 = wref(pi2)
self.assertEquals(getrefcount(p1), rc_p1)
self.assertEquals(getrefcount(pi1), rc_pi1)
del m, d, p1, pi1, p2, pi2
self.assertEquals(getrefcount(w_p1()), rc_p1 - 1)
self.assertEquals(getrefcount(w_pi1()), rc_pi1 - 1)
factory.flush()
self.gc_collect()
self.failUnless(w_p1() is None, getrefcount(w_p1()))
self.failUnless(w_pi1() is None, getrefcount(w_p1()))
self.failUnless(w_p2() is None, getrefcount(w_p1()))
self.failUnless(w_pi2() is None, getrefcount(w_p1()))
self.failUnless(w_m() is None, getrefcount(w_p1()))
self.failUnless(w_d() is None, getrefcount(w_p1()))
def testPackage1(self):
self._testTemplate1(UML.Package, items.PackageItem)
def testPackage2(self):
self._testTemplate2(UML.Package, items.PackageItem)
def testClass1(self):
self._testTemplate1(UML.Class, items.ClassItem)
def testClass2(self):
self._testTemplate2(UML.Class, items.ClassItem)
def testDependency1(self):
self._testTemplate1(UML.Dependency, items.DependencyItem)
def testDependency2(self):
self._testTemplate2(UML.Dependency, items.DependencyItem)
def _testTemplate3(self, uml_class, item_class):
import gtk
import gaphor.ui as ui
m = factory.create(UML.Package)
d = factory.create(UML.Diagram)
d.package = m
p1 = factory.create(uml_class)
p1.package = m
pi1 = d.create(item_class)
pi1.subject = p1
rc_p1 = getrefcount(p1)
rc_pi1 = getrefcount(pi1)
p2 = factory.create(uml_class)
p2.package = m
pi2 = d.create(item_class)
pi2.subject = p2
self.assertEquals(getrefcount(p1), rc_p1)
self.assertEquals(getrefcount(pi1), rc_pi1)
d.canvas.update_now()
win = gtk.Window()
view = ui.DiagramView(d)
win.add(view)
win.show_all()
win.destroy()
self.assertEquals(getrefcount(p1), rc_p1)
self.assertEquals(getrefcount(pi1), rc_pi1)
w_m = wref(m)
w_d = wref(d)
w_p1 = wref(p1)
w_pi1 = wref(pi1)
w_p2 = wref(p2)
w_pi2 = wref(pi2)
self.assertEquals(getrefcount(p1), rc_p1)
self.assertEquals(getrefcount(pi1), rc_pi1)
del m, d, p1, pi1, p2, pi2
del win, view
self.assertEquals(getrefcount(w_p1()), rc_p1 - 1)
self.assertEquals(getrefcount(w_pi1()), rc_pi1 - 1)
factory.flush()
self.gc_collect()
self.failUnless(w_p1() is None, getrefcount(w_p1()))
self.failUnless(w_pi1() is None, getrefcount(w_p1()))
self.failUnless(w_p2() is None, getrefcount(w_p1()))
self.failUnless(w_pi2() is None, getrefcount(w_p1()))
self.failUnless(w_m() is None, getrefcount(w_p1()))
self.failUnless(w_d() is None, getrefcount(w_p1()))
if __name__ == '__main__':
unittest.main()
|
