3 d5@sddlZddlZddlZddlZddlmZddlZddlZddlZddl m Z m Z ddl m Z ddljZddlmZddlmZdZd Zejjejjejjejejd Zd d ZddZGdddej Z!Gdddej Z"xrej#ejjedejj$d D]NZ%ejj&ejj'e%dZ(de(Z)e(d!kr:qddZ*e*e%Z+e,e)e+_-qWGdddeeZ.e-dkr~ej/dS)"N)etree) Namespace QualifiedName)PROV) AllTestsBase)RoundTripTestCaseexhttp://example.com/ns/ex#trhttp://example.com/ns/tr#xmlcCsB|jdk r|jjdkrd|_x|D]}tj|r$t|q$WdS)N)textstripr iselementremove_empty_tags)treeelemr3/tmp/pip-build-7vycvbft/prov/prov/tests/test_xml.pyrs   rc Csty|jddWntk r$YnXy|jddWntk rJYnXtj|}tj|}x(|jjdD]}|j}|j|qpWx(|jjdD]}|j}|j|qWt|jt|jt j }|j ||jdd|j j }dd|jD}t j }|j ||jdd|j j }dd|jD}tj||} dj| } | rpd} t| | dS) z| Helper function to compare two XML files. It will parse both once again and write them in a canonical fashion. rz //comment()cSsg|]}|jr|jqSr)r).0_irrr Bszcompare_xml..cSsg|]}|jr|jqSr)r)rrrrrrHs zStrings are not equal. N)seekAttributeErrorrparsegetrootZxpathZ getparentremoverioBytesIO write_c14nreaddecode splitlinesdifflib unified_diffjoinAssertionError) Zdoc1Zdoc2Zobj1obj2cpbufZstr1Zstr2r&err_msgmsgrrr compare_xml"sB            r/c@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)ProvXMLTestCasec Csptj}|jt}|jt|jdtj|dfd ftj(}|j d|dt t j j td|WdQRXdS) zc Test the serialization of example 6 which is a simple entity description. ztr:WD-prov-dm-20111215Document ex:version2r )format destinationzexample_06.xmlN)r2r3)prov ProvDocument add_namespaceEX_NSEX_TRentity PROV_TYPErr serializer/ospathr' DATA_PATH)selfdocumentex_nsactualrrrtest_serialization_example_6Ss   z,ProvXMLTestCase.test_serialization_example_6c Csptj}|jt|jdddtjtjdtjfd gtj (}|j d|dt t j jtd |Wd QRXd S) zP Test the serialization of example 7 which is a basic activity. zex:a1z2011-11-16T16:05:00z2011-11-16T16:06:00zex:editex:hostserver.example.orgr )r4r5zexample_07.xmlN)rFrG)r6r7r8r9activityr<LiteralZ XSD_QNAMErr r=r/r>r?r'r@)rArBrDrrrtest_serialization_example_7es  z,ProvXMLTestCase.test_serialization_example_7c Cstj}|jt|jd}|jd}|j||dddid|jd}|j||ddd idtj(}|j d |d t t j j td |Wd QRXd S)zR Test the serialization of example 8 which deals with generation. zex:e1zex:a1z2001-10-26T21:32:52zex:portp1)r;rHtimeZother_attributeszex:e2z2001-10-26T10:00:00p2r )r4r5zexample_08.xmlN)r6r7r8r9r;rHZwasGeneratedByrr r=r/r>r?r'r@)rArBZe1Za1e2rDrrrtest_serialization_example_8zs$       z,ProvXMLTestCase.test_serialization_example_8cCs`tjjtjjtddd}tj}|jt}|jt |j dtj |dfd f|j ||dS) ze Test the deserialization of example 6 which is a simple entity description. zexample_06.xmlr )sourcer4ztr:WD-prov-dm-20111215r1 ex:versionr3N)rQr3) r6r7 deserializer>r?r'r@r8r9r:r;r< assertEqual)rA actual_docexpected_documentrCrrrtest_deserialization_example_6s  z.ProvXMLTestCase.test_deserialization_example_6cCsdtjjtjjtddd}tj}tt}|j ||j dddtj t |dfd g|j ||d S) zg Test the deserialization of example 7 which is a simple activity description. zexample_07.xmlr )rPr4zex:a1z2011-11-16T16:05:00z2011-11-16T16:06:00Zeditex:hostserver.example.orgN)rWrX)r6r7rRr>r?r'r@rr9r8rHr<rrS)rArTrUrCrrrtest_deserialization_example_7s z.ProvXMLTestCase.test_deserialization_example_7cCs4d}tj,}|j||jddtjj|dd}WdQRXtj}tt}|j ||j t |j dtj t |dftj tdff|j||d d }tj,}|j||jddtjj|dd}WdQRXtj}|j t|j t |j dtj t |dftj td ftj tdff|j||d dS) a# Example 4 and 5 have a different type specification. They use an xsi:type as an attribute on an entity. This can be read but if written again it will become an XML child element. This is semantically identical but cannot be tested with a round trip. a ex:Workflow rr )rPr4Nztr:WD-prov-dm-20111215ZWorkflowZPlanZ example_04a ex:Workflow prov:Plan prov:Entity EntityZ example_05)rStringIOwriterr6r7rRrr9r8r:r;r<rrrS)rA xml_stringr Zactual_documentrUrCrrr&test_deserialization_example_04_and_05s8           z6ProvXMLTestCase.test_deserialization_example_04_and_05cCsd}tjddJ}tjdtj,}|j||jddtjj |dd}WdQRXWdQRX|j t |d |j d t |djk|j t |jddS) z PROV XML uses the element to enable the storage of non PROV information in a PROV XML document. It will be ignored by this library a warning will be raised informing the user. a bar T)recordalwaysrr )rPr4Nz[Document contains non-PROV information in . It will be ignored in this package.)warningscatch_warnings simplefilterrr[r\rr6r7rRrSlenZ assertTruestrmessage_records)rAr]wr docrrrtest_other_elementss    $z#ProvXMLTestCase.test_other_elementscCsztjjtd}tjj|dd}tdd}|jt |j d|j|j ||j|j dj j ||j|j dj jdd S) zu Tests that a default namespace that is defined in a lower level tag is written to a bundle. znested_default_namespace.xmlr )rPr4r zhttp://example.org/0/rarZe001N)r>r?r'r@r6r7rRrrSrerhZget_default_namespace identifier namespaceZ localpart)rAfilenamerjnsrrrtest_nested_default_namespaceAs z-ProvXMLTestCase.test_nested_default_namespacecCstjjtd}tjj|dd}|jt|j dt dd}|j|j dj ddj |tjjtd}tjj|dd}|j dj ddj }|j |||j|jd d S) zX Test the behaviour when namespaces are redefined at the element level. z+namespace_redefined_but_does_not_change.xmlr )rPr4rarzhttp://example.com/ns/ex#rznamespace_redefined.xmlzhttp://example.com/ns/new_ex#N)r>r?r'r@r6r7rRrSrerhr attributesrmZassertNotEqualuri)rArnrjroZnew_nsrrrtest_redefining_namespacesPs  z*ProvXMLTestCase.test_redefining_namespacesN) __name__ __module__ __qualname__rErJrOrVrYr^rkrprsrrrrr0RsQ+r0c@seZdZdddZdS) ProvXMLRoundTripFromFileTestCaseFc CsBtjj|dd}tj }|jd||dt||WdQRXdS)Nr )rPr4)r4r5 force_types)r6r7rRrr r=r/)rArnrxrBZnew_xmlrrr_perform_round_tripfs   z4ProvXMLRoundTripFromFileTestCase._perform_round_tripN)F)rtrurvryrrrrrwesrw*ztest_roundtrip_from_xml_%snested_default_namespace!nested_changing_default_namespace'namespace_redefined_but_does_not_changenamespace_redefinedcs$tdkrdndfdd}|S)Npc1TFcs|jddS)N)rx)ry)rA)frxrrfctszget_fct..fct)r)name)rrr)rrxrget_fcts rc@seZdZdZdS)RoundTripXMLTestsr N)rtrurvZFORMATrrrrrsr__main__)rr )r r )r{r|r}r~)0r%globinspectrlxmlrr>ZunittestrbZprov.identifierrrZprov.constantsrZ prov.modelmodelr6Zprov.tests.test_modelrZprov.tests.utilityrr9r:r?r'dirnameabspathgetfile currentframer@rr/ZTestCaser0rwiglobextseprnsplitextbasenamerZ test_namerrrfrtrmainrrrrsJ     "0&