def test_class_defined_bundle_change(self):
cl1 = CClass(self.mcl, "Class1", bundles=self.b1)
cl2 = CClass(self.mcl, "Class2", bundles=self.b1)
cl3 = CClass(self.mcl, "Class3", bundles=self.b1)
mcl = CMetaclass("MCL", bundles=self.b1)
b = CBundle()
cl2.bundles = b
cl3.bundles = None
self.mcl.bundles = b
eq_(set(self.b1.elements), {mcl, cl1})
eq_(set(self.b1.get_elements(type=CClass)), {cl1})
eq_(set(b.elements), {cl2, self.mcl})
eq_(set(b.get_elements(type=CClass)), {cl2})
eq_(cl1.bundles, [self.b1])
eq_(cl2.bundles, [b])
eq_(cl3.bundles, [])
def test_stereotype_defined_bundle_change(self):
s1 = CStereotype("s1", bundles=self.b1)
s2 = CStereotype("s2", bundles=self.b1)
s3 = CStereotype("s3", bundles=self.b1)
cl1 = CClass(self.mcl, "C1", bundles=self.b1)
cl2 = CClass(self.mcl, "C2", bundles=self.b1)
b = CBundle()
s2.bundles = b
s3.bundles = None
cl2.bundles = b
eq_(set(self.b1.elements), {cl1, s1})
eq_(set(self.b1.get_elements(type=CStereotype)), {s1})
eq_(set(b.elements), {s2, cl2})
eq_(set(b.get_elements(type=CStereotype)), {s2})
eq_(s1.bundles, [self.b1])
eq_(s2.bundles, [b])
eq_(s3.bundles, [])
def test_metaclass_defined_bundle_change(self):
m1 = CMetaclass("M1", bundles=self.b1)
m2 = CMetaclass("M2", bundles=self.b1)
m3 = CMetaclass("M3", bundles=self.b1)
cl1 = CClass(m1, "C1", bundles=self.b1)
cl2 = CClass(m1, "C2", bundles=self.b1)
b = CBundle()
m2.bundles = b
m3.bundles = None
cl2.bundles = b
eq_(set(self.b1.elements), {cl1, m1})
eq_(set(self.b1.get_elements(type=CMetaclass)), {m1})
eq_(set(b.elements), {m2, cl2})
eq_(set(b.get_elements(type=CMetaclass)), {m2})
eq_(m1.bundles, [self.b1])
eq_(m2.bundles, [b])
eq_(m3.bundles, [])
