
    )Jf0	                     x    d dl Z d dlmZ d dlmZ d dlmZ d dlmZm	Z	 ddl
mZ ddlmZ  G d	 d
ee	          ZdS )    N)SocialAccount)OAuth2Error)OAuth2TestsMixin)MockedResponseTestCase   )BattleNetProvider)_check_errorsc                   T    e Zd Zej        ZdZdZd Zd Z	d Z
d Zd Zd Zd	 Zd
 ZdS )BattleNetTestsi[zLuckyDragon#1953c                 d    | j         | j        d}t          dt          j        |                    S )N)	battletagid   )
_battletag_uidr   jsondumps)selfdatas     r/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/allauth/socialaccount/providers/battlenet/tests.pyget_mocked_responsez"BattleNetTests.get_mocked_response   s-    !_DI>>c4:d#3#3444    c                     ddi}t          dt          j        |                    }|                     t	          |          |           d S )Nr   i90  r   )r   r   r   assertEqualr
   r   r   responses      r    test_valid_response_no_battletagz/BattleNetTests.test_valid_response_no_battletag   sH    e}!#tz$'7'788x00$77777r   c                     t          dt          j        i                     }|                     t                    5  t          |           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   assertRaisesr   r
   r   r   s     r   test_invalid_dataz BattleNetTests.test_invalid_data   s    !#tz"~~66{++ 	$ 	$(###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$s   AA!Ac                     dddd}t          dt          j        |                    }|                     t                    5  t          |           d d d            d S # 1 swxY w Y   d S )Ni  	ForbiddenzAccount Inactive)codetypedetaili  r    r   s      r   test_profile_invalid_responsez,BattleNetTests.test_profile_invalid_response    s    [<NOO!#tz$'7'788{++ 	$ 	$(###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$s   A  A$'A$c                     t          j        ddi          }t          d|          }|                     t                    5  t          |           d d d            d S # 1 swxY w Y   d S )Nerrorinvalid_tokeni  )r   r   r   r!   r   r
   )r   bodyr   s      r   test_error_responsez"BattleNetTests.test_error_response(   s    z7O455!#t,,{++ 	$ 	$(###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$s   AA"%A"c                     t          dd          }|                     t                    5  t          |           d d d            d S # 1 swxY w Y   d S )NiT  z<h1>596 Service Not Found</h1>r   r!   r   r
   r"   s     r   test_service_not_foundz%BattleNetTests.test_service_not_found0   s    !#'GHH{++ 	$ 	$(###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$   AAAc                     t          dd          }|                     t                    5  t          |           d d d            d S # 1 swxY w Y   d S )Nr   zinvalid json datar0   r"   s     r   test_invalid_responsez$BattleNetTests.test_invalid_response6   s    !#':;;{++ 	$ 	$(###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$r2   c                    |                      |                                            t          j                            t          | j                            }|                     |j        d         | j	                   |                     |j        d         | j                   |                     |j        d         d           d S )N)uidr   r   regionus)
loginr   r   objectsgetstrr   r   
extra_datar   )r   accounts     r   test_extra_datazBattleNetTests.test_extra_data<   s    

4++--...'++DI+??+K8$/JJJ+D149===+H5t<<<<<r   N)__name__
__module____qualname__r	   r   provider_idr   r   r   r   r#   r)   r.   r1   r4   r?    r   r   r   r      s        #&KD#J5 5 58 8 8
$ $ $$ $ $$ $ $$ $ $$ $ $= = = = =r   r   )r   allauth.socialaccount.modelsr   -allauth.socialaccount.providers.oauth2.clientr   allauth.socialaccount.testsr   allauth.testsr   r   providerr	   viewsr
   r   rD   r   r   <module>rK      s     6 6 6 6 6 6 E E E E E E 8 8 8 8 8 8 2 2 2 2 2 2 2 2 ' ' ' ' ' '            5= 5= 5= 5= 5=%x 5= 5= 5= 5= 5=r   