o
    tBhH                     @   sh   d dl mZ d dl mZmZ d dl mZ d dlmZ dZdZdZ	dZ
d	Zd
Zdd Zdd Zdd ZdS )    )AccessTokenClient)CredentialsCredentialProvider)AuthorizationError)FromCodeCredentialProviderz<refresh_token>z<lwa_app_id>z<lwa_client_secret>z<aws_secret_access_key>z<aws_access_key_id>z
<role_arn>c                  C   s:   t ddd} | jdksJ | jdksJ | jdksJ d S )N   Fooi  )r   status_code
error_codemessage)e r   m/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/tests/client/test_auth.pytest_auth_exception   s   r   c                  C   sJ   t  } | jjd usJ | jjd usJ | jjd usJ | jjd us#J d S )N)r   credentials
lwa_app_idlwa_client_secretaws_secret_keyaws_access_key)xr   r   r   test_credentials   s
   r   c               
   C   s   t ttttttttddj	d} | 
d}|ddksJ z| d W n ty? } zt|ts5J W Y d }~nd }~ww z
| di i  W d S  tyd } zt|tsYJ W Y d }~d S d }~ww )N)refresh_tokenr   r   r   r   role_arn)r   foo
grant_typeauthorization_codez,https://jsonplaceholder.typicode.com/posts/1)r   r   dictr   r   r   r   r   r   r   _auth_code_request_bodygetauthorize_auth_coder   
isinstance_request)clientr   r   r   r   r   test_auth_client   s0   
r#   N)sp_api.baser   r   r   r   sp_api.base.credential_providerr   r   r   r   r   r   r   r   r   r#   r   r   r   r   <module>   s    