[XMc@sdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z mZmZddl mZmZd Zd efd YZd efd YZdS(s[ Tests for L{twisted.cred._digest} and the associated bits in L{twisted.cred.credentials}. i(t verifyObject(tTestCase(tmd5tsha1(t IPv4Address(t LoginFailed(tcalcHA1tcalcHA2tIUsernameDigestHash(t calcResponsetDigestCredentialFactorycCs|jdjS(Ntbase64(tencodetstrip(ts((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyt b64encodestFakeDigestCredentialFactorycBs)eZdZdZdZdZRS(s\ A Fake Digest Credential Factory that generates a predictable nonce and opaque cOs&tt|j||d|_dS(Nt0(tsuperRt__init__t privateKey(tselftargstkwargs((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyRscCsdS(s) Generate a static nonce t178288758716122392881254770685((R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyt_generateNoncescCsdS(s& Return a stable time i((R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyt_getTime&s(t__name__t __module__t__doc__RRR(((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyRs  tDigestAuthTestscBsveZdZdZdedZdZdZdedZdedZ dZ d Z d Z d Z ded Zd ZdZdedZdZdZedZdZedZdZdZdZdZdZdZdZdZdZ dZ!dZ"d Z#d!Z$d"Z%d#Z&d$Z'd%Z(RS(&s L{TestCase} mixin class which defines a number of tests for L{DigestCredentialFactory}. Because this mixin defines C{setUp}, it must be inherited before L{TestCase}. cCsyd|_d|_d|_d|_d|_d|_d|_tdd d |_d |_ t |j|j|_ d S( s> Create a DigestCredentialFactory for testing tfoobartbazquuxs test realmRt 29fc54aa1641c6fa0e151419361c8f23tauths/write/tTCPs10.2.3.4iutGETN( tusernametpasswordtrealmt algorithmtcnoncetqopturiRt clientAddresstmethodR tcredentialFactory(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pytsetUp4s        RcCsod}t||j|j|j||j}d|j|j|jf}||j}|j||dS(s L{calcHA1} accepts the C{'md5'} algorithm and returns an MD5 hash of its parameters, excluding the nonce and cnonce. t abc123xyzs%s:%s:%sN(RR%R'R&R)t hexdigestt assertEqual(Rt _algorithmt_hashtnoncethashA1ta1texpected((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5HashA1Es cCsd}td|j|j|j||j}d|j|j|jf}t|j}d|||jf}t|j}|j||dS(s L{calcHA1} accepts the C{'md5-sess'} algorithm and returns an MD5 hash of its parameters, including the nonce and cnonce. t xyz321abcsmd5-sesss%s:%s:%sN( RR%R'R&R)RtdigestR1R2(RR5R6R7tha1R8((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5SessionHashA1RscCs|jdtdS(s L{calcHA1} accepts the C{'sha'} algorithm and returns a SHA hash of its parameters, excluding the nonce and cnonce. tshaN(R9R(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_SHAHashA1ascCsZd}t|||jdd}d||jf}||j}|j||dS(s L{calcHA2} accepts the C{'md5'} algorithm and returns an MD5 hash of its arguments, excluding the entity hash for QOP other than C{'auth-int'}. R$R"s%s:%sN(RR+tNoneR1R2(RR3R4R-thashA2ta2R8((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5HashA2Authis cCscd}d}t|||jd|}d||j|f}||j}|j||dS(s L{calcHA2} accepts the C{'md5'} algorithm and returns an MD5 hash of its arguments, including the entity hash for QOP of C{'auth-int'}. R$t foobarbazsauth-ints%s:%s:%sN(RR+R1R2(RR3R4R-thentityRARBR8((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5HashA2AuthIntvs cCs|jddS(s L{calcHA2} accepts the C{'md5-sess'} algorithm and QOP of C{'auth'} and returns the same value as it does for the C{'md5'} algorithm. smd5-sessN(RC(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5SessHashA2AuthscCs|jddS(s L{calcHA2} accepts the C{'md5-sess'} algorithm and QOP of C{'auth-int'} and returns the same value as it does for the C{'md5'} algorithm. smd5-sessN(RF(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5SessHashA2AuthIntscCs|jdtdS(s L{calcHA2} accepts the C{'sha'} algorithm and returns a SHA hash of its arguments, excluding the entity hash for QOP other than C{'auth-int'}. R>N(RCR(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_SHAHashA2AuthscCs|jdtdS(s L{calcHA2} accepts the C{'sha'} algorithm and returns a SHA hash of its arguments, including the entity hash for QOP of C{'auth-int'}. R>N(RFR(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_SHAHashA2AuthIntsc Csid}d}d}d|||f}||j}t||||ddd}|j||dS(s L{calcResponse} accepts the C{'md5'} algorithm and returns an MD5 hash of its parameters, excluding the nonce count, client nonce, and QoP value if the nonce count and client nonce are C{None} tabc123t789xyztlmnopqs%s:%s:%sN(R1R R@R2( RR3R4R6RAR5tresponseR8R;((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5HashResponses cCs|jddS(s L{calcResponse} accepts the C{'md5-sess'} algorithm and returns an MD5 hash of its parameters, excluding the nonce count, client nonce, and QoP value if the nonce count and client nonce are C{None} smd5-sessN(RO(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5SessionHashResponsescCs|jdtdS(s L{calcResponse} accepts the C{'sha'} algorithm and returns a SHA hash of its parameters, excluding the nonce count, client nonce, and QoP value if the nonce count and client nonce are C{None} R>N(ROR(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_SHAHashResponsesc Csd}d}d}d}d}d}d||||||f} || j} t|||||||} |j| | dS( s L{calcResponse} accepts the C{'md5'} algorithm and returns an MD5 hash of its parameters, including the nonce count, client nonce, and QoP value if they are specified. RKRLRMt00000004t abcxyz123R"s%s:%s:%s:%s:%s:%sN(R1R R2( RR3R4R6RAR5t nonceCountt clientNonceR*RNR8R;((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_MD5HashResponseExtrascCs|jddS(s L{calcResponse} accepts the C{'md5-sess'} algorithm and returns an MD5 hash of its parameters, including the nonce count, client nonce, and QoP value if they are specified. smd5-sessN(RV(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyt test_MD5SessionHashResponseExtrascCs|jdtdS(s L{calcResponse} accepts the C{'sha'} algorithm and returns a SHA hash of its parameters, including the nonce count, client nonce, and QoP value if they are specified. R>N(RVR(R((s@/usr/lib/python2.7/dist-packages/twisted/test/test_digestauth.pyttest_SHAHashResponseExtrascKsd|kr|j|ds