î
'CRb  ã               @   s…   d  d l  m Z d  d l Z d  d l m Z m Z d  d l m Z m Z d  d l m	 Z	 d  d l
 m Z Gd d „  d e j ƒ Z d S)	é    )Úunicode_literalsN)ÚWebApplicationClientÚMobileApplicationClient)ÚLegacyApplicationClientÚBackendApplicationClient)ÚRequest)ÚOAuth2c               @   sL   e  Z d  Z d d „  Z d d „  Z d d „  Z d d „  Z d	 d
 „  Z d S)ÚOAuth2AuthTestc             C   sd   i d d 6d d 6d d 6|  _  d |  _ t |  j ƒ t |  j ƒ t |  j ƒ t |  j ƒ g |  _ d  S)NÚBearerÚ
token_typeZasdfoiw37850234lkjsdfsdfÚaccess_tokenZ3600Ú
expires_inÚfoo)ÚtokenÚ	client_idr   r   r   r   Úclients)Úself© r   úU/Users/cory/Documents/Python/requests_org/requests-oauthlib/tests/test_oauth2_auth.pyÚsetUp   s    	zOAuth2AuthTest.setUpc             C   s‚   d } | d |  j  d } x` |  j D]U } d | _ t d | d |  j  ƒ } t d | d | ƒj ƒ  } |  j | j | ƒ q% Wd  S)	Nz$https://example.com/resource?foo=barz&access_token=r   ÚqueryÚclientr   ÚGETÚauth)r   r   Údefault_token_placementr   r   ÚprepareÚassertEqualÚurl)r   r   Únew_urlr   r   Úrr   r   r   Útest_add_token_to_url   s    	z$OAuth2AuthTest.test_add_token_to_urlc             C   ss   d |  j  d } x[ |  j D]P } t d | d |  j  ƒ } t d d d | ƒj ƒ  } |  j | j d | ƒ q Wd  S)	NzBearer r   r   r   r   zhttps://i.br   ÚAuthorization)r   r   r   r   r   r   Úheaders)r   r   r   r   r   r   r   r   Útest_add_token_to_headers#   s
    z(OAuth2AuthTest.test_add_token_to_headersc             C   sˆ   d } | d |  j  d } xf |  j D][ } d | _ t d | d |  j  ƒ } t d d d	 | d
 | ƒj ƒ  } |  j | j | ƒ q% Wd  S)Nzfoo=barz&access_token=r   Úbodyr   r   r   zhttps://i.bÚdatar   )r   r   r   r   r   r   r   r$   )r   r$   Znew_bodyr   r   r   r   r   r   Útest_add_token_to_body*   s    	!z%OAuth2AuthTest.test_add_token_to_bodyc             C   sO   xH |  j  D]= } t d | ƒ } t d d d | ƒ} |  j t | j ƒ q
 Wd  S)Nr   r   zhttps://i.br   )r   r   r   ÚassertRaisesÚ
ValueErrorr   )r   r   r   r   r   r   r   Útest_add_nonexisting_token3   s    z)OAuth2AuthTest.test_add_nonexisting_tokenN)Ú__name__Ú
__module__Ú__qualname__r   r    r#   r&   r)   r   r   r   r   r	   
   s
   		r	   )Ú
__future__r   ÚunittestÚoauthlib.oauth2r   r   r   r   Úrequestsr   Úrequests_oauthlibr   ÚTestCaser	   r   r   r   r   Ú<module>   s   