def get(self, name):
from_namespace = self._get_from_namespace(name)
if from_namespace:
return from_namespace
if name not in self.types:
return None
el, parsed = self.types[name]
if not parsed:
# Create an "empty" element to avoid circular dependency
# If it exists in self.types, we'll return it in the subsequent
# calls of 'element_parser', after we needs to fill this element with
# proper data
ref = ElementType(name)
self.types[name][1] = ref
parse_result = self.element_parser(el)
if parse_result:
_, parse_element = parse_result
ref.fill_with(parse_element)
parsed = ref
return parsed
def test_element_get_child_by_name_must_returns_expected_child():
element_type = ElementType('xsd:string')
name_elem = Element(name='Name', element_type=element_type)
surname_elem = Element(name='Surname', element_type=element_type)
f = ElementType(name='ModelType', children=[name_elem, surname_elem])
assert f.get_child_by_name('Surname') == surname_elem
assert f.get_child_by_name('SurnameNotFound') is None
def test_element_type_with_in_operator():
elements = [ElementType(name='xsd:string')]
f = ElementType(name='xsd:string')
f2 = ElementType(name='xsd:token')
assert f in elements
assert f2 not in elements
def test_element_type_with_same_elements_must_be_equal():
element_type = ElementType('xsd:string')
children = [Element(name='Name', element_type=element_type)]
f = ElementType(name='ModelType', children=children)
f2 = ElementType(name='ModelType', children=children)
assert f == f2
assert hash(f) == hash(f2)
def test_element_type_change_children_must_recalculate_hash():
element_type = ElementType('xsd:string')
children = [Element(name='Name', element_type=element_type)]
f = ElementType(name='ModelType', children=children)
current_hash = hash(f)
owner = Element(name='Owner', element_type=element_type)
f.add_child(owner)
assert hash(f) != current_hash
def test_element_type_same_children_in_different_order_must_have_different_hash(
):
element_type = ElementType('xsd:string')
child1 = Element(name='Name', element_type=element_type)
child2 = Element(name='Owner', element_type=element_type)
f = ElementType(name='ModelType', children=[child1, child2])
f2 = ElementType(name='ModelType', children=[child2, child1])
assert f != f2
assert hash(f) != hash(f2)
def test_element_type_add_child_must_add_item_properly():
element_type = ElementType('xsd:string')
children = [Element(name='Name', element_type=element_type)]
f = ElementType(name='ModelType', children=children)
current_hash = hash(f)
code = Element(name='Code', element_type=element_type)
f.add_child(code)
assert hash(f) != current_hash
assert f.children[0] == code
def test_element_children_original_order_must_keep_order():
element_type = ElementType('xsd:string')
# A not alphabetical order...
children = [
Element(name='Surname', element_type=element_type),
Element(name='Name', element_type=element_type),
Element(name='City', element_type=element_type),
]
f = ElementType(name='ModelType', children=children)
assert f.children != children
assert f.children_original_order == children
def test_similarity_must_return_expected_ratio_considering_annotation():
f = ElementType(name='ModelType')
f.annotation = 'A simple documentation for type'
f2 = ElementType(name='AnotherElementType')
f2.annotation = 'A simple documentation for type'
assert f2.similarity(f) > 0.8
# With a small change in documentation
f2.annotation = 'A simple documentation'
assert f2.similarity(f) > 0.7
def test_element_to_lxml_element_with_children_must_use_original_order_to_add_it(
string_type, lxml_root):
children = [
Element('Name', element_type=string_type),
Element('Surname', element_type=string_type),
Element('Owner', element_type=string_type),
Element('Country', element_type=string_type),
]
el_type = ElementType('EmployeeType', children=children)
el = Element('Employee', element_type=el_type)
el.to_lxml_element(lxml_root)
result = et.tostring(lxml_root).decode()
# Same order that was added to element type
expected = (
'<xsl:stylesheet xmlns:xsl="http://www.w3.org/1999/XSL/Transform" version="1.0">'
'<xsl:element name="Employee">'
'<xsl:element name="Name">autogenerated</xsl:element>'
'<xsl:element name="Surname">autogenerated</xsl:element>'
'<xsl:element name="Owner">autogenerated</xsl:element>'
'<xsl:element name="Country">autogenerated</xsl:element>'
'</xsl:element>'
'</xsl:stylesheet>')
assert result == expected
def get_or_create(self, name):
parsed = self.get(name)
if not parsed:
parsed = ElementType(name)
self.types[name] = [None, parsed]
return parsed
def test_element_similarity_with_equal_elements_must_return_as_expected():
element_type = ElementType('xsd:string')
name_elem = Element(name='Name', element_type=element_type)
surname_elem = Element(name='Surname', element_type=element_type)
first = ElementType(name='Contract', children=[name_elem, surname_elem])
first.annotation = 'foo'
second = ElementType(name='Contract', children=[name_elem, surname_elem])
second.annotation = 'foo'
assert first.similarity(second) == 1.0
def test_element_children_original_order_must_keep_order_when_added_child():
element_type = ElementType('xsd:string')
# A not alphabetical order...
children = [
Element(name='Surname', element_type=element_type),
Element(name='Name', element_type=element_type),
Element(name='City', element_type=element_type),
]
f = ElementType(name='ModelType', children=children)
nick = Element(name='Nickname', element_type=element_type)
f.add_child(nick)
expected_children = children + [nick]
assert f.children != expected_children
assert f.children_original_order == expected_children
def test_element_fill_with_must_copy_all_data_from_other_element():
element_type = ElementType('xsd:string')
name_elem = Element(name='Name', element_type=element_type)
surname_elem = Element(name='Surname', element_type=element_type)
f = ElementType(name='ModelType', children=[name_elem, surname_elem])
f.annotation = 'A simple documentation for type'
f2 = ElementType('')
f2.fill_with(f)
assert f2.name == f.name
assert f2.children == f.children
assert f2.annotation == 'A simple documentation for type'
assert f2.children_original_order == f.children_original_order
def test_element_remove_child_must_remove_properly():
element_type = ElementType('xsd:string')
name_elem = Element(name='Name', element_type=element_type)
surname_elem = Element(name='Surname', element_type=element_type)
f = ElementType(name='ModelType', children=[name_elem, surname_elem])
# Removing surname
f.remove_child(surname_elem)
assert f.children == [name_elem]
# It's hash must be changed
f_removed = ElementType(name='ModelType', children=[name_elem])
assert f_removed == f
def test_similarity_must_return_expected_ratio_considering_children():
element_type = ElementType('xsd:string')
name_elem = Element(name='Name', element_type=element_type)
surname_elem = Element(name='Surname', element_type=element_type)
f = ElementType(name='ContactType', children=[name_elem, surname_elem])
other_name_elem = Element(name='Name', element_type=element_type)
f2 = ElementType(name='OtherContactType',
children=[other_name_elem, surname_elem])
assert f2.similarity(f) > 0.9
other_name_elem.name = 'FirstName'
assert f2.similarity(f) > 0.8
def string_type():
return ElementType('xsd:string')
def complex_type(string_type):
return ElementType('EmployeeType',
children=[Element('Name', element_type=string_type)])
def test_element_type_must_have_same_hash_of_same_type_name():
f = ElementType(name='xsd:boolean')
f2 = ElementType(name='xsd:boolean')
assert hash(f) == hash(f2)
def test_create_element_type():
f = ElementType(name='xsd:boolean')
assert f.name == 'xsd:boolean'
assert f.children == []
assert f.children_original_order == []
assert str(f) == 'xsd:boolean'