>X5@sddlZddljjZddlZddlmZddlm Z ddlm Z ddZ ddZ d d Z d d Zd dZddZddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Z d1d2Z!d3d4Z"d5d6Z#d7d8Z$d9d:Z%d;d<Z&d=d>Z'd?d@Z(dAdBZ)dCdDZ*dEdFZ+dGdHZ,dIdJZ-dKdLZ.dMdNZ/dOdPZ0dQdRZ1dSdTZ2dUdVZ3dWdXZ4dYdZZ5d[d\Z6ej7j8Z8d]d^Z9d_d`Z:dadbZ;dcddZ<dedfZ=dgdhZ>didjZ?dkdlZ@dS)mN) rules_yacc) parse_rules) scan_itemcCs||}t|gf}||k}|s tjd |fd ||fitj|d6tj|d6dtjkstj|rtj|ndd6dtjkstj|rtj|ndd6}di|d 6}ttj|nd}}}dS)N!=0%(py3)s {%(py3)s = %(py0)s(%(py1)s) } != %(py6)spy3py6 matchfuncpy0datapy1assert %(py8)spy8)r)rr) len @pytest_ar_call_reprcompare _saferepr @py_builtinslocals_should_repr_global_nameAssertionError_format_explanation)r r @py_assert2 @py_assert5 @py_assert4 @py_format7 @py_format9rT/home/kop/biz/barnett/enforcer_project/BRIEF-Script/Enforcer/tests/test_scan_item.py not_matchs  r cCs||}t|gf}||k}|s tjd |fd ||fitj|d6tj|d6dtjkstj|rtj|ndd6dtjkstj|rtj|ndd6}di|d 6}ttj|nd}}}dS)N==0%(py3)s {%(py3)s = %(py0)s(%(py1)s) } == %(py6)srrr r r r r assert %(py8)sr)r!)r"r#) rrrrrrrrr)r r rrrrrrrr does_matchs  r$cCseidtjdd6}tj}d}d}|||}||k}|sKtjd|fd||fitj|d6dtjkstjtrtjtndd6tj|d 6d tjkstj|rtj|nd d 6tj|d 6tj|d 6}di|d6}ttj |nd}}}}}dS)z%Can a ScanItem be used as a hashable?barfoorinX%(py8)s {%(py8)s = %(py2)s {%(py2)s = %(py0)s.ScanItem }(%(py4)s, %(py6)s) } in %(py10)srrr py4dctpy10rpy2r assert %(py12)spy12N)r')r(r-) rScanItemrrrrrrrr)r* @py_assert1 @py_assert3r @py_assert7 @py_assert9 @py_format11 @py_format13rrrtest_scanitem_hashable's  r6c Cs3tjttjddjdWdQXdS)Nr&rr%)pytestraisesNotImplementedErrorrr/matchesrrrrtest_scanitem_match-sr;c Csidtjddd6}tj}d}d}d}||||}||k}|sgtjd|fd||fitj|d6dtjkstjtrtjtndd 6tj|d 6d tjkstj|rtj|nd d 6tj|d 6tj|d6tj|d6}di|d6}ttj |nd}}}}}}dS)z%Can a ScanItem be used as a hashable?Zbazr&rr%r'i%(py10)s {%(py10)s = %(py2)s {%(py2)s = %(py0)s.ValuedScanItem }(%(py4)s, %(py6)s, %(py8)s) } in %(py12)srrr r)r*r.r+rr,r assert %(py14)spy14N)r')r<r=) rValuedScanItemrrrrrrrr) r*r0r1rr2r3 @py_assert11r5 @py_format15rrrtest_valuedscanitem_hashable3s  rBcCs)tjd}d}t|j|dS)z$Does the match match a valid string?rzSome-99 StringN)rUserStrScanItemr$r:)itemr rrrtest_userstrscanitem_matches:srEc Csxtjd}d}|j}d}||}||}t|gf}||k}|sZtjd|fd||fidtjkstj|rtj |ndd6tj |d6tj |d 6d tjkstj|rtj |nd d 6tj |d 6tj |d 6}di|d6} t tj | nd}}}}}}dS)z)Does the match stop after a valid string?rzSome-99 Stringz - more Textr!Z%(py8)s {%(py8)s = %(py2)s {%(py2)s = %(py0)s.matches }((%(py3)s + %(py5)s)) } == %(py11)sr rrpy11rDr py5r,r assert %(py13)spy13N)r!)rFrI) rrCr:rrrrrrrrr) rDr r0r @py_assert6r2 @py_assert10r3 @py_format12 @py_format14rrrtest_userstrscanitem_stopsAs   rOc Cstj}d}d}d}||||}tj}d}d}|||}||k} | rtjdf| fdf||fi tj|d6tj|d6dtjkptjtrtjtndd 6tj|d 6dtjkptjtr"tjtndd 6tj|d 6tj|d 6tj|d6tj|d6tj|d6tj|d6} ddi| d6} t tj | nd}}}}}} }}}}dS)zWA SpecialUserStrScanItem compares different from a UserStrScanItem of the same typez z value rz%(py10)s {%(py10)s = %(py2)s {%(py2)s = %(py0)s.SpecialUserStrScanItem }(%(py4)s, %(py6)s, %(py8)s) } != %(py20)s {%(py20)s = %(py14)s {%(py14)s = %(py12)s.UserStrScanItem }(%(py16)s, %(py18)s) }py16rrr r)r.py20r+py18rr,r>r zassert %(py22)spy22N) rSpecialUserStrScanItemrCrrrrrrrr) r0r1rr2r3 @py_assert13 @py_assert15 @py_assert17 @py_assert19r@ @py_format21 @py_format23rrr&test_specialuserstrscanitem_notspecialIs    *r\c Cstj}d}d}d}||||}tj}d}d}d}||||} || k} | rtjdf| fdf|| fi tj|d6tj|d6d tjkptjtrtjtnd d 6tj|d 6tj|d 6tj| d 6tj|d6tj|d6tj|d6d tjkpitjtr{tjtnd d6tj|d6tj|d6} ddi| d6} ttj | nd}}}}}} }}}}} dS)zR2 objects compare different when the strings following the differsrPz z value0 z value1 rz%(py10)s {%(py10)s = %(py2)s {%(py2)s = %(py0)s.SpecialUserStrScanItem }(%(py4)s, %(py6)s, %(py8)s) } != %(py22)s {%(py22)s = %(py14)s {%(py14)s = %(py12)s.SpecialUserStrScanItem }(%(py16)s, %(py18)s, %(py20)s) }rQrSrr.rrrTr)r+rRr r,r>r zassert %(py24)spy24N) rrUrrrrrrrr) r0r1rr2r3rVrWrXrY @py_assert21r@r[ @py_format25rrr%test_specialuserstrscanitem_differentPs"   :r`c Cstj}d}d}d}||||}tj}d}d}d}||||} || k} | rtjdf| fdf|| fi tj|d6tj|d6dtjkptjtrtjtndd 6tj|d 6tj|d 6tj| d 6tj|d 6tj|d6tj|d6dtjkpitjtr{tjtndd6tj|d6tj|d6} ddi| d6} ttj | nd}}}}}} }}}}} dS)zS2 objects compare equal when the strings followig the do not differrPz z value r!z%(py10)s {%(py10)s = %(py2)s {%(py2)s = %(py0)s.SpecialUserStrScanItem }(%(py4)s, %(py6)s, %(py8)s) } == %(py22)s {%(py22)s = %(py14)s {%(py14)s = %(py12)s.SpecialUserStrScanItem }(%(py16)s, %(py18)s, %(py20)s) }rQrSrr.rrrTr)r+rRr r,r>r zassert %(py24)sr]N) rrUrrrrrrrr) r0r1rr2r3rVrWrXrYr^r@r[r_rrr test_specialuserstrscanitem_sameWs"   :rac Cs~tjddd}d}|j}d}||}||}t|gf}||k}|s`tjd|fd||fidtjkstj|rtj |ndd 6tj |d 6tj |d 6d tjkstj|r tj |nd d 6tj |d6tj |d6}di|d6} t tj | nd}}}}}}dS)z>Comparison works when there's a trailing string where expectedrPz z value zSome Element or Anotherz value and morer!Z%(py8)s {%(py8)s = %(py2)s {%(py2)s = %(py0)s.matches }((%(py3)s + %(py5)s)) } == %(py11)sr rrrGrDr rHr,r assert %(py13)srJN)r!)rbrc) rrUr:rrrrrrrrr) rDr r0rrKr2rLr3rMrNrrr)test_specialuserstrscanitem_trail_succeed^s   rdcCs/tjddd}d}t|j|dS)z0Comparison works when there's no trailing stringrPz z value zSome Element or AnotherN)rrUr$r:)rDr rrr+test_specialuserstrscanitem_notrail_succeedesrec Cstjddd}d}|j|d}t|gf}||k}|stjd|fd||fitj|d6d tjkstj |rtj|nd d 6}di|d 6}t tj |nd}}|d}d}t|} || k}|stjd|fd|| fidtjksatj trptjtndd6tj| d6tj|d 6tj|d6} di| d6} t tj | nd}}}} dS)zBComparison fails when there's a trailing string where not expectedrPz z value zSome random value or Anotherz value and morer%(py0)s != %(py3)srresultr r assert %(py5)srHNrz Some randomr!0%(py1)s == %(py7)s {%(py7)s = %(py3)s(%(py5)s) }rpy7r assert %(py9)spy9)r)rfrh)r!)rirk) rrUr:rrrrrrrrr) rDr rgrr0 @py_format4 @py_format6 @py_assert0rrK @py_format8 @py_format10rrr&test_specialuserstrscanitem_trail_faills& l   rrcCsHtjddd}tjddd}t|}t|}||k}| rtjdf|fdf||fitj|d6dtjkptj|rtj|ndd 6tj|d 6d tjkptjtrtjtnd d 6d tjkp(tjtr:tjtnd d 6dtjkp_tj|rqtj|ndd6}ddi|d6}t tj |nd}}}t|}t|}||k}| r6tjdf|fdf||fitj|d6dtjkp7tj|rItj|ndd 6tj|d 6d tjkp~tjtrtjtnd d 6d tjkptjtrtjtnd d 6dtjkptj|rtj|ndd6}ddi|d6}t tj |nd}}}dS)z1The hash function works with special user stringsrPz z value0 z value1 r!zN%(py3)s {%(py3)s = %(py0)s(%(py1)s) } == %(py8)s {%(py8)s = %(py5)s(%(py6)s) }ritem0rrhashr rHr r zassert %(py10)sr+NrzN%(py3)s {%(py3)s = %(py0)s(%(py1)s) } != %(py8)s {%(py8)s = %(py5)s(%(py6)s) }item1) rrUrtrrrrrrrr)rsrurr2rrr4rrr test_specialuserstrscanitem_hashus( ( (rvcCs)tjd}d}t|j|dS)zA hash scan item matches digitsr2340N)r HashScanItemr$r:)rDr rrrtest_hashscanitem_matches~srycCs)tjd}d}t|j|dS)z2A hash scan item matches digits with leading zerosrZ02340N)rrxr$r:)rDr rrr"test_hashscanitem_matches_leading0srzcCs)tjd}d}t|j|dS)z2A hash scan item does not match digits and lettersrZ02a340N)rrxr r:)rDr rrrtest_hashscanitem_failssr{cCs)tjd}d}t|j|dS)z*An entity matches a reasonable entity namerz Some EntityN)rEntityScanItemr$r:)rDr rrrtest_entity_scan_itemsr}cCs)tjd}d}t|j|dS)z%A date scan item matches a valid daterz 2017-01-01N)r DateScanItemr$r:)rDr rrr$test_date_scan_item_matches_gooddatesrcCs)tjd}d}t|j|dS)z(A date scan item matches an invalid daterz 0000-00-00N)rr~r$r:)rDr rrr#test_date_scan_item_matches_baddatesrcCs)tjd}d}t|j|dS)z,A date scan item does not match a short daterz2017-1-1N)rr~r r:)rDr rrr"test_date_scan_item_mismatch_shortsrcCs)tjd}d}t|j|dS)z4A date scan item does not match non-date-like stringrz2017-sandwitch-1N)rr~r r:)rDr rrr$test_date_scan_item_mismatch_letterssrcCs)tjd}d}t|j|dS)z$A year scan item matches a good yearrZ2017N)r YearScanItemr$r:)rDr rrrtest_year_scan_item_match_goodsrcCs)tjd}d}t|j|dS)z#A year scan item matches a bad yearrZ0017N)rrr$r:)rDr rrrtest_year_scan_item_match_badsrcCs)tjd}d}t|j|dS)z+A year scan item does not match alpha charsrZ20a17N)rrr r:)rDr rrr"test_year_scan_item_mismatch_alphssrcCs)tjd}d}t|j|dS)zA hash scan item matches digitsrrwN)rLast4DigitsScanItemr$r:)rDr rrr test_last4digitsscanitem_matchessrcCs)tjd}d}t|j|dS)z2A hash scan item matches digits with leading zerosrZ0234N)rrr$r:)rDr rrr)test_last4digitsscanitem_matches_leading0srcCs)tjd}d}t|j|dS)z2A hash scan item does not match digits and lettersrZ2a340N)rrr r:)rDr rrrtest_last4digitsscanitem_failssrcCs)tjd}d}t|j|dS)zA version number matchesrz v10N)rVersionScanItemr$r:)rDr rrrtest_versionscanitem_matchessrcCs)tjd}d}t|j|dS)z A version with a leading 0 failsrz v01N)rrr r:)rDr rrr#test_versionscanitem_fails_leading0srcCs,tjdd}d}t|j|dS)z2A parenthetical with no optional user text matchesrz (some parenthetical)N)r ParenScanItemr$r:)rDr rrr test_parenscanitem_matches_plainsrc Csdtjdd}d}d|d}|j}||}t||gf}||k}|sNtjd|fd||fidtjkstj|rtj |ndd 6d tjkstj|rtj |nd d 6tj |d 6tj |d 6tj |d6}di|d6}t tj |nd}}}}dS)ziA parenthetical with valid optional user text matches and returns the expected content to be reportedrz (some parenthetical)zsome parenthetical; Extra Stuffz ()r!M%(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.matches }(%(py3)s) } == %(py8)sr rrDr rHr,rr assert %(py10)sr+N)r!)rr) rrr:rrrrrrrrr) rDcontentr r0rr2rKrr4rrr test_parenscanitem_matches_extras  rcCs,tjdd}d}t|j|dS)z5A parenthetical with different content fails to matchrz (some parenthetical)z (some other parenthetical)N)rrr r:)rDr rrr!test_parenscanitem_nomatch_differsrcCs,tjdd}d}t|j|dS)z>A parenthetical with invalid optional user text does not matchrz (some parenthetical)z (some parenthetical; ===)N)rrr r:)rDr rrr test_parenscanitem_nomatch_extra srcCs,tjdd}d}t|j|dS)z.A parenthetical does not match non-parenthesisrz (some parenthetical)z some parentheticalN)rrr r:)rDr rrr#test_parenscanitem_nomatch_notparensrc Csatjd}d}d|d}|j}||}t||gf}||k}|sKtjd|fd||fidtjkstj|rtj |ndd6d tjkstj|rtj |nd d 6tj |d 6tj |d 6tj |d 6}di|d6}t tj |nd}}}}dS)z:A repeated parens scan item matches a single parentheticalrzsome text and morez (rr!M%(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.matches }(%(py3)s) } == %(py8)sr rrDr rHr,rr assert %(py10)sr+N)r!)rr) rRepeatedParensScanItemr:rrrrrrrrr) rDrr r0rr2rKrr4rrr$test_repeatedparensscanitem_matches1s  rc Csrtjd}d}d}d|d|d}|j}||}t|||gf}||k}|s\tjd|fd||fid tjkstj|rtj |nd d 6d tjkstj|rtj |nd d 6tj |d 6tj |d6tj |d6}di|d6} t tj | nd}}}}dS)z6A repeated parens scan item matches two parentheticalsrzsome text and morezanother parentheticalz (z) (rr!M%(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.matches }(%(py3)s) } == %(py8)sr rrDr rHr,rr assert %(py10)sr+N)r!)rr) rrr:rrrrrrrrr) rDcontent1content2r r0rr2rKrr4rrr$test_repeatedparensscanitem_matches2"s  rcCs)tjd}d}t|j|dS)z3A repeated parens scan item fails on trailing charsrz (some text; and more) N)rrr r:)rDr rrr*test_repeatedparensscanitem_fails_trailing+srcCs)tjd}d}t|j|dS)zBA repeated parens scan item fails when parenthetical is not closedrz (some text; and more N)rrr r:)rDr rrr+test_repeatedparensscanitem_fails_notclosed2srcCs)tjd}d}t|j|dS)z9A repeated parens scan item fails without a leading spacerz(some text; and more)N)rrr r:)rDr rrr0test_repeatedparensscanitem_fails_noleadingspace9srcCs)tjd}d}t|j|dS)z?A repeated parens scan item fails when space follows open parenrz( some text; and more)N)rrr r:)rDr rrr+test_repeatedparensscanitem_fails_openspace@srcCs)tjd}d}t|j|dS)z>A repeated parens scan item fails when space leads close parenrz(some text; and more )N)rrr r:)rDr rrr,test_repeatedparensscanitem_fails_closespaceGsrcCs)tjd}d}t|j|dS)z,A repeated parens scan item fails when emptyrz()N)rrr r:)rDr rrr+test_repeatedparensscanitem_fails_nocontentNsrc Cs9tjdg}tjt|jdWdQXdS)z5Trying to match an optional scan item raises an errorrZanythingN)rOptionalScanItemr7r8r9r:)rDrrr$test_optionalscanitem_does_not_matchVsrc CsWd}d}d|d|d}tj}||}||g}||k}|sAtjd|fd||fidtjkstj|rtj|ndd 6d tjkstjtrtjtnd d 6tj|d 6tj|d 6tj|d6}di|d6}ttj |nd}}}}dS)zReturns what's expectedzsome text and morezanother parentheticalz (z) (rr!\%(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.repeated_paren_content }(%(py3)s) } == %(py8)sr rrr rHr,rr assert %(py10)sr+N)r!)rr) rrepeated_paren_contentrrrrrrrr) rrr r0rr2rKrr4rrrtest_repeated_paren_contenths rcCstjdgd}|j}d}||k}|stjd|fd||fidtjksxtj|rtj|ndd6tj|d 6tj|d 6}di|d 6}t tj |nd }}}d S)zpTest that a scan item list of one item produces a specialuserstrscanitem with an empty string for a next_struser_str rr r!0%(py2)s {%(py2)s = %(py0)s.next_str } == %(py5)srDr rHr,assert %(py7)srjN)zuser_strr)r!)rr) rscan_to_specialuserstrscanitemnext_strrrrrrrrr)rDr0rr1rnrprrr#test_scan_to_specialuserstrscanitemqs  |rc Cstjdgd}t|}d}||k}|stjd|fd||fitj|d6tj|d6d tjkstjtrtjtnd d 6d tjkstj|rtj|nd d 6}di|d6}t tj |nd}}}|d}|j }d}||k}|stjd|fd||fidtjkstj|rtj|ndd 6tj|d6tj|d6} di| d6} t tj | nd}}}|j }d}||k}|stjd |fd!||fidtjksdtj|rstj|ndd 6tj|d6tj|d6} d"i| d6} t tj | nd}}}dS)#z1Does an entity tuple produce an entity scan item?entity Some EntityrPr!0%(py3)s {%(py3)s = %(py0)s(%(py1)s) } == %(py6)srrrr itemsr r assert %(py8)srNr,%(py2)s {%(py2)s = %(py0)s.type } == %(py5)srDrHr,assert %(py7)srj+%(py2)s {%(py2)s = %(py0)s.pos } == %(py5)s)zentityr)r!)rr)r!)rr)r!)rr) rtuple_to_itemsrrrrrrrrrtypepos) rrrrrrrDr0r1rnrprrrtest_tuple_to_items_entity~s4     |  |rc Cs-tjttjdgdWdQXdS)z2Do we get an exception when supplying a bad tuple? unknowntype somevaluerPN)rr)r7r8 ValueErrorrrrrrrtest_tuple_to_items_badtuplesrcCstjjd}tj|d}|d}|j}d}||k}|stjd|fd||fitj|d6tj|d6tj|d 6}di|d 6}t tj |nd }}}}d S)zNA speicaluserstrscanitem gets the correct next_str when given parsed inputz by rz by r!0%(py3)s {%(py3)s = %(py1)s.next_str } == %(py6)srrr r assert %(py8)srN)r!)rr) rparserparserrrrrrrr)scanrrorrrrrrrrtest_tuple_to_items_parseds  UrcCsJtjtd}|d}|j}d}||k}|stjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}|d }|j }d }||k}|sytjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}|d }|j}d }||k}|s4tjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}d S)z3Do we get valid position information in scan items?ztwo words
rr!+%(py3)s {%(py3)s = %(py1)s.pos } == %(py6)srrr r assert %(py8)srNraddress,%(py3)s {%(py3)s = %(py1)s.type } == %(py6)s)r!)rr)r!)rr)r!)rr) r to_scan_itemsrrrrrrrr)rrorrrrrrrrtest_to_scan_item_pos_simples8  U  U  UrcCstjtd}|d}|j}d}||k}|stjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}|d}|j }d }||k}|sytjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}d S)z3Do we get valid position information in scan items?z"two words
- dater!,%(py3)s {%(py3)s = %(py1)s.type } == %(py6)srrr r assert %(py8)srN+%(py3)s {%(py3)s = %(py1)s.pos } == %(py6)s)r!)rr)r!)rr) rrrrrrrrrr)rrorrrrrrrrtest_to_scan_item_pos_complexs(  U  UrcCstjtd}|d}|j}d}||k}|stjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}|d}|j }d }||k}|sytjd|fd||fitj|d6tj|d6tj|d6}di|d 6}ttj|nd }}}}|dj d}|j}d}||k}|s;tjd|fd ||fitj|d6tj|d6tj|d6}d!i|d 6}ttj|nd }}}}|dj d}|j }d}||k}|stjd"|fd#||fitj|d6tj|d6tj|d6}d$i|d 6}ttj|nd }}}}|dj d}|j}d}||k}|stjd%|fd&||fitj|d6tj|d6tj|d6}d'i|d 6}ttj|nd }}}}|dj d}|j }d}||k}|stjd(|fd)||fitj|d6tj|d6tj|d6}d*i|d 6}ttj|nd }}}}|d}|j}d}||k}|s<tjd+|fd,||fitj|d6tj|d6tj|d6}d-i|d 6}ttj|nd }}}}|d}|j }d}||k}|stjd.|fd/||fitj|d6tj|d6tj|d6}d0i|d 6}ttj|nd }}}}|dj d}|j}d}||k}|stjd1|fd2||fitj|d6tj|d6tj|d6}d3i|d 6}ttj|nd }}}}|dj d}|j }d}||k}|s{tjd4|fd5||fitj|d6tj|d6tj|d6}d6i|d 6}ttj|nd }}}}d S)7z?Is position information valid with optional pattern components?z8two words
[ - ][ - ] optionalr!,%(py3)s {%(py3)s = %(py1)s.type } == %(py6)srrr r assert %(py8)srN+%(py3)s {%(py3)s = %(py1)s.pos } == %(py6)srstringrPr $ last_4_digits()r!)rr)r!)rr)r!)rr)r!)rr)r!)rr)r!)rr)r!)rr)r!)rr)r!)rr)r!)rr) rrrrrrrrrrvalue)rrorrrrrrrrtest_to_scan_item_optional_poss  U  U  U  U  U  U  U  U  U  Ur)Abuiltinsr_pytest.assertion.rewrite assertionrewriterr7enforcerrrrr r$r6r;rBrErOr\r`rardrerrrvryrzr{r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrst