def test_copy_extensions(self):
target = ClassFactory.create(extensions=ExtensionFactory.list(1))
source = ClassFactory.create(extensions=ExtensionFactory.list(2))
link_extension = ExtensionFactory.create()
link_extension.restrictions.max_occurs = 2
ClassUtils.copy_extensions(source, target, link_extension)
self.assertEqual(3, len(target.extensions))
self.assertEqual(2, target.extensions[1].restrictions.max_occurs)
self.assertEqual(2, target.extensions[2].restrictions.max_occurs)
def test_process_enum_extension_with_complex_source(self):
source = ClassFactory.create(
tag=Tag.COMPLEX_TYPE,
attrs=[
AttrFactory.create(tag=Tag.ATTRIBUTE),
AttrFactory.create(tag=Tag.RESTRICTION),
],
extensions=ExtensionFactory.list(2),
status=Status.PROCESSED,
)
target = ClassFactory.enumeration(1)
target.attrs[0].default = "Yes"
extension = ExtensionFactory.reference(source.qname)
target.extensions.append(extension)
expected = target.clone()
self.processor.container.add(source)
self.processor.container.add(target)
self.processor.process_dependency_extension(target, extension)
expected.attrs = [attr.clone() for attr in source.attrs]
expected.extensions = [ext.clone() for ext in source.extensions]
expected.attrs[1].default = "Yes"
expected.attrs[1].fixed = True
self.assertEqual(expected, target)
self.assertIsNone(target.attrs[0].default)
self.assertFalse(target.attrs[0].fixed)
self.assertEqual("Yes", target.attrs[1].default)
self.assertTrue(target.attrs[1].fixed)
def test_class_references(self):
target = ClassFactory.elements(
2,
inner=ClassFactory.list(2, attrs=AttrFactory.list(1)),
extensions=ExtensionFactory.list(1),
)
actual = ClassAnalyzer.class_references(target)
# +1 target
# +2 attrs
# +2 attr types
# +1 extension
# +1 extension type
# +2 inner classes
# +2 inner classes attrs
# +2 inner classes attr types
self.assertEqual(13, len(actual))
self.assertEqual(id(target), actual[0])