o LfjO@sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlmZddlm Z m!Z!m"Z"ddl#m$Z$ddl%m&Z&Gddde&Z'ddZ(ddZ)ddZ*ddZ+dS)N)get_user_model) AnonymousUser)MessageMiddleware)SessionMiddleware)RequestFactory)override_settings)reverse) app_settings) EmailAddress) user_email user_username)context) providers)complete_social_login) SocialAccount SocialApp SocialLogin)signup)TestCasec seZdZfddZeddejjdddZedddddd d d Z eddd ddd d dZ eddd ddd ddZ edgddddddddZ ddZ edddd dd dddZedddd dd dddZed dddd dd dddZedddddd dd d!ZZS)" SignupTestscsnttjD]*}|jdkrq tjj|j|jdddd}t j j r4ddl m }|j}|j|q dS)Nopenid_connectapp123id123dummy)providername client_idkeysecretr)Site)supersetUprregistryget_class_listidrobjectscreateallauthr SITES_ENABLEDdjango.contrib.sites.modelsr get_currentsitesadd)selfrapprsite __class__h/var/www/html/humari/django-venv/lib/python3.10/site-packages/allauth/socialaccount/tests/test_signup.pyr!s"     zSignupTests.setUpTN)SOCIALACCOUNT_AUTO_SIGNUPACCOUNT_SIGNUP_FORM_CLASSACCOUNT_EMAIL_VERIFICATIONcCst}|d}t|_tdd|tdd|t}|}t|t j dt|t j dt ddd }t ||d }t| t||Wdn1sSwY|jjd it j di}|t jj||jd |tjj|t|d dS)Nz/accounts/login/callback/cSdSNr2requestr2r2r32z8SignupTests.test_email_address_created..cSr7r8r2r9r2r2r3r;3r<testtest@example.comopenidrruiduseraccount)rCrA)rCemailr2)rgetrrCrprocess_requestrrsetattraccount_settingsUSER_MODEL_USERNAME_FIELDUSER_MODEL_EMAIL_FIELDrrr request_contextrr% assertTruefilterrAexistsr r r-factoryr:UserrCrD socialloginr2r2r3test_email_address_created)s*     z&SignupTests.test_email_address_createdrE)ACCOUNT_EMAIL_REQUIREDACCOUNT_UNIQUE_EMAILACCOUNT_USERNAME_REQUIREDACCOUNT_AUTHENTICATION_METHODr4cCs|dd\}}||dtdd|_ddd|_t| t|}Wd n1s/wY||dd tj j d it j di}|t |dd S) z%Test clash on both username and emailr=r>locationsocialaccount_signupPOSTotherother@example.comusernamerEN/accounts/profile/r2)_email_address_clash assertEqualrmethodr[r rLrrr%rFrIrKr r-r:resprCr2r2r3*test_email_address_clash_username_requiredGs     z6SignupTests.test_email_address_clash_username_requiredFcCs|dd\}}||dtdd|_ddi|_t| t|}Wdn1s.wY||dd tj j d it j di}| t|ddS) z)Test clash while username is not requiredr=r>rYrZr[rEr]Nr`r2)rarbrrcr[r rLrrr%rFrIrKassertNotEqualr rdr2r2r3.test_email_address_clash_username_not_required^s     z:SignupTests.test_email_address_clash_username_not_requiredcCsN|dd\}}||ddtjjditjdi}|t|ddS)Nr=r]rYr`r2) rarbrr%rFrIrKrgr rdr2r2r3-test_email_address_clash_username_auto_signupus   z9SignupTests.test_email_address_clash_username_auto_signup)r_ username1 username2)rUACCOUNT_USERNAME_BLACKLISTrVrWrXr4cCst}|d}t|_tdd|tdd|t}|}t|t j dt|t j dt ddd }t ||d }t| t||Wdn1sSwY||jjt jdS) N!/accounts/twitter/login/callback/cSr7r8r2r9r2r2r3r;r<zASignupTests.test_populate_username_in_blacklist..cSr7r8r2r9r2r2r3r;r<r_zusername@example.comtwitterrr@rB)rrFrrCrrGrrrHrIrJrKrrr rLr assertNotInr_USERNAME_BLACKLISTrPr2r2r3#test_populate_username_in_blacklists&     z/SignupTests.test_populate_username_in_blacklistc Cst}|}t|dd}t|||tjj||dddtddd}|}t||t||t||d}t } | d } t | _ t d d | td d | t| t| |} Wd| | fS1sowY| | fS) Nr=r>TrCrEverifiedprimaryrnrr@rBrmcSr7r8r2r9r2r2r3r;r<z2SignupTests._email_address_clash..cSr7r8r2r9r2r2r3r;r<)rr r saver r%r&rrrrFrrCrrGrr rLr) r-r_rErRexi_userexi_user_emailrDrCrSrQr:rer2r2r3ras2          z SignupTests._email_address_clash mandatory)rUr6rVrWrXr4cCs|jj}t}t|ddtddtddddgd}||d<||jt d }|j d }| |d  d|jj t d d d id }|j|ddd|jd}| t|d|tjj|dddd|tjj|d ddddS)z Test scenario for when the user changes email at social signup. Current behavior is that both the unverified and verified email are added, and that the user is allowed to pass because he did provide a verified one. zverified@example.comrEgooglerTrErsrtrCrDemail_addressessocialaccount_socialloginrZformrEunverified@example.orgdatar`F)fetch_redirect_responserrrNclientsessionrrrr serializerurFrr rbvaluepostassertRedirectsr%allr rMrNrOr-rrRrSrerrCr2r2r3$test_verified_email_change_at_signupsL   z0SignupTests.test_verified_email_change_at_signupcCs|jj}t}t|ddtddtddddgd}||d <||jt d }|j d }| |d  d|jj t d d d id}||t d|jd}| t|d |tjj|dddd|tjj|d ddddS)z Test scenario for when the user changes email at social signup, while his provider did not provide a verified email. In that case, email verification will kick in. Here, both email addresses are added as well. zunverified@example.comryrzr{FTr|r}rrZrrErraccount_email_verification_sentrrrNrrr2r2r3&test_unverified_email_change_at_signupsT  z2SignupTests.test_unverified_email_change_at_signup)ACCOUNT_PREVENT_ENUMERATIONrUr6rVrWrXr4cCs|jj}t}d}|jj|d}tjj||ddt|ddtddtddddgd}||d <| |j t d }|j d }| |d ||jjt d d |id }tjdkrj||j d d ddS||d d ddS)Nzme@example.comryT)rErCrsrzr{r|r}rrZrrEr)zzAn account already exists with this email address. Please sign in to that account first, then connect your Google account.)rrrr%r&r rrrrurFrr rbrrdjangoVERSIONassertFormError)r-rrRrErCrSrerr2r2r3#test_unique_email_validation_signup/s:     z/SignupTests.test_unique_email_validation_signupcCs|jj}t}t|ddtddd}||d<||jtd}|j d}| |d  d|jj tdd dd d }| |j d | |jd| tjd|jtd}||tddS)a Test scenario for when the user signs up with a social account and uses email address in that social account. But upon seeing the verification screen, they realize that email address is somehow unusable for them, and so backs up and enters a different email address (and is forced to choose a new username) while providing the same social account token which is owned by their first attempt. zme1@example.comryfacebookr{rBrrZrrEme1r^ri.r account_loginN)rrrrrrrurFrr rbrr status_coder%countr)r-rrRrSrerr2r2r3#test_social_account_taken_at_signup]s(  z/SignupTests.test_social_account_taken_at_signup)__name__ __module__ __qualname__r!rrIEmailVerificationMethodNONErTrfrhrirqrarrrr __classcell__r2r2r0r3rs        , 2 %rc Csd|_d|_d|_d|_d|_d|_|dd}|d}i|_t|_ t ||}|dt dks2J|j}| |d <| |t ddd i}|dt d ksSJd S) zgTests that when the email address is missing from the sociallogin email verification kicks in. TFrErx with_email/rYrZrzother@example.orgrN)rUrVrWrXr6r4rFrrrCrrrrur) dbsettingssociallogin_factoryrrfrSr:rerr2r2r34test_email_address_required_missing_from_sociallogins"    rc Csd|_d|_d|_d|_d|_d|_|}|dd}|d} i| _t| _ t | |} | dt dks5J|j} | | d <| |t dd|ji} |d jd ksVJ| dt d ks`Jd S)yTests that when an already existing email is given at the social signup form, enumeration preventation kicks in. TFrErxrrrYrZrr$[example.com] Account Already ExistsrN)rUrVrWrXr6r4rFrrrCrrrrurrEsubject rr user_factoryrrr mailoutboxrCrSr:rerr2r2r31test_email_address_conflict_at_social_signup_forms&    rc Csd|_d|_d|_d|_d|_d|_|}||jdd}|d} i| _t | _ t | |} | dt dks7J|d j d ks@Jd S) ztTests that when an already existing email is received from the provider, enumeration preventation kicks in. TFrErx)rErrrYrrrN)rUrVrWrXr6r4rErFrrrCrrr) rrrrrrrrCrSr:rer2r2r3.test_email_address_conflict_during_auto_signups  rc Csd|_d|_d|_d|_d|_d|_d|_|dd}||jdd}|d} i| _ t | _ t | |} | dt d kss2              q!