gRA@ddlmZddlmZddlmZddlmZddlZddl m Z m Z ddl m ZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZej8dZej8dZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,y))patch)get_user_model) AnonymousUser)reverseN)assertFormErrorassertRedirects) app_settings) EmailAddress) user_email user_username)context)complete_social_login) SocialAccount)signupcfd}|S)Nc<jd}t|_tj|5t ||}|j }|j jD] \}}|||< |j|cdddS#1swYyxYw)N/) getruserr request_contextrsessionitemssave)client socialloginrequestresprkvrequest_factorys e/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/allauth/socialaccount/tests/test_signup.pyfz!setup_sociallogin_flow..fs!%%c*$  $ $W - (+>DnnG--/ 1   LLN    s ABB)r r"s` r!setup_sociallogin_flowr$s  Hcfd}|S)Nct}|}t|dd}t|||jtj j ||dd d||}jd}t|_ tj|5t||}ddd||fS#1swY|fSxYw)Ntesttest@example.comTremailverifiedprimarytwitterproviderusernamer+!/accounts/twitter/login/callback/) rr r rr objectscreaterrrr rr) r1r+Userexi_userexi_user_emailrrrr sociallogin_factorys r!_email_address_clashz1email_address_clash.._email_address_clash(s6h'+8^, ##$ $ * "%%&IJ$  $ $W - ?(+>D ?} ?}s B99Cr#)r r8twitter_provider_settingsr9s`` r!email_address_clashr;$s0 r%cd|_d|_tjj|_|ddd}|||t jjditjdi}tjj||jjjsJtjj|t!|jsJy) NTr)Fr(r+email_verifiedr1)ruid)rr+r#)SOCIALACCOUNT_AUTO_SIGNUPACCOUNT_SIGNUP_FORM_CLASSaccount_settingsEmailVerificationMethodNONEACCOUNT_EMAIL_VERIFICATIONrr3rUSER_MODEL_USERNAME_FIELDrfilteraccountr?existsr r )settingsdbrr$r8rrs r!test_email_address_createdrLCs*.H&)-H&*:*R*R*W*WH'% K6;/ '>  # # ' '   5 5v > D  ' 'T{7J7J7N7N ' O V V XX X    & &D 48H & I P P RR Rr%cd|_d|_d|_dh|_d|_|dd\}}|dt dk(sJd|_dd d |_tj|5t|}d d d |dd k(sJtjjd itjd i}t!|dk(sJy #1swYXxYw)z%Test clash on both username and emailTr+r(r)locationsocialaccount_signupPOSTotherother@example.comr1r+N/accounts/profile/r#ACCOUNT_EMAIL_REQUIREDACCOUNT_UNIQUE_EMAILACCOUNT_USERNAME_REQUIREDACCOUNT_LOGIN_METHODSr@rmethodrPr rrrr3rrBUSER_MODEL_EMAIL_FIELDr )rKrrJr;rrrs r!*test_email_address_clash_username_requiredr\Vs'+H#$(H!)-H&&-YH")-H&'0BCMGT  w'=> >> >GN '2EFGL   )g  3 33 3 '>  # # ' '   2 24G H D  ' )) ) s * C  Ccd|_d|_d|_dh|_d|_|dd\}}|dt dk(sJd|_dd i|_tj|5t|}d d d |dd k(sJtjjd itjd i}t!|dk7sJy #1swYXxYw) z)Test clash while username is not requiredTFr+r(r)rNrOrPrRNrTr#rUrKrJr;rrrs r!.test_email_address_clash_username_not_requiredr_ns&*H#$(H!).H&&-YH")-H&'0BCMGT  w'=> >> >GN01GL   )g  3 33 3 '>  # # ' '   2 24G H D  & (( ( s ) C  Ccd|_d|_d|_dh|_d|_|dd\}}|ddk(sJt j jditjdi}t|dk7sJy) NTFr+r(rRrNrTr#) rVrWrXrYr@rr3rrBr[r r^s r!-test_email_address_clash_username_auto_signupras&*H#$(H!).H&&-YH")-H&'0CDMGT  3 33 3 '>  # # ' '   2 24G H D  & (( (r%cpd|_gd|_d|_d|_dh|_d|_|j d}t|_|ddd}tj|5t||ddd|jjtjvsJy#1swY2xYw) NT)r1 username1 username2r+r2r.r1zusername@example.comr/)rVACCOUNT_USERNAME_BLACKLISTrWrXrYr@rrrr rrr1rBUSERNAME_BLACKLIST)rKrJr r8r:rrs r!#test_populate_username_in_blacklistrgs'+H#*PH'$(H!)-H&&-YH")-H&!!"EFG ?GL%Z7MK   )4g{34 << (8(K(K KK K44s . B,,B5cd|_d|_d|_d|_dh|_d|_|d}||||j td}|jd}|djdk(sJ|jtddd i }t|d d tjjd }t|dk(sJt jj#|dddj%sJt jj#|d ddj%sJy)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. T mandatoryFr+zverified@example.comr+rOformunverified@example.orgdatarT)fetch_redirect_responserr*N)rVrErWrXrYr@rrr valuepostrrr3allr r rGrI) rKrrJr8r$rrrkrs r!$test_verified_email_change_at_signuprssi'+H#*5H'$(H!).H&&-YH").H&%,BCK6;/ ::g45 6D << D =   $: :: : ;;&'/ 0  DD.N   # # ' ' )! ,D d 5 55 5    & & $ '   fh       & & & '   fh   r%cd|_d|_d|_d|_dh|_d|_t }|dd}||||jtd}|jd}|djdk(sJ|jtddd i }t|td |jjd } t| d k(sJt jj#| ddd j%sJt jj#| d dd j%sJy)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. TriFr+zunverified@example.comr+r>rOrkrlrmaccount_email_verification_sentrr*N)rVrErWrXrYr@rrrr rprqrr3rrr r rGrI) rKrrJr8r$r5rrrkrs r!&test_unverified_email_change_at_signuprwsu'+H#*5H'$(H!).H&&-YH").H&  D%&uK6;/ ::g45 6D << D =   $< << < ;;&'/ 0  D D'"CDE <<   a D d 7 77 7    & & & '   fh       & & & '   fh   r%cd|_d|_d|_d|_d|_dh|_d|_t}d}|jj|}tjj||d|d}||||jtd} | jd } | dj|k(sJ|jtdd|i } t!| jd dd y) NFTrir+zme@example.comrj)r+rr,rOrkrmzAn account already exists with this email address. Please sign in to that account first, then connect your Unittest Server account.)ACCOUNT_PREVENT_ENUMERATIONrVrErWrXrYr@rr3r4r rrr rprqr) rKrr8rJr$r5r+rrrrks r!#test_unique_email_validation_signuprzs,1H(&*H#*5H'$(H!).H&&-YH").H&  D E <<  U  +De$F%,<=K6;/ ::g45 6D << D =   E )) ) ;;w56gu=M; ND V )r%cXd|_d|_d|_d|_dh|_d|_t }|dd}||||jtd}|jd}|djdk(sJ|jtdd dd  }|jd k(sJ|jjd k(sJtjjd k(sJ|jtd}t!|tdy)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. Trir+Fzme1@example.comrurOrkme1rSrmi. account_loginN)rVrErWrXrYr@rrrr rprq status_coder3countrr) rKrr8rJr$r5rrrks r!#test_social_account_taken_at_signuprs)'+H#*5H'$(H!)-H&&-YH").H&  D%,=eTK6;/ ::g45 6D << D =   $5 55 5 ;;&'*; <  D   s "" " <<   1 $$ $  & & (A -- - ::g45 6DD'/23r%cd|_d|_d|_dh|_d|_d|_|d}|||}|dt dk(sJ|jt dddi}|dt d k(sJy ) zgTests that when the email address is missing from the sociallogin email verification kicks in. TFr+ri with_emailrNrOzother@example.orgrvN)rVrWrXrYrEr@rrq)rKrJr8rr$rrs r!4test_email_address_required_missing_from_socialloginr?s '+H#$(H!).H&&-YH"*5H')-H&%7K !&+ 6D  w'=> >> > ;;w56BU8V WD  w'HI II Ir%cNd|_d|_d|_dh|_d|_d|_|}|d}|||} | dt dk(sJ|jt dd|ji} |djd k(sJ| dt d k(sJy ) yTests that when an already existing email is given at the social signup form, enumeration preventation kicks in. TFr+rirrNrOr$[example.com] Account Already ExistsrvN) rVrWrXrYrEr@rrqr+subject) rKrJ user_factoryr8rr$ mailoutboxrrrs r!1test_email_address_conflict_at_social_signup_formrTs'+H#$(H!).H&&-YH"*5H')-H& >D%7K !&+ 6D  w'=> >> > ;;w56$**8M ND a= $J JJ J  w'HI II Ir%cd|_d|_d|_dh|_d|_d|_|}||j d}|||} | dtdk(sJ|djd k(sJy ) ztTests that when an already existing email is received from the provider, enumeration preventation kicks in. TFr+ri)r+rrNrvrrN) rVrWrXrYrEr@r+rr rKrJrr8rrr$rrrs r!.test_email_address_conflict_during_auto_signuprts'+H#$(H!).H&&-YH"*5H')-H& >D%DJJ4HK !&+ 6D  w'HI II I a= $J JJ Jr%cd|_d|_d|_dh|_d|_d|_d|_|d}||jd}|||} | dtdk(sJ|jtddd i} |d jd k(sJ| d|jk(sJtjj|j jd k(sJy)rTFr+optional)r>rurNrOzother@email.orgrz/[example.com] Please Confirm Your Email Addressrjr}N)rVrWrXrYrEr@"SOCIALACCOUNT_EMAIL_AUTHENTICATIONr+rrqrLOGIN_REDIRECT_URLr r3rGrrs r!5test_email_address_conflict_removes_conflicting_emailrs'+H#$(H!).H&&-YH"*4H')-H&27H/ u -D%DJJuMK !&+ 6D  w'=> >> > ;;w56BS8T UD a= $U UU U  x:: :: :    & &TZZ & 8 > > @A EE Er%c|ddd}td5}d|_|||}ddddjvsJtjj rJy#1swY>xYw)Nr)Fr(r=zLallauth.socialaccount.adapter.DefaultSocialAccountAdapter.is_open_for_signupsSign Up Closed)r return_valuecontentrr3rI)rJrKrr$r8riofsrs r!test_signup_closedrs& K V ; !%fk: ;  ,, ,''..00 00 ;;s A%%A.)- unittest.mockrdjango.contrib.authrdjango.contrib.auth.modelsr django.urlsrpytestpytest_django.assertsrrallauth.accountr rBallauth.account.modelsr allauth.account.utilsr r allauth.corer allauth.socialaccount.helpersrallauth.socialaccount.modelsrallauth.socialaccount.viewsrfixturer$r;rLr\r_rargrsrwrzrrrrrrr#r%r!rs.4 B</; ?6.      <S&*0), )L(&R+\:!4HJ*J@K4 FF1r%