def testPO(self):
class SKU(StringPattern):
pattern = '\\d{3}-[A-Z]{2}'
class Comment(String):
xmltagname = 'comment'
class Quantity(Int):
pass
class USAddress(Object):
country = String.T(
default='US', optional=True, xmlstyle='attribute')
name = String.T()
street = String.T()
city = String.T()
state = String.T()
zip = Float.T()
class Item(Object):
part_num = SKU.T(xmlstyle='attribute')
product_name = String.T()
quantity = Quantity.T()
us_price = Float.T(xmltagname='USPrice')
comment = Comment.T(optional=True)
ship_date = DateTimestamp.T(optional=True)
class Items(Object):
item_list = List.T(Item.T())
class PurchaseOrderType(Object):
order_date = DateTimestamp.T(optional=True, xmlstyle='attribute')
ship_to = USAddress.T()
bill_to = USAddress.T()
comment = Comment.T(optional=True)
items = Items.T()
class PurchaseOrder(PurchaseOrderType):
xmltagname = 'purchaseOrder'
xml = '''<?xml version="1.0"?>
<purchaseOrder orderDate="1999-10-20">
<shipTo country="US">
<name>Alice Smith</name>
<street>123 Maple Street</street>
<city>Mill Valley</city>
<state>CA</state>
<zip>90952</zip>
</shipTo>
<billTo country="US">
<name>Robert Smith</name>
<street>8 Oak Avenue</street>
<city>Old Town</city>
<state>PA</state>
<zip>95819</zip>
</billTo>
<comment>Hurry, my lawn is going wild</comment>
<items>
<item partNum="872-AA">
<productName>Lawnmower</productName>
<quantity>1</quantity>
<USPrice>148.95</USPrice>
<comment>Confirm this is electric</comment>
</item>
<item partNum="926-AA">
<productName>Baby Monitor</productName>
<quantity>1</quantity>
<USPrice>39.98</USPrice>
<shipDate>1999-05-21</shipDate>
</item>
</items>
</purchaseOrder>
'''
po1 = load_xml_string(xml)
po2 = load_xml_string(po1.dump_xml())
for (path1, obj1), (path2, obj2) in zip(walk(po1), walk(po2)):
assert path1 == path2
assert path_to_str(path1) == path_to_str(path2)
assert type(obj1) is type(obj2)
if not isinstance(obj1, Object):
assert obj1 == obj2
for _ in zip_walk(po1):
pass
self.assertEqual(po1.dump(), po2.dump())
def testPO(self):
class SKU(StringPattern):
pattern = '\\d{3}-[A-Z]{2}'
class Comment(String):
xmltagname = 'comment'
class Quantity(Int):
pass
class USAddress(Object):
country = String.T(
default='US', optional=True, xmlstyle='attribute')
name = String.T()
street = String.T()
city = String.T()
state = String.T()
zip = Float.T()
class Item(Object):
part_num = SKU.T(xmlstyle='attribute')
product_name = String.T()
quantity = Quantity.T()
us_price = Float.T(xmltagname='USPrice')
comment = Comment.T(optional=True)
ship_date = DateTimestamp.T(optional=True)
class Items(Object):
item_list = List.T(Item.T())
class PurchaseOrderType(Object):
order_date = DateTimestamp.T(optional=True, xmlstyle='attribute')
ship_to = USAddress.T()
bill_to = USAddress.T()
comment = Comment.T(optional=True)
items = Items.T()
class PurchaseOrder(PurchaseOrderType):
xmltagname = 'purchaseOrder'
xml = '''<?xml version="1.0"?>
<purchaseOrder orderDate="1999-10-20">
<shipTo country="US">
<name>Alice Smith</name>
<street>123 Maple Street</street>
<city>Mill Valley</city>
<state>CA</state>
<zip>90952</zip>
</shipTo>
<billTo country="US">
<name>Robert Smith</name>
<street>8 Oak Avenue</street>
<city>Old Town</city>
<state>PA</state>
<zip>95819</zip>
</billTo>
<comment>Hurry, my lawn is going wild</comment>
<items>
<item partNum="872-AA">
<productName>Lawnmower</productName>
<quantity>1</quantity>
<USPrice>148.95</USPrice>
<comment>Confirm this is electric</comment>
</item>
<item partNum="926-AA">
<productName>Baby Monitor</productName>
<quantity>1</quantity>
<USPrice>39.98</USPrice>
<shipDate>1999-05-21</shipDate>
</item>
</items>
</purchaseOrder>
'''
po1 = load_xml_string(xml)
po2 = load_xml_string(po1.dump_xml())
for (path1, obj1), (path2, obj2) in zip(walk(po1), walk(po2)):
assert path1 == path2
assert path_to_str(path1) == path_to_str(path2)
assert type(obj1) is type(obj2)
if not isinstance(obj1, Object):
assert obj1 == obj2
for _ in zip_walk(po1):
pass
self.assertEqual(po1.dump(), po2.dump())