o b @sTdZddlZddlmZddlmZddlmZmZddl m Z Gddde Z dS) z9 Tests for L{twisted.cred}'s implementation of CRAM-MD5. N)hexlify)HMAC)CramMD5CredentialsIUsernameHashedPassword)TestCasec@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)CramMD5CredentialsTestsz* Tests for L{CramMD5Credentials}. cCs"t}|}|||dS)z The same L{CramMD5Credentials} will always provide the same challenge, no matter how many times it is called. N)r getChallenge assertEqualselfcchalrA/usr/lib/python3/dist-packages/twisted/cred/test/test_cramauth.pytest_idempotentChallengesz0CramMD5CredentialsTests.test_idempotentChallengecCs<t}|}ttd|tjd|_|| ddS)a2 When a valid response (which is a hex digest of the challenge that has been encrypted by the user's shared secret) is set on the L{CramMD5Credentials} that created the challenge, and C{checkPassword} is called with the user's shared secret, it will return L{True}. secret digestmodN) rrrrhashlibmd5digestresponse assertTrue checkPasswordr rrrtest_checkPasswordsz*CramMD5CredentialsTests.test_checkPasswordcCst}||ddS)zg When there is no response set, calling C{checkPassword} will return L{False}. rN)r assertFalser)r r rrrtest_noResponse+sz'CramMD5CredentialsTests.test_noResponsecCs<t}|}ttd|tjd|_|| ddS)a  When an invalid response is set on the L{CramMD5Credentials} (one that is not the hex digest of the challenge, encrypted with the user's shared secret) and C{checkPassword} is called with the user's correct shared secret, it will return L{False}. sthewrongsecretrrN) rrrrrrrrrrr rrrtest_wrongPassword3s z*CramMD5CredentialsTests.test_wrongPasswordc CsXt}|}|ddttd|tjdf| | d| |j ddS)z When C{setResponse} is called with a string that is the username and the hashed challenge separated with a space, they will be set on the L{CramMD5Credentials}.  ssquirrels supersecretrN) rr setResponsejoinrrrrrrrr usernamer rrrtest_setResponseAsz(CramMD5CredentialsTests.test_setResponsecCs|ttdS)zd L{CramMD5Credentials} implements the L{IUsernameHashedPassword} interface. N)rr implementedByr)r rrrtest_interfaceTsz&CramMD5CredentialsTests.test_interfaceN) __name__ __module__ __qualname____doc__rrrrr"r$rrrrrs   r) r(rbinasciirhmacrtwisted.cred.credentialsrrtwisted.trial.unittestrrrrrrs