U 2ܟj @sTdZddlZddlmZddlmZddlmZmZddl m Z Gddde Z dS) z9 Tests for L{twisted.cred}'s implementation of CRAM-MD5. N)hexlify)HMAC)CramMD5CredentialsIUsernameHashedPassword)TestCasec@sdeZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ dS)CramMD5CredentialsTestsz* Tests for L{CramMD5Credentials}. N)returncCs"t}|}|||dS)z The same L{CramMD5Credentials} will always provide the same challenge, no matter how many times it is called. N)r getChallenge assertEqualselfcchalrC/tmp/pip-unpacked-wheel-4b529toj/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) rr rrhashlibmd5digestresponse 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) rr rrrrrrrrr 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 Z setResponsejoinrrrrrrrr usernamer rrrtest_setResponseAsz(CramMD5CredentialsTests.test_setResponsecCs|ttdS)zd L{CramMD5Credentials} implements the L{IUsernameHashedPassword} interface. N)rrZ implementedByr)r rrrtest_interfaceTsz&CramMD5CredentialsTests.test_interface) __name__ __module__ __qualname____doc__rrrrr"r#rrrrrs  r) r'rbinasciirhmacrZtwisted.cred.credentialsrrZtwisted.trial.unittestrrrrrrs