o gf@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z dZedZGdd d ejZGd d d eZGd d d eZGdddeZGdddeZGdddejZdS)N) SGMLParser)settings)Clientreversez Mozilla/5.0z \d;\s*url=csteZdZdZdZdZfddZfddZfddZd d Z dd d Z dddZ ddZ ddZ ddZZS)SocialAuthTestsCasez Base class for social auth testsNcsTi}|jr |j|d<|jr|j|d<d|_tdi||_tt|j|i|dS)N SERVER_NAME SERVER_PORT)rr jarrclientsuperr__init__)selfargskwargs client_kwargs __class__r W/var/www/html/humari/django-venv/lib/python3.10/site-packages/social_auth/tests/base.pyrs  zSocialAuthTestsCase.__init__cs,ddlm}|j|_d|_tt|dS)Nrbackends)z5social_auth.backends.pipeline.social.social_auth_userz:social_auth.backends.pipeline.associate.associate_by_emailz/social_auth.backends.pipeline.user.get_usernamez.social_auth.backends.pipeline.user.create_userz3social_auth.backends.pipeline.social.associate_userz4social_auth.backends.pipeline.social.load_extra_dataz6social_auth.backends.pipeline.user.update_user_details) social_authrPIPELINE old_PIPELINEr rsetUprrrr rr!s  zSocialAuthTestsCase.setUpcs&ddlm}|j|_tt|dS)Nrr)rrrrr rtearDownrrr rr/s zSocialAuthTestsCase.tearDowncCsnz|jWn tyYdSw|jtjvrdSddlm}i|_|j| d|j| |j|j vdS)z5Ensure that the backend for the testcase gets cached.Nrrsocialauth_begin) nameAttributeErrorrSOCIAL_AUTH_ENABLED_BACKENDSrrBACKENDSr getr assertTrue BACKENDSCACHErr r rtest_backend_cache4s    z&SocialAuthTestsCase.test_backend_cacheFcCsd|r tj|ddp |}t|}t}|r |t||dt d |j ||d S)yReturn content for given url, if data is not None, then a POST request will be issued, otherwise GET will be usedTdoseq User-Agentdata) urllib urlencodeurllib2Request build_opener add_handlerHTTPCookieProcessorget_jar add_header USER_AGENTjoinopen readlinesrurlr- use_cookiesrequestagentr r r get_contentCs  zSocialAuthTestsCase.get_contentcCs^|r tj|ddp |}t|}tt}|r"|t|| dt |j ||dS)r'Tr(r*r,) r.r/r0r1r2RedirectHandlerr3r4r5r6r7r9r;r r r get_redirectOs   z SocialAuthTestsCase.get_redirectcCs|jst|_|jSN)r cookielib CookieJarrr r rr5[s zSocialAuthTestsCase.get_jarcCst||fdS)zReverses backend URL by name)rr)rrbackendr r rr`szSocialAuthTestsCase.reversecCs(t|}tdd|j|j|j|jfS)z5Converst URL to relative, useful for server responsesr+)urlparse urlunparsepathparamsqueryfragment)rvalueparsedr r r make_relativeds z!SocialAuthTestsCase.make_relativeNF)__name__ __module__ __qualname____doc__rr rrrr&r@rBr5rrP __classcell__r r rrrs     rc@seZdZdZddZdS) CustomParserz6Custom SGMLParser that closes the parser once it's fedcCst|||dSrC)rfeedclose)rr-r r rrXms  zCustomParser.feedN)rRrSrTrUrXr r r rrWks rWc@s8eZdZdZddZddZddZdd Zd d Zd S) FormParserz5Form parser, load form data and action for given formcOs.tj|g|Ri|d|_d|_i|_dSrQ)rWr inside_formactionvaluesrrrr r rrts zFormParser.__init__cCs,t|}||rd|_|d|_dSdS)z9Start form parsing detecting if form is the one requestedTr\N)dictin_formr[r#r\r attributesattrsr r r start_formzs  zFormParser.start_formcCsdS)zOverride belowTr rrcr r rr`szFormParser.in_formcCs d|_dS)z(End form parsing, unset inside_form flagFN)r[rFr r rend_forms zFormParser.end_formcCsVt|}|jr%|d|d|d}}}|r'|dvr)||j|<dSdSdSdS)ztParse input fields, we only keep data for fields of type text, hidden or password and that has a valid name.typerrN)texthiddenpasswordN)r_r[r#r])rrbrcrgrrNr r r start_inputs zFormParser.start_inputN) rRrSrTrUrrdr`rfrkr r r rrZrs rZc@ eZdZdZddZddZdS)FormParserByIDzNForm parser, load form data and action for given form identified by its idcOs"tj|g|Ri|||_dSrC)rZrform_id)rrnrrr r rr zFormParserByID.__init__cCs|d|jkS)Nid)r#rnrer r rr`szFormParserByID.in_formN)rRrSrTrUrr`r r r rrms rmc@rl) RefreshParserzDRefresh parser, will check refresh by meta tag and store refresh URLcOs"tj|g|Ri|d|_dSrC)rWrrNr^r r rrrozRefreshParser.__init__cCs8t|}|ddkrtd|dd|_dSdS)z3Start meta parsing checking by http-equiv attributez http-equivrefreshr+content'N)r_r# REFRESH_REsubstriprNrar r r start_metaszRefreshParser.start_metaN)rRrSrTrUrrxr r r rrqs rqc@seZdZddZdS)rAcCs|SrCr )rreqfpcodemsgheadersr r rhttp_error_302szRedirectHandler.http_error_302N)rRrSrTr~r r r rrAs rA)rer0rDr.rHunittestsgmllibr django.confrdjango.test.clientrdjango.core.urlresolversrr7compileruTestCaserrWrZrmrqHTTPRedirectHandlerrAr r r rs$     Y#