
    fs	                     p    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y)    N)SocialAccount)OAuth2Error)OAuth2TestsMixin)MockedResponseTestCase   )BattleNetProvider)_check_errorsc                   b    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 Zy)BattleNetTestsi[zLuckyDragon#1953c                 r    | j                   | j                  d}t        dt        j                  |            S )N)	battletagid   )
_battletag_uidr   jsondumps)selfdatas     j/var/www/cs2snipe.com/venv/lib/python3.12/site-packages/allauth/socialaccount/providers/battlenet/tests.pyget_mocked_responsez"BattleNetTests.get_mocked_response   s*    !__DII>c4::d#344    c                     | j                   S )N)r   )r   s    r   get_expected_to_strz"BattleNetTests.get_expected_to_str   s    r   c                     ddi}t        dt        j                  |            }| j                  t	        |      |       y )Nr   i90  r   )r   r   r   assertEqualr
   r   r   responses      r    test_valid_response_no_battletagz/BattleNetTests.test_valid_response_no_battletag   s5    e}!#tzz$'78x0$7r   c                     t        dt        j                  i             }| j                  t              5  t        |       d d d        y # 1 sw Y   y xY w)Nr   r   r   r   assertRaisesr   r
   r   r   s     r   test_invalid_dataz BattleNetTests.test_invalid_data   sA    !#tzz"~6{+ 	$(#	$ 	$ 	$s   A

Ac                     dddd}t        dt        j                  |            }| j                  t              5  t        |       d d d        y # 1 sw Y   y xY w)Ni  	ForbiddenzAccount Inactive)codetypedetaili  r"   r   s      r   test_profile_invalid_responsez,BattleNetTests.test_profile_invalid_response#   sO    [<NO!#tzz$'78{+ 	$(#	$ 	$ 	$s   AAc                     t        j                  ddi      }t        d|      }| j                  t              5  t        |       d d d        y # 1 sw Y   y xY w)Nerrorinvalid_tokeni  )r   r   r   r#   r   r
   )r   bodyr   s      r   test_error_responsez"BattleNetTests.test_error_response+   sL    zz7O45!#t,{+ 	$(#	$ 	$ 	$s   AAc                     t        dd      }| j                  t              5  t        |       d d d        y # 1 sw Y   y xY w)NiT  z<h1>596 Service Not Found</h1>r   r#   r   r
   r$   s     r   test_service_not_foundz%BattleNetTests.test_service_not_found3   s:    !#'GH{+ 	$(#	$ 	$ 	$	   7A c                     t        dd      }| j                  t              5  t        |       d d d        y # 1 sw Y   y xY w)Nr   zinvalid json datar2   r$   s     r   test_invalid_responsez$BattleNetTests.test_invalid_response9   s:    !#':;{+ 	$(#	$ 	$ 	$r4   c                    | j                  | j                                t        j                  j	                  t        | j                              }| j                  |j                  d   | j                         | j                  |j                  d   | j                         | j                  |j                  d   d       y )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++-.''++DII+?++K8$//J++D1499=++H5t<r   N)__name__
__module____qualname__r	   r   provider_idr   r   r   r   r    r%   r+   r0   r3   r6   rA    r   r   r   r      sD    #&&KD#J58
$$$$$=r   r   )r   allauth.socialaccount.modelsr   -allauth.socialaccount.providers.oauth2.clientr   allauth.socialaccount.testsr   allauth.testsr   r   providerr	   viewsr
   r   rF   r   r   <module>rM      s)     6 E 8 2 '  8=%x 8=r   