o
    ˷e0	                     @   sj   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
 d
ee	ZdS )    N)SocialAccount)OAuth2Error)OAuth2TestsMixin)MockedResponseTestCase   )BattleNetProvider)_check_errorsc                   @   sZ   e Zd ZejZdZdZdd Zdd Z	dd Z
d	d
 Zdd Zdd Zdd Zdd ZdS )BattleNetTestsi[zLuckyDragon#1953c                 C   s   | j | jd}tdt|S )N)	battletagid   )
_battletag_uidr   jsondumps)selfdata r   f/var/www/ideatree/venv/lib/python3.10/site-packages/allauth/socialaccount/providers/battlenet/tests.pyget_mocked_response   s   z"BattleNetTests.get_mocked_responsec                 C   s,   ddi}t dt|}| t|| d S )Nr   i90  r   )r   r   r   assertEqualr	   r   r   responser   r   r    test_valid_response_no_battletag   s   z/BattleNetTests.test_valid_response_no_battletagc                 C   sH   t dti }| t t| W d    d S 1 sw   Y  d S )Nr   r   r   r   assertRaisesr   r	   r   r   r   r   r   test_invalid_data   s   
"z BattleNetTests.test_invalid_datac                 C   sT   dddd}t dt|}| t t| W d    d S 1 s#w   Y  d S )Ni  	ForbiddenzAccount Inactive)codetypedetaili  r   r   r   r   r   test_profile_invalid_response    s
   
"z,BattleNetTests.test_profile_invalid_responsec                 C   sP   t ddi}td|}| t t| W d    d S 1 s!w   Y  d S )Nerrorinvalid_tokeni  )r   r   r   r   r   r	   )r   bodyr   r   r   r   test_error_response(   s
   

"z"BattleNetTests.test_error_responsec                 C   B   t dd}| t t| W d    d S 1 sw   Y  d S )NiT  z<h1>596 Service Not Found</h1>r   r   r   r	   r   r   r   r   test_service_not_found0      

"z%BattleNetTests.test_service_not_foundc                 C   r(   )Nr   zinvalid json datar)   r   r   r   r   test_invalid_response6   r+   z$BattleNetTests.test_invalid_responsec                 C   s`   |  |   tjjt| jd}| |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   accountr   r   r   test_extra_data<   s
   zBattleNetTests.test_extra_dataN)__name__
__module____qualname__r   r   provider_idr   r   r   r   r   r#   r'   r*   r,   r6   r   r   r   r   r
      s    r
   )r   allauth.socialaccount.modelsr   -allauth.socialaccount.providers.oauth2.clientr   allauth.socialaccount.testsr   allauth.testsr   r   providerr   viewsr	   r
   r   r   r   r   <module>   s    