a
    a]0	                     @   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   g/var/www/html/Ranjet/env/lib/python3.9/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   sF   t dti }| t t| W d    n1 s80    Y  d S )Nr   r   r   r   assertRaisesr   r	   r   r   r   r   r   test_invalid_data   s    z BattleNetTests.test_invalid_datac                 C   sR   dddd}t dt|}| t t| W d    n1 sD0    Y  d S )Ni  Z	ForbiddenzAccount Inactive)codetypeZdetaili  r   r   r   r   r   test_profile_invalid_response    s    z,BattleNetTests.test_profile_invalid_responsec                 C   sN   t ddi}td|}| t t| W d    n1 s@0    Y  d S )NerrorZinvalid_tokeni  )r   r   r   r   r   r	   )r   bodyr   r   r   r   test_error_response(   s    
z"BattleNetTests.test_error_responsec                 C   s@   t dd}| t t| W d    n1 s20    Y  d S )NiT  z<h1>596 Service Not Found</h1>r   r   r   r	   r   r   r   r   test_service_not_found0   s    
z%BattleNetTests.test_service_not_foundc                 C   s@   t dd}| t t| W d    n1 s20    Y  d S )Nr   zinvalid json datar%   r   r   r   r   test_invalid_response6   s    
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   Zregionus)
Zloginr   r   objectsgetstrr   r   
extra_datar   )r   Zaccountr   r   r   test_extra_data<   s
    zBattleNetTests.test_extra_dataN)__name__
__module____qualname__r   r   Zprovider_idr   r   r   r   r   r!   r$   r&   r'   r.   r   r   r   r   r
      s   r
   )r   Zallauth.socialaccount.modelsr   Z-allauth.socialaccount.providers.oauth2.clientr   Zallauth.socialaccount.testsr   Zallauth.testsr   r   Zproviderr   Zviewsr	   r
   r   r   r   r   <module>   s   