�
�w�g�$ � �0 � S r SSKJr SSKJrJr SSKrSSKrSSKJ r SSK
J r SSK J
r
Jr SSKJr S r\" 0 5 R% S
5 r " S S 5 r " S
S\5 r " S S\
5 r " S S\R. 5 r " S S\5 r " S S\5 rg)z,Tests for the LaunchpadOAuthAwareHTTP class.� )�deque)�dumps�JSONDecodeErrorN)� Unauthorized)� UnencryptedFileCredentialStore)� Launchpad�LaunchpadOAuthAwareHttp)� NoNetworkAuthorizationEngines� <?xml version="1.0"?>
<application xmlns="http://research.sun.com/wadl/2006/10">
<resources base="http://www.example.com/">
<resource path="" type="#service-root"/>
</resources>
<resource_type id="service-root">
<method name="GET" id="service-root-get">
<response>
<representation href="#service-root-json"/>
</response>
</method>
</resource_type>
<representation id="service-root-json" mediaType="application/json"/>
</application>
zutf-8c � � \ rS rSrSrS rSrg)�Response�8 z A fake HTTP response object.c � � Xl X l g �N)�status�content)�selfr r s �>/usr/lib/python3/dist-packages/launchpadlib/tests/test_http.py�__init__�Response.__init__; s � � � � � � )r r N)�__name__�
__module__� __qualname__�__firstlineno__�__doc__r �__static_attributes__� r r r r 8 s
� �&� r r c �2 ^ � \ rS rSrSrU 4S jrS rSrU =r$ )�SimulatedResponsesHttp�@ z<Responds to HTTP requests by shifting responses off a stack.c �H >� [ TU ] " U6 / U l Xl SU l g)zcConstructor.
:param responses: A list of HttpResponse objects to use
in response to requests.
N)�superr �sent_responses�unsent_responses�cache)r � responses�args� __class__s �r r � SimulatedResponsesHttp.__init__C s' �� � ���$� � ��� )� ���
r c � � U R R 5 nU R R U5 U R " X"R
/UQ76 $ r )r$ �popleftr# �append�retry_on_bad_tokenr )r r' �responses r �_request� SimulatedResponsesHttp._requestN sF � ��(�(�0�0�2�� ���"�"�8�,��&�&�x�1A�1A�I�D�I�Ir )r% r# r$ ) r r r r r r r/ r �
__classcell__)r( s @r r r @ s �� �F� �J� Jr r c �. � \ rS rSr/ rS r\S 5 rSrg)�SimulatedResponsesLaunchpad�T c �V � [ [ U R 5 X R /UQ76 $ r )r r r& �authorization_engine)r r' s r � httpFactory�'SimulatedResponsesLaunchpad.httpFactoryZ s+ � �%� �$�.�.� !�4�)B�)B�
�EI�
�
r c �F � [ [ R " 5 S U5 $ )N� )r �tempfile�mkstemp)�cls�credential_save_faileds r �credential_store_factory�4SimulatedResponsesLaunchpad.credential_store_factory_ s# � �-� � � � �q� !�#9�
�
r r N) r r r r r& r7 � classmethodr? r r r r r3 r3 T s" � � �I�
�
�
� �
r r3 c �$ � \ rS rSrSrS rS rSrg)�SimulatedResponsesTestCase�f zATest cases that give fake responses to launchpad's HTTP requests.c �< � / [ l [ SS5 U l g)z*Clear out the list of simulated responses.zhttp://api.example.com/�application nameN)r3 r& r
�engine�r s r �setUp� SimulatedResponsesTestCase.setUpi s � �02�#�-�2� %�'9�
�� r c �T � U[ l [ R SU R S9$ )aA Use simulated HTTP responses to get a Launchpad object.
The given Response objects will be sent, in order, in response
to launchpadlib's requests.
:param responses: Some number of Response objects.
:return: The Launchpad object, assuming that errors in the
simulated requests didn't prevent one from being created.
rF �r6 )r3 r& �
login_withrG )r r& s r �launchpad_with_responses�3SimulatedResponsesTestCase.launchpad_with_responsesp s. � � 1:�#�-�*�5�5� �T�[�[� 6�
�
r )rG N)r r r r r rI rN r r r r rC rC f s � �K�
�
r rC c �* � \ rS rSrSrS rS rS rSrg)�TestAbilityToParseData� aG Test launchpadlib's ability to handle the sample data.
To create a Launchpad object, two HTTP requests must succeed and
return usable data: the requests for the WADL and JSON
representations of the service root. This test shows that the
minimal data in SIMPLE_WADL and SIMPLE_JSON is good enough to
create a Launchpad object.
c �` � U R [ S[ 5 [ S[ 5 5 g)z5Make sure that launchpadlib can use the minimal data.�� N)rN r � SIMPLE_WADL� SIMPLE_JSONrH s r �test_minimal_data�(TestAbilityToParseData.test_minimal_data� s# � � �%�%� �S�+� &���k�(B�
r c �x � U R [ U R [ SS5 [ S[ 5 5 g)z'Show that bad WADL causes an exception.rT s This is not WADL.N)� assertRaises� SyntaxErrorrN r rV rH s r �
test_bad_wadl�$TestAbilityToParseData.test_bad_wadl� s1 � � ��� � � )� )� �S�.� /� �S�+� &�
r c �x � U R [ U R [ S[ 5 [ SS5 5 g)z'Show that bad JSON causes an exception.rT s This is not JSON.N)rZ r rN r rU rH s r �
test_bad_json�$TestAbilityToParseData.test_bad_json� s1 � � ��� � � )� )� �S�+� &� �S�.� /�
r r N) r r r r r rW r\ r_ r r r r rQ rQ � s � ��
�
�
r rQ c �B � \ rS rSrSrS rS rS rS rS r S r
S r S
r g ) � TestTokenFailureDuringRequest� a Test access token failures during a request.
launchpadlib makes two HTTP requests on startup, to get the WADL
and JSON representations of the service root. If Launchpad
receives a 401 error during this process, it will acquire a fresh
access token and try again.
c �* � [ S[ 5 [ S[ 5 /[ l U R U R R S5 [ R SU R S9 U R U R R S5 g)z/If our token is good, we never get another one.rT r rF rL r: N� r rU rV r3 r& � assertEqualrG �access_tokens_obtainedrM rH s r �test_good_token�-TestTokenFailureDuringRequest.test_good_token� sw � �
�S�+� &� �S�+� &�1
�#�-�
������;�;�Q�?�#�.�.� �T�[�[� /�
�
������;�;�Q�?r c �@ � [ SS5 [ S[ 5 [ S[ 5 /[ l U R U R R S5 [ R SU R S9 U R U R R S5 g) z(If our token is bad, we get another one.� � Invalid token.rT r rF rL � Nre rH s r �test_bad_token�,TestTokenFailureDuringRequest.test_bad_token� s� � �
�S�+� ,� �S�+� &� �S�+� &�1
�#�-�
������;�;�Q�?�#�.�.� �T�[�[� /�
�
������;�;�Q�?r c �@ � [ SS5 [ S[ 5 [ S[ 5 /[ l U R U R R S5 [ R SU R S9 U R U R R S5 g) z,If our token is expired, we get another one.rk � Expired token.rT r rF rL rm Nre rH s r �test_expired_token�0TestTokenFailureDuringRequest.test_expired_token� s� � �
�S�+� ,� �S�+� &� �S�+� &�1
�#�-�
������;�;�Q�?�#�.�.� �T�[�[� /�
�
������;�;�Q�?r c �@ � [ SS5 [ S[ 5 [ S[ 5 /[ l U R U R R S5 [ R SU R S9 U R U R R S5 g) z,If our token is unknown, we get another one.rk s Unknown access token.rT r rF rL rm Nre rH s r �test_unknown_token�0TestTokenFailureDuringRequest.test_unknown_token� s� � �
�S�2� 3� �S�+� &� �S�+� &�1
�#�-�
������;�;�Q�?�#�.�.� �T�[�[� /�
�
������;�;�Q�?r c �@ � [ S[ 5 [ SS5 [ S[ 5 /[ l U R U R R S5 [ R SU R S9 U R U R R S5 g) z6We get another token no matter when the error happens.rT rk rq r rF rL rm Nre rH s r �test_delayed_error�0TestTokenFailureDuringRequest.test_delayed_error� s� � �
�S�+� &� �S�+� ,� �S�+� &�1
�#�-�
������;�;�Q�?�#�.�.� �T�[�[� /�
�
������;�;�Q�?r c �l � [ SS5 [ S[ 5 [ SS5 [ SS5 [ S[ 5 /[ l U R U R R S5 [ R SU R S9 U R U R R S5 g )
z@We'll keep getting new tokens as long as tokens are the problem.rk rl rT rq r rF rL � Nre rH s r �test_many_errors�.TestTokenFailureDuringRequest.test_many_errors� s� � �
�S�+� ,� �S�+� &� �S�+� ,� �S�+� ,� �S�+� &� 1
�#�-�
������;�;�Q�?�#�.�.� �T�[�[� /�
�
������;�;�Q�?r c � � [ SS5 /[ l U R [ [ R
SU R S9 g)z>If the token is not at fault, a 401 error raises an exception.rk s Some other error.rF rL N)r r3 r& rZ r rM rG rH s r �test_other_unauthorized�5TestTokenFailureDuringRequest.test_other_unauthorized sE � �
�S�.� /�1
�#�-�
��� � '� 2� 2� �!%��� �
r r N)
r r r r r rh rn rr ru rx r| r r r r r rb rb � s0 � �� @� @�
@�
@� @�
@�
r rb )r � collectionsr �jsonr r r; �unittest�launchpadlib.errorsr �launchpadlib.credentialsr �launchpadlib.launchpadr r � launchpadlib.testing.helpersr
rU �encoderV r r r3 �TestCaserC rQ rb r r r �<module>r� s� ��" 3� � '� � � ,� C�� F�� �& �B�i� � �w�'� � � �J� 4� J�(
�)�
�$
��!2�!2�
�4
� 7�
�Fk
�$>� k
r