581e4f2d88
git-svn-id: file:///Users/arjan/backup/gaphor/gaphor/trunk@2156 a8418922-720d-0410-834f-a69b97ada669
149 lines
4.7 KiB
Python
149 lines
4.7 KiB
Python
# 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()
|
|
|