diff --git a/EditorialModel/test/test_types.py b/EditorialModel/test/test_types.py index 2334fab..43302d4 100644 --- a/EditorialModel/test/test_types.py +++ b/EditorialModel/test/test_types.py @@ -8,7 +8,7 @@ import unittest from EditorialModel.types import EmType from EditorialModel.classes import EmClass -from EditorialModel.classtypes import EmClassType +from EditorialModel.classtypes import EmClassType, EmNature from EditorialModel.components import EmComponent from EditorialModel.fieldgroups import EmFieldGroup from EditorialModel.fieldtypes import * @@ -40,14 +40,14 @@ class TypeTestCase(TestCase): self.emclass1 = EmClass.create("entity1", EmClassType.entity) self.emclass2 = EmClass.create("entity2", EmClassType.entity) - - self.emtype = EmType.create('type1',self.emclass2) - self.emfieldgroup = EmFieldGroup.create('fieldgroup1',self.emclass1) + self.emtype = EmType.create(name='type1', em_class=self.emclass2) + self.emtype2 = EmType.create(name='type2', em_class=self.emclass2) + self.emtype3 = EmType.create(name='type3', em_class=self.emclass2) + self.emfieldgroup = EmFieldGroup.create(name='fieldgroup1', em_class=self.emclass1) self.emfieldtype = get_field_type('integer') self.emfield = EmField.create(name='field1', fieldgroup=self.emfieldgroup, fieldtype=self.emfieldtype, rel_to_type_id=self.emtype.uid) restoreDbState(TEST_TYPE_DBNAME) - pass class TestSelectField(TypeTestCase): def testSelectField(self): @@ -58,3 +58,14 @@ class TestSelectField(TypeTestCase): self.emtype.unselect_field(self.emfield) self.assertFalse(Em_Field_Type(self.emtype.uid, self.emfield.uid).exists()) +class TestLinkedTypes(TypeTestCase): + def testLinkedtypes(self): + self.emtype.add_superior(self.emtype2, EmNature.PARENT) + self.emtype3.add_superior(self.emtype, EmNature.PARENT) + + linked_types = self.emtype.linked_types() + + self.assertEqual(len(linked_types),2) + self.assertNotIn(self.emtype,linked_types) + self.assertIn(self.emtype2, linked_types) + self.assertIn(self.emtype3, linked_types)