git-svn-id: file:///Users/arjan/backup/gaphor/trunk/gaphor@376 a8418922-720d-0410-834f-a69b97ada669
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
# vim:sw=4:et:ai
|
|
|
|
import unittest
|
|
|
|
import gaphor.UML as UML
|
|
|
|
class TestUML2(unittest.TestCase):
|
|
|
|
def test1(self):
|
|
factory = UML.ElementFactory()
|
|
c = factory.create(UML.Class)
|
|
p = factory.create(UML.Package)
|
|
c.package = p
|
|
self.assertEquals(c.package, p)
|
|
self.assertEquals(c.namespace, p)
|
|
self.failUnless(c in p.ownedElement)
|
|
|
|
def __on_owned_member(self, pspec, name):
|
|
self.owned_member_called = True
|
|
|
|
def testOwnedMember(self):
|
|
factory = UML.ElementFactory()
|
|
c = factory.create(UML.Class)
|
|
p = factory.create(UML.Package)
|
|
self.owned_member_called = False
|
|
p.connect('ownedMember', self.__on_owned_member)
|
|
c.package = p
|
|
self.assertEquals(self.owned_member_called, True)
|
|
|
|
def testOwnedMember_Unlink(self):
|
|
factory = UML.ElementFactory()
|
|
c = factory.create(UML.Class)
|
|
p = factory.create(UML.Package)
|
|
self.owned_member_called = False
|
|
p.connect('ownedMember', self.__on_owned_member)
|
|
c.package = p
|
|
self.assertEquals(self.owned_member_called, True)
|
|
self.owned_member_called = False
|
|
c.unlink()
|
|
self.assertEquals(self.owned_member_called, True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|