a zew@sddlZddlmZddlZddlZddlZddlmZddlmZGdddej Z Gdddej Z Gd d d ej Z e d kredS) N)support)array) script_helperc@s4eZdZddZddZddZddZd d Zd S) LegacyBase64TestCasecCsZ|t|d|t|gtddd}|t||tdd}|t||dS)Ns1234B)rI) assertRaises TypeError memoryviewcast)selffmultidimensionalint_datarD/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_base64.pycheck_type_errors s z&LegacyBase64TestCase.check_type_errorscCs|j}|tdd|tdd|tdd|tdd|td d |td d |ttdd|ttdd|ttd dd|tjdS) Nwww.python.orgd3d3LnB5dGhvbi5vcmc= aYQ== abYWI= abcYWJj RabcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#0^&*();:<>,. []{}rYWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0 NTY3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ== r) assertEqualbase64Z encodebytes bytearrayr rrreqrrrtest_encodebytess z%LegacyBase64TestCase.test_encodebytescCs|j}|tdd|tdd|tdd|tdd|td d |td d |ttdd|ttdd|ttd dd|tjdS) Nrrrrrrrrrrrr)r r!Z decodebytesr"r rrr#rrrtest_decodebytes(s z%LegacyBase64TestCase.test_decodebytescCs|j}ddlm}m}|d}|}t||||d|ttj|d||ttj|d||ttj|d|dS)NrBytesIOStringIOrrabcr) r ior(r)r!encodegetvaluer r )rr$r(r)infpoutfprrr test_encode;s z LegacyBase64TestCase.test_encodecCsddlm}m}|d}|}t||||d|ttj |d||ttj |d||ttj |d|dS)Nrr'd3d3LnB5dGhvbi5vcmc=rzYWJj r) r+r(r)r!decoder r-r r r,)rr(r)r.r/rrr test_decodeLs z LegacyBase64TestCase.test_decodeN)__name__ __module__ __qualname__rr%r&r0r3rrrrr s rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3S)4BaseXYTestCasecCs |t|d|t|gdS)Nrr r rrrrrcheck_encode_type_errors\sz'BaseXYTestCase.check_encode_type_errorscCs|t|gdSNr8r9rrrcheck_decode_type_errors`sz'BaseXYTestCase.check_decode_type_errorscCsl|j}t|}|||||||||t||||td|||tj||tj|dS)Nr)r r"r rcheck_nonbyte_element_formatr! b64encodecheck_multidimensional)rr bytes_dataZexpectedr$brrrcheck_other_typescs z BaseXYTestCase.check_other_typescCsTt|drdnd}||}t|ddf}t|d|}|||||dS)Nrrrlenr r r )rrdatapaddingr@shaperrrrr?os z%BaseXYTestCase.check_multidimensionalcCsBddt|d}||}t|d}|||||dS)NrCr rD)rrrFrGr@rrrrr=vsz+BaseXYTestCase.check_nonbyte_element_formatcCs|j}|tdd|tdd|tdd|tdd|td d |td d |td d |tjdddd|tjdtddd|tjdtddd|tjdtdddd|tjdd|tj|jt tjd dd|t dd|t dd|t dd|t d d |t d d |t d d |tj dd|tj |t dd|tj dd|tj dS)Nrr1rCAA==rYQ==rYWI=rYWJjrrspYWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0NTY3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==Vo*$altchars01a*b$cdrabcdsYWJjZA==z*$01a-b_cd) r r!r>r"r rrBr:r r Zstandard_b64encodeZurlsafe_b64encoder#rrrtest_b64encode}sP    zBaseXYTestCase.test_b64encodec Cs|j}dddddddd}|D].\}}|t|||t|d |q"|tjd d|tjd d i}|D]p\\}}}|d }|d }|tj||d ||tj||d ||tj||d ||tj||d |q~|D].\}}|t|||t|d |q|tjd d|tjd dd} | D]0\}}|t|||t|d |qV|tjdd |tjdS)NrrCrrrrr)r1rJrKrLrMsqYWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0 NTY3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==rasciirM)rRrOrNrP)rTrrT) r itemsr! b64decoder2rBr<standard_b64decodeurlsafe_b64decode) rr$testsrFresZtests_altcharsrQdata_strZ altchars_strZ tests_urlsaferrrtest_b64decodesJ     zBaseXYTestCase.test_b64decodecCs(|tjtjd|tjtjddS)Nrr*)r binasciiErrorr!rXrrrrtest_b64decode_padding_errorsz+BaseXYTestCase.test_b64decode_padding_errorc Cs(d}tjtjtjf}|D]\}}|D]X}|j||d6|||||||d|Wdq$1sr0Yq$|tj tj|ddWdn1s0Y|tj $tj|dddWdq1s0Yqd}|tdd||td |dS) N) )s%3d==)s$3d==rc)s[==r)sYW]3=sam)s3{d==rc)s3d}==rc)s@@r)!r)rr)s YWJj YWI=sabcab)bstrfuncrVT)validatess++[[//]]s[]s++--//__) r!rXrYrZsubTestr r2r r_r`)rr[Zfuncsrer\rfrrrtest_b64decode_invalid_charss"  6,4z+BaseXYTestCase.test_b64decode_invalid_charscCs|j}|tdd|tdd|tdd|tdd|tdd |td d |td d |tjd d |tjdS)NrrCAA======rME======rMFRA====rMFRGG===rSMFRGGZA=abcdeMFRGGZDF)r r!Z b32encoderBr:r#rrrtest_b32encode szBaseXYTestCase.test_b32encodecCsr|j}dddddddd}|D].\}}|t|||t|d |q"|tjd d|tjdS) NrrCrrrrSro)rrjrkrlrmrnrprVrm)r rWr! b32decoder2rBr<rr$r[rFr\rrrtest_b32decodeszBaseXYTestCase.test_b32decodec CsN|j}dddddddddddd }|D]2\}}|t|d||t|d d|q*|tjtjd |tjtjd |td d |tdd d dd}|D]\\}}}|d }|d }|tj||d||tj||d||tj||d||tj||d||tjtj||tjtj|qdS)NrrrrrSro) rrkrlrmrnrpme======smfra====smfrgg===smfrggza=smfrggzdfTrVruzme======sMLO23456sbݭZMLO23456sb))M1023456L)rvI)map01)r rWr!rrr2r r_r`) rr$r[rFr\Z map_testsryr]Z map01_strrrrtest_b32decode_casefold)s@  z&BaseXYTestCase.test_b32decode_casefoldc Cs gd}gd}tddD]B}|r0|d||D]&}t||dkr4||d|q4q|D]}|j|d|tjt|Wdn1s0Y|tj t| dWdn1s0YWdqb1s0YqbdS) N)rsABCDEF==s==ABCDEF)MsMEsMFRAsMFRGGsMFRGGZArpr=)rFrV) rangeappendrErhr r_r`r!rrr2)rr[prefixesiprefixrFrrrtest_b32decode_errorRs(z#BaseXYTestCase.test_b32decode_errorcCsF|j}|tdd|tdd|tjdd|tjdS)N 0102ABCDEFrC00)r r!Z b16encoderBr:r#rrrtest_b16encodebs zBaseXYTestCase.test_b16encodecCs|j}|tdd|tdd|tdd|tdd|tjtjd|tjtjd|tdd d|tdd d|tjdd|tj|ttdd d|tt dd d|tt d dd d|tjtjd |tjtjd dS) NrrZ 0102ABCDEFrrCZ00s 0102abcdefZ 0102abcdefTrZ0102AGZ010) r r! b16decoder r_r`rBr<r"r rr#rrrtest_b16decodeks0  zBaseXYTestCase.test_b16decodecCs\|j}ddddttdddddd d d d d ddddddddddddi }|D]N\}}|t||||tj|dd|||tj|ddd|d|qN|tjdd|ttjd|tjdd dd!d"|tjd#d dd!d$|tjdd dd!d%|tjd&ddd'd(|tjd)ddd'd*|tjd+ddd'd,|tjd-ddd'd.dS)/NrrGB\6`E-ZP=Df.1GEb>s?!!*-'"9eu7#RLhG$k3[W&.oNg'GVB"(`=52*$$(B+<_pR,UFcb-n-Vr/1iJ-0JP==1c70M3&s#]4?Ykm5X@_(6q'R884cEH9MJ8X:f1+h<)lt#=BSg3>[:ZC?t!MSA7]@cBPD3sCi+'.E,fo>FEMbNG^4U^I!pHnJ:W<)KS>/9Ll%"IN/`jYOHG]iPa.Q$R$jD4S=Q7DTV8*TUnsrdW2ZetXKAY/Yd(L?['d?O\@K2_]Y2%o^qmn*`5Ta:aN;TJbg"GZd*^:jeCE.%f\,!5gtgiEi8N\UjQ5OekiqBum-X60nF?)@o_%qPq"ad`r;HTrg@:E_WAS,RgBkhF"D/O92EH6,BF`qtRH$VbC6UX@47n?3D92&&T:Jand;cHat='/U/0JP==1c70M3&r-I,;,E,oN2F(oQ1zzero compressionH=_,8+Cf>,E,oN2F(oQ1!!!! Boundary:6>q!aA79M(3WK-[!!Space compr: ;fH/TAKYK$D/aMV+r)ZwrapcolrsGB\6`E- ZP=Df.1 GEb>swww.python.orgszGB\6`E -ZP=Df. 1GEb>s<~GB\6` E-ZP=Df .1GEb> ~> Z foldspacesryy y+}2SXo+ITwPvYU}0ioWMyV&XlZI|Y;A6DaB*^Tbai%jczJqze0_d@fPsR8goTEOh>41ejE#[:ZC?t!MSA7]@cBPD3sCi+'.E,fo>FEMbNG^4U^I!pHnJ:W<)KS>/9Ll%"IN/`jYOHG]iPa.Q$R$jD4S=Q7DTV8*TUnsrdW2ZetXKAY/Yd(L?['d?O\@K2_]Y2%o^qmn*`5Ta:aN;TJbg"GZd*^:jeCE.%f\,!5gtgiEi8N\UjQ5OekiqBum-X60nF?)@o_%qPq"ad`r;HTrrrrrrrrrrFrrVrrTz<~%s~>rrrrrrrrrr)r rrrWr! a85decoder2rBrsrrrtest_a85decodes>  zBaseXYTestCase.test_a85decodecCsz|j}ddttddddddd d d d d d }|D].\}}|t|||t|d|q6|tjdddS)Nrrrrrrrrrrrrr) rrrrrrrrrrrrrrVr)r rrrWr! b85decoder2rBrsrrrtest_b85decodes*  zBaseXYTestCase.test_b85decodecCs|j}|tjdddd|tjdddd|tjdddd|tjd ddd |tjd ddd |tdd |tdd|tdd|td d |td ddS)NxTZpadsGQ7^DxxsG^'2gxxxsG^+H5xxxxsG^+IXxxxxxs G^+IXGQ7^Dxxxxxxxxxxx)r r!rrr#rrrtest_a85_padding*szBaseXYTestCase.test_a85_paddingcCs|j}|tjdddd|tjdddd|tjdddd|tjd ddd |tjd ddd |tdd |tdd|tdd|td d |td ddS)NrTrscmMzZrscz6H+rsczAdKrsczAetrs czAetcmMzZrrrr)r r!rrr#rrrtest_b85_padding9szBaseXYTestCase.test_b85_paddingc CsttdttddBtd}|D]}|jtt|gd$tdt|gWdn1sh0Y|jtt|gd(tjdt|gddWdn1s0Y|jtt|gd,tjd t|gd d dWdq&1s0Yq&|jttjd d d|jttjd d d|ttjd|jttjdddtjdd d|jttjddd|jttjddd|jttjdddd|jttjddd|jttjddd|jttjddd|jttjddd|jttjddddS)N vs msgs!!!!Frs<~!!!!rTs malformeds<~still malformeds<~~>sabcxsabcdeysa b cr)rZ ignorecharssss8ss8Wss8W-ss8W-")setrr ValueErrorrr!rrZillegalcrrrtest_a85decode_errorsHsF"26>       z$BaseXYTestCase.test_a85decode_errorsc Csttdtdttdd}|D]L}|jtt|gd$tdt|gWdq&1sh0Yq&|ttjd|ttjd|ttjd |ttjd |ttjd dS) N!s "',./:[\]rrs0000|s|Ns|Nss|NsCs|NsC1)listrr rrr!rrrrrtest_b85decode_errorsks  4z$BaseXYTestCase.test_b85decode_errorscCs<tjtjtjtjtjtjtjf}|D]}|t |dq$dS)Nuwith non-ascii Ë) r!rXrYrZrrrrrr r)rZ decode_funcsrrrrtest_decode_nonascii_strysz'BaseXYTestCase.test_decode_nonascii_strcCs|ttjtdSr;)Z assertTrue issubclassr_r`rrarrrtest_ErrorHeritagesz!BaseXYTestCase.test_ErrorHeritageN)r4r5r6r:r<rBr?r=rUr^rbrirqrtrzrrrrrrrrrrrrrrrrrr7Xs2 28  ) 2"-## r7c@s<eZdZddZddZddZddZd d Zd d Zd S)TestMaincCstjtjrttjdSr;)ospathexistsrTESTFNunlinkrarrrtearDownszTestMain.tearDowncGstjddg|RjS)N-mr!)rZassert_python_okout)rargsrrr get_outputszTestMain.get_outputcCs|d}||ddS)Nz-t)b'Aladdin:open sesame's!b'QWxhZGRpbjpvcGVuIHNlc2FtZQ==\n'r)rZassertSequenceEqual splitlines)routputrrrtest_encode_decodes zTestMain.test_encode_decodecCsXttjd}|dWdn1s,0Y|dtj}||ddS)Nwbab -eYf9iCg==openrrwriterr rstriprfprrrrtest_encode_files(zTestMain.test_encode_filecCsZtddd}|d\}}Wdn1s20Y||d||dS)Nrr!rrr)rZ spawn_python communicater rZ assertIsNone)rprocrerrrrrtest_encode_from_stdins,zTestMain.test_encode_from_stdincCsXttjd}|dWdn1s,0Y|dtj}||ddS)Nrrz-dsabrrrrrr3s(zTestMain.test_decodeN) r4r5r6rrrrrr3rrrrrs r__main__)Zunittesttestrr!r_rrZ test.supportrZTestCaserr7rr4mainrrrrs   N4"