554798ce47
git-svn-id: file:///Users/arjan/backup/gaphor/trunk/gaphor@219 a8418922-720d-0410-834f-a69b97ada669
107 lines
2.5 KiB
Python
107 lines
2.5 KiB
Python
# vim:sw=4:et:ai
|
|
|
|
import gaphor.UML as UML
|
|
from gaphor.ui.namespace import NamespaceModel
|
|
|
|
factory = UML.ElementFactory()
|
|
|
|
m = factory.create(UML.Package)
|
|
m.name = 'm'
|
|
a = factory.create(UML.Package)
|
|
a.name = 'a'
|
|
a.package = m
|
|
assert a.package is m
|
|
assert a in m.ownedMember
|
|
assert a.namespace is m
|
|
|
|
b = factory.create(UML.Package)
|
|
b.name = 'b'
|
|
b.package = a
|
|
assert b in a.ownedMember
|
|
assert b.namespace is a
|
|
|
|
c = factory.create(UML.Class)
|
|
c.name = 'c'
|
|
c.package = b
|
|
d = factory.create(UML.Class)
|
|
d.name = 'd'
|
|
d.package = a
|
|
e = factory.create(UML.Class)
|
|
e.name = 'e'
|
|
e.package = b
|
|
|
|
assert c in b.ownedMember
|
|
assert c.namespace is b
|
|
assert d in a.ownedMember
|
|
assert d.namespace is a
|
|
assert e in b.ownedMember
|
|
assert e.namespace is b
|
|
|
|
|
|
ns = NamespaceModel(factory)
|
|
# We have a model loaded. Use it!
|
|
factory.notify_model()
|
|
|
|
print '---'
|
|
ns.dump()
|
|
assert ns.path_from_element(m) == (0,)
|
|
assert ns.path_from_element(a) == (0, 0)
|
|
assert ns.path_from_element(b) == (0, 0, 1)
|
|
assert ns.path_from_element(c) == (0, 0, 1, 0)
|
|
assert ns.path_from_element(d) == (0, 0, 0)
|
|
assert ns.path_from_element(e) == (0, 0, 1, 1)
|
|
|
|
print '--- del.b.ownedClassifier[c]'
|
|
del b.ownedClassifier[c]
|
|
ns.dump()
|
|
assert ns.path_from_element(m) == (0,)
|
|
assert ns.path_from_element(a) == (0, 0)
|
|
assert ns.path_from_element(b) == (0, 0, 1)
|
|
assert ns.path_from_element(d) == (0, 0, 0)
|
|
assert ns.path_from_element(e) == (0, 0, 1, 0)
|
|
try:
|
|
ns.path_from_element(c)
|
|
except AttributeError:
|
|
pass # Yes, should raise an exception
|
|
else:
|
|
assert ns.path_from_element(c) is not None
|
|
|
|
print '--- c.package = a'
|
|
c.package = a
|
|
ns.dump()
|
|
assert ns.path_from_element(m) == (0,)
|
|
assert ns.path_from_element(a) == (0, 0)
|
|
assert ns.path_from_element(b) == (0, 0, 1)
|
|
assert ns.path_from_element(c) == (0, 0, 2)
|
|
assert ns.path_from_element(d) == (0, 0, 0)
|
|
assert ns.path_from_element(e) == (0, 0, 1, 0)
|
|
|
|
print '--- b.package = m'
|
|
b.package = m
|
|
ns.dump()
|
|
assert ns.path_from_element(m) == (0,)
|
|
assert ns.path_from_element(a) == (0, 0)
|
|
assert ns.path_from_element(b) == (0, 1)
|
|
assert ns.path_from_element(c) == (0, 0, 1)
|
|
assert ns.path_from_element(d) == (0, 0, 0)
|
|
assert ns.path_from_element(e) == (0, 1, 0)
|
|
|
|
print '--- e.unlink()'
|
|
e.unlink()
|
|
ns.dump()
|
|
|
|
print '--- a.unlink()'
|
|
def on_unlink(name, element):
|
|
print 'unlink: %s' % element.name
|
|
a.connect('__unlink__', on_unlink, a)
|
|
b.connect('__unlink__', on_unlink, b)
|
|
c.connect('__unlink__', on_unlink, c)
|
|
d.connect('__unlink__', on_unlink, d)
|
|
|
|
a.unlink()
|
|
ns.dump()
|
|
print '--- TODO: e.relink()'
|
|
|
|
print UML.Class.package
|
|
print UML.Package.ownedClassifier
|