ó EÂWc@s3ddlmZddlZddlZddlZddlmZddlmZyddl m Z Wn!e k r‹ddl m Z nXddl mZddlmZmZddlmZdd lmZmZdd lmZmZdd lmZmZejƒZd „Zd e fd„ƒYZdS(iÿÿÿÿ(tunicode_literalsN(t b64encode(tdeepcopy(tTestCase(t urlencode(tTokenExpiredErrort OAuth2Error(tMismatchingStateError(tWebApplicationClienttMobileApplicationClient(tLegacyApplicationClienttBackendApplicationClient(t OAuth2Sessiont TokenUpdatedcs‡fd†}|S(Ncs"tjƒ}tjˆƒ|_|S(N(tmockt MagicMocktjsontdumpsttext(trtkwargstresp(ttoken(sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyt fake_sends ((RR((RsK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyt fake_tokenstOAuth2SessionTestcBsÝeZd„Zd„Zd„Zejddd„ƒd„ƒZejddd„ƒd„ƒZejddd „ƒd „ƒZ d „Z d „Z d „Z d„Z d„Zd„Zejddd„ƒd„ƒZRS(cs°tˆdƒs$‡fd†ˆ_nidd6dd6dd6d d 6td d 6ˆ_d ˆ_tˆjddƒtˆjƒtˆjƒgˆ_ˆjt ˆjƒgˆ_ dS(NuassertIncsˆj||kƒS(N(t assertTrue(tatb(tself(sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyt%suBeareru token_typeuasdfoiw37850234lkjsdfsdfu access_tokenusldvafkjw34509s8dfsdfu refresh_tokenu3600u expires_iniu expires_atufootcodeu asdf345xdf( thasattrtassertInt fake_timeRt client_idRR R tclientsR t all_clients(R((RsK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pytsetUp"s  csidˆjd‰‡‡fd†}x?ˆjD]4}td|dˆjƒ}||_|jdƒq-WdS(NuBearer u access_tokencso|jjddƒ}djdƒ|jkrF|jdjdƒ}nˆj|ˆƒtjƒ}g|_|S(Nu Authorizationuutf-8(theaderstgettNonetencodet assertEqualRRtcookes(RRt auth_headerR(RR(sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pytverifier9s  tclientRu https://i.b(RR%R tsendR((RR.R/tauth((RRsK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyttest_add_token6s   cCsÖd}t|jƒ}td|ƒ}|j|ƒ\}}|j||ƒ|j|j|ƒ|jd|ƒt|jƒ}td|ƒ}|j|ƒ\}}|j||ƒ|j|j|ƒ|jd|ƒdS(Nu%https://example.com/authorize?foo=barR/uresponse_type=codeuresponse_type=token(RR#R tauthorization_urlR!R (Rturltwebtstauth_urltstatetmobile((sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyttest_authorization_urlGsu time.timetnewcCstS(N(R"(((sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyRXsc s•tˆjƒˆ_dˆjd<ˆjd=‡fd†}x?ˆjD]4}td|dˆjƒ}ˆjt|jdƒqBWxNˆjD]C}td|dˆjdd ƒ}||_ˆjt |jdƒq„W‡fd †}xKˆjD]@}td|dˆjdd d |ƒ}||_|jdƒqäW‡fd †}xWˆjD]L}td|dˆjdd d |ƒ}||_|jdd dddƒqAWdS(Nu-1u expires_inu expires_atcsJd|jkr%ˆjd|jƒntjƒ}tjˆjƒ|_|S(Nu/refreshu Authorization( R4t assertNotInR'RRRRRR(RRR(R(sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyt fake_refresh^s  R/Ru https://i.btauto_refresh_urluhttps://i.b/refreshcsˆj|ˆjƒdS(N(R+R(R(R(sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyt token_updaterrsR?csqd|jkrLˆjd|jƒtdƒ}ˆj|jdd|ƒntjƒ}tjˆj ƒ|_ |S(Nu/refreshu Authorizationsfoo:barsBasic ( R4R!R'RR+RRRRRR(RRtencodedR(R(sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pytfake_refresh_with_auth|s  R#ufoot client_secretubar( tdictRt expired_tokenR$R t assertRaisesRR(R0R (RR=R/R1R?RA((RsK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyttest_refresh_token_requestXs4         cCstS(N(R"(((sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyRŒscCsWt|jƒ}dt|jjƒƒ}td|ƒ}|j|j|ƒ|jƒdS(Nuhttps://i.b/callback#R/(R R#RRtitemsR R+ttoken_from_fragment(RR9t response_urlR1((sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyttest_token_from_fragmentŒscCstS(N(R"(((sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyR“scCsÂd}xW|jD]L}td|d|jƒ}t|jƒ|_|j|j|ƒ|jƒqWidd6}xN|jD]C}td|d|jƒ}t|ƒ|_|jt|j|ƒqwWdS(Nuhttps://example.com/tokenR/Ruinvalid_requestuerror( R$R RRR0R+t fetch_tokenRER(RR4R/R1terror((sK/Users/singingwolfboy/clones/requests-oauthlib/tests/test_oauth2_session.pyttest_fetch_token“s  c sÆt|jƒ}tjƒd}tjƒ‰||jd<ˆd|ds$