← Back
Editing: test_launchpad.cpython-38.pyc
U ��^�| � @ s� d Z eZddlmZ ddlZddlZddlZddlZddl Z ddl Z zddl m Z W n e k rt ddlm Z Y nX ddlZddlmZ ddlmZmZ ddlmZ ddlZddlmZ dd lmZmZm Z m Z m Z m Z m Z dd lm!Z! d Z"G d d � d e j#�Z$G dd� de j#�Z%G dd� de j#�Z&G dd� de j#�Z'G dd� de j#�Z(G dd� de j#�Z)G dd� de)�Z*G dd� de)�Z+G d d � d e j#�Z,G d d � d e j#�Z-d d!� Z.dS )"z Tests for the Launchpad class.� )�contextmanagerN)�patch)� ServiceRoot)� AccessToken� Credentials)�uris)� Launchpad)�assert_keyring_not_imported�BadSaveKeyring� fake_keyring�FauxSocketModule�InMemoryKeyring� NoNetworkAuthorizationEngine�NoNetworkLaunchpad)�KeyringCredentialStore�http://api.example.com/c @ s e Zd ZdZdd� ZdS )�TestResourceTypeClassesz<launchpadlib must know about restfulclient's resource types.c C s* t j�� D ]\}}| �tj| |� q d S �N)r ZRESOURCE_TYPE_CLASSES�items� assertEqualr )�self�name�cls� r �C/usr/lib/python3/dist-packages/launchpadlib/tests/test_launchpad.py�test_resource_types? s z+TestResourceTypeClasses.test_resource_typesN)�__name__� __module__� __qualname__�__doc__r r r r r r < s r c @ sl e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� ZdS )�TestNameLookupsz0Test the utility functions in the 'uris' module.c C s t dddddddg�| _d S )N� productionZ qastaging�stagingZdogfoodZdevZtest_dev�edge)�sorted�aliases�r r r r �setUpI s ��zTestNameLookups.setUpc c sd t jdd��N}t �d� d V | �t|�d� |\}| �t|jt�� | � dt |�� W 5 Q R X d S )NT��record�always� zno longer exists) �warnings�catch_warnings� simplefilterr �len� assertTrue� issubclass�category�DeprecationWarning�assertIn�str)r �caughtZwarningr r r �edge_deprecation_errorN s z&TestNameLookups.edge_deprecation_errorc C s4 | � ttj�� �| j� | � ttj�� �| j� d S r )r r$ r � service_roots�keysr% � web_rootsr& r r r �test_short_names[ s z TestNameLookups.test_short_namesc C s0 | � � � | �t�d�t�d�� W 5 Q R X d S �Nr# r! �r7 r r �lookup_service_rootr& r r r �$test_edge_service_root_is_production` s �z4TestNameLookups.test_edge_service_root_is_productionc C s0 | � � � | �t�d�t�d�� W 5 Q R X d S r< �r7 r r �lookup_web_rootr& r r r � test_edge_web_root_is_productiong s �z0TestNameLookups.test_edge_web_root_is_productionc C s2 | � � � | �t�tj�t�d�� W 5 Q R X d S �Nr! �r7 r r r> ZEDGE_SERVICE_ROOTr& r r r �-test_edge_service_root_url_becomes_productionn s �z=TestNameLookups.test_edge_service_root_url_becomes_productionc C s2 | � � � | �t�tj�t�d�� W 5 Q R X d S rC )r7 r r rA Z EDGE_WEB_ROOTr& r r r �)test_edge_web_root_url_becomes_productions s �z9TestNameLookups.test_edge_web_root_url_becomes_productionc C s2 | � � � | �t�tj�t�d�� W 5 Q R X d S rC rD r& r r r �/test_top_level_edge_constant_becomes_productionx s �z?TestNameLookups.test_top_level_edge_constant_becomes_productionc C s0 | � � � | �t�d�t�d�� W 5 Q R X d S )Nz https://api.edge.launchpad.net/r! r= r& r r r �5test_edge_server_equivalent_string_becomes_production} s �zETestNameLookups.test_edge_server_equivalent_string_becomes_productionc C s0 | � � � | �t�d�t�d�� W 5 Q R X d S )Nzhttps://edge.launchpad.net/r! r@ r& r r r �9test_edge_web_server_equivalent_string_becomes_production� s �zITestNameLookups.test_edge_web_server_equivalent_string_becomes_productionc C s� | � � �* | jD ] }| �t�|�tj| � qW 5 Q R X | � � �* | jD ] }| �t�|�tj| � qHW 5 Q R X d}| �t�|�|� | �t�|�|� d}| �t tj|� | �t tj|� dS )z=Ensure that short service names turn into long service names.z http://some-other-server.comz not-a-urlN) r7 r% r r r> r8 rA r: � assertRaises� ValueError)r �aliasZ other_rootZ not_a_urlr r r � test_lookups� s$ � �z TestNameLookups.test_lookupsN)r r r r r' r r7 r; r? rB rE rF rG rH rI rM r r r r r F s r c @ s e Zd ZdZdd� ZdS )�"TestServiceNameWithEmbeddedVersiona3 Reject service roots that include the version at the end of the URL. If the service root is "http://api.launchpad.net/beta/" and the version is "beta", the launchpadlib constructor will raise an exception. This happens with scripts that were written against old versions of launchpadlib. The alternative is to try to silently fix it (the fix will eventually break as new versions of the web service are released) or to go ahead and make a request to http://api.launchpad.net/beta/beta/, and cause an unhelpful 404 error. c C s� d}t jd | }ztd d d ||d� W n6 tk r^ } z| �t|��d�� W 5 d }~X Y n X td��|d7 }| jttd d d ||d� t j }t jd | d }| jttd d d |d� d S )Nz version-foor" )� service_root�versionzoIt looks like you're using a service root that incorporates the name of the web service version ("version-foo")z*Expected a ValueError that was not thrown!�/�rO ) r r8 r rK r0 r5 � startswith�AssertionErrorrJ r �DEFAULT_VERSION)r rP �root�eZdefault_versionr r r �'test_service_name_with_embedded_version� s* �� ��zJTestServiceNameWithEmbeddedVersion.test_service_name_with_embedded_versionN)r r r r rX r r r r rN � s rN c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �#TestRequestTokenAuthorizationEnginez4Tests for the RequestTokenAuthorizationEngine class.c C s | � ttt� d S r �rJ rK r � SERVICE_ROOTr& r r r �test_app_must_be_identified� s �z?TestRequestTokenAuthorizationEngine.test_app_must_be_identifiedc C s t tdd� d S �Nr )�application_name�r r[ r& r r r �$test_application_name_identifies_app� s zHTestRequestTokenAuthorizationEngine.test_application_name_identifies_appc C s t tdd� d S �Nr �� consumer_namer_ r& r r r �!test_consumer_name_identifies_app� s zETestRequestTokenAuthorizationEngine.test_consumer_name_identifies_appc C s, | j tttddd� | j tttddd� d S )NZname1Zname2)r^ rc r rZ r& r r r �#test_conflicting_app_identification� s � �zGTestRequestTokenAuthorizationEngine.test_conflicting_app_identificationN)r r r r r\ r` rd re r r r r rY � s rY c @ s e Zd ZdZdd� Zdd� ZdS )�%TestLaunchpadLoginWithCredentialsFilez9Tests for Launchpad.login_with() with a credentials file.c C sH t �� \}}tjd|d�}t�|�}| �|jj|j jj� t � |� d S )N� not important)r^ �credentials_file) �tempfile�mkstempr � login_withr Zload_from_pathr �consumer�key� credentials�os�remove)r �ignore�filename� launchpadrn r r r � test_filename� s � �z3TestLaunchpadLoginWithCredentialsFile.test_filenamec C s6 t �� \}}t� }| jttjd||d� t�|� d S )Nrg )r^ rh �credential_store) ri rj r rJ rK r rk ro rp )r rq rr �storer r r �+test_cannot_specify_both_filename_and_store� s �zQTestLaunchpadLoginWithCredentialsFile.test_cannot_specify_both_filename_and_storeN)r r r r rt rw r r r r rf � s rf c @ s e Zd ZdZdd� Zdd� ZdS )� KeyringTestz*Base class for tests that use the keyring.c C s t � t� tj_d S r )r r � launchpadlibrn �keyringr& r r r r' s zKeyringTest.setUpc C s t j`d S r )ry rn rz r& r r r �tearDown s zKeyringTest.tearDownN)r r r r r' r{ r r r r rx s rx c s� e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z� ZS )0�TestLaunchpadLoginWithz!Tests for Launchpad.login_with().c s t t| ��� t�� | _d S r )�superr| r' ri �mkdtemp�temp_dirr& �� __class__r r r' s z TestLaunchpadLoginWith.setUpc s t t| ��� t�| j� d S r )r} r| r{ �shutil�rmtreer r& r� r r r{ s z TestLaunchpadLoginWith.tearDownc C s� t j�| jd�}tjdt|d� | �t j�|�� t j�|d�}| �t j�|�� | �t j�t j�|d��� t j�|d�}| � t j�|�� d S )Nry rg �rO �launchpadlib_dir�api.example.com�cachern ) ro �path�joinr r rk r[ r0 �isdir� assertFalse)r r� Z service_path�credentials_pathr r r �test_dirs_created s ��z(TestLaunchpadLoginWith.test_dirs_createdc C s� t j�| jd�}t �|� t �|d� | �t j�|�� t �|�}t� |j �}| � |tj tj B tjB � tjdt|d� t �|�}t� |j �}| �|tj tj B tjB � d S )Nry i� rg r� )ro r� r� r �mkdir�chmodr0 r� �stat�S_IMODE�st_mode�assertNotEqual�S_IWRITE�S_IREAD�S_IEXECr rk r[ r �r r� Zstatinfo�moder r r �'test_dirs_created_are_changed_to_secure4 s � z>TestLaunchpadLoginWith.test_dirs_created_are_changed_to_securec C sf t j�| jd�}tjdt|d� | �t j�|�� t � |�}t � |j �}| � |t j t jB t jB � d S )Nry rg r� )ro r� r� r r rk r[ r0 r� r� r� r� r r� r� r� r� r r r � test_dirs_created_are_secureE s � z3TestLaunchpadLoginWith.test_dirs_created_are_securec C s\ t j�| jd�}tjdt|dd�}| �|jd d� tjdt|dd�}| �|jd d� d S )Nry rg �foo)rO r� rP rP Zbar) ro r� r� r r rk r[ r �passed_in_args�r r� rs r r r �test_version_is_propagatedP s � �z1TestLaunchpadLoginWith.test_version_is_propagatedc C sX t j�| jd�}tjdt|d�}| �|jj j d� tjdt|d�}| �|jj j d� d S )Nry zvery importantr� ) ro r� r� r r rk r[ r rn rl r^ r� r r r �#test_application_name_is_propagatedc s& � � � �z:TestLaunchpadLoginWith.test_application_name_is_propagatedc C s6 t td�}tj|d� | �|jd� | �|jd� d S )N�application name��authorization_enginer+ )r r[ r rk r Zrequest_tokens_obtainedZaccess_tokens_obtained�r Zenginer r r �'test_authorization_engine_is_propagatedz s � z>TestLaunchpadLoginWith.test_authorization_engine_is_propagatedc C s | � ttj� d S r )rJ rK r rk r& r r r �)test_login_with_must_identify_application� s z@TestLaunchpadLoginWith.test_login_with_must_identify_applicationc C s t jdd� d S r] �r rk r& r r r r` � s z;TestLaunchpadLoginWith.test_application_name_identifies_appc C s t jdd� d S ra r� r& r r r rd � s z8TestLaunchpadLoginWith.test_consumer_name_identifies_appc C s" t td�}| jttjd|d� dS )�;Catch an attempt to specify inconsistent application_names.zapplication name1zapplication name2r� N�r r[ rJ rK r rk r� r r r �+test_inconsistent_application_name_rejected� s � �zBTestLaunchpadLoginWith.test_inconsistent_application_name_rejectedc C s& t tddd�}| jttjd|d� dS )r� NZconsumer_name1rb Zconsumer_name2r� r� r� r r r �(test_inconsistent_consumer_name_rejected� s � �z?TestLaunchpadLoginWith.test_inconsistent_consumer_name_rejectedc C s. t tddgd�}| jttjdddg|d� dS )z=Catch an attempt to specify inconsistent allow_access_levels.rl �FOO�rc �allow_access_levelsNZBAR)rc r� r� r� r� r r r �.test_inconsistent_allow_access_levels_rejected� s � �zETestLaunchpadLoginWith.test_inconsistent_allow_access_levels_rejectedc C s4 dd� }t |d�}dd� }| jttjd||d� d S )Nc S s d S r r r r r r � callback1� s zRTestLaunchpadLoginWith.test_inconsistent_credential_save_failed.<locals>.callback1)�credential_save_failedc S s d S r r r r r r � callback2� s zRTestLaunchpadLoginWith.test_inconsistent_credential_save_failed.<locals>.callback2�app name)ru r� )r rJ rK r rk )r r� rv r� r r r �(test_inconsistent_credential_save_failed� s �z?TestLaunchpadLoginWith.test_inconsistent_credential_save_failedc C sJ t jddgd�}| �|jjjd� | �|jjjd � | �|jjdg� d S )Nrl r� r� ) r rk r rn rl rm r^ r� r� �r rs r r r � test_non_desktop_integration� s � � �z3TestLaunchpadLoginWith.test_non_desktop_integrationc C s& t jddgd�}| �|jjdg� d S )Nr� r� )r� �DESKTOP_INTEGRATION)r rk r r� r� r� r r r �<test_desktop_integration_doesnt_happen_without_consumer_name� s � �zSTestLaunchpadLoginWith.test_desktop_integration_doesnt_happen_without_consumer_namec C s� t � }t � }tjd| jt||d�}| �|jjjt j � | �|jj j d� | �|j jdg� tttj�| jdd�||tjd�}| �|j|� d S )Nr� )r� rO �timeout� proxy_infor� r� r� )rO r� r� r� rP )�objectr rk r r[ r rn � access_tokenrm r ZACCESS_TOKEN_KEYrl r^ r� r� �dictro r� r� rU r� )r r� r� rs �expected_argumentsr r r �*test_no_credentials_creates_new_credential� s4 � � � ��zATestLaunchpadLoginWith.test_no_credentials_creates_new_credentialc C s` t jd| jtd�}| �|jjjd� | �|jjjd� t j � | jddd�}| � t j � |�� dS )z)Test the anonymous login helper function.zanonymous access)r� rO � r� rn N)r Zlogin_anonymouslyr r[ r rn r� rm �secretro r� r� r� �exists)r rs r� r r r �test_anonymous_login� s � �z+TestLaunchpadLoginWith.test_anonymous_loginc C s� t �t j�| jdd�� t j�| jddd�}tddtdd�d�}|�|� t� }t� }d}t j d| jt |||d �}t t |||t j�| jdd �d �}|� � D ] \}} |j| } | �| | � q�d S ) Nr� rn r� zconsumer_secret:42z access_key:84zaccess_secret:168)Zconsumer_secretr� r� )r� rO r� r� rP r� )rO r� r� rP r� )ro �makedirsr� r� r r r Z save_to_pathr� r rk r[ r� r r� r ) r Zcredentials_file_pathrn r� r� rP rs r� rm ZexpectedZactualr r r �-test_existing_credentials_arguments_passed_on� sH � � � �� zDTestLaunchpadLoginWith.test_existing_credentials_arguments_passed_onc C s� t j�d�}| jt jd< tjdtd�}|d k r:|t jd< nt jd= |jd }t j� t j� |dd��}| � |t j� | jd�� | � t j� t j� |dd��� d S )N�HOMEr� rR r� z..z .launchpadlibr� )ro �environ�getr r rk r[ r� r� �abspathr� r r0 r� )r Zold_homers Z cache_dirr� r r r �test_None_launchpadlib_dir s( � � � �z1TestLaunchpadLoginWith.test_None_launchpadlib_dirc C sb t �dd�}| �|jd d� t �dtjd �}| �|jd tjd � d}| �tt jdd� d S )Nr� r" rO z"https://api.staging.launchpad.net/)r� zhttps://r� )r rk r r� r r8 rJ rK r� r r r �test_short_service_name* s( � �� �z.TestLaunchpadLoginWith.test_short_service_namec C s t jddd� d S )Nrg � )Zmax_failed_attemptsr� r& r r r �!test_max_failed_attempts_accepted? s �z8TestLaunchpadLoginWith.test_max_failed_attempts_accepted) r r r r r' r{ r� r� r� r� r� r� r� r` rd r� r� r� r� r� r� r� r� r� r� r� r� � __classcell__r r r� r r| s0 r| c @ s e Zd ZdZdd� Zdd� ZdS )�TestDeprecatedLoginMethodsz2Make sure the deprecated login methods still work.c C sb t jdd��L}t �d� t �dt� t�ddd� | �t|�d� | �|d jt � W 5 Q R X d S ) NTr( r* rq rl �tokenr� r+ r ) r, r- r. �PendingDeprecationWarningr Zloginr r/ r2 r3 �r r6 r r r �test_login_is_deprecatedI s z3TestDeprecatedLoginMethods.test_login_is_deprecatedc C s^ t jdd��H}t �d� t �dt� t�d� | �t|�d� | �|d jt � W 5 Q R X d S )NTr( r* rq rl r+ r ) r, r- r. r� r Zget_token_and_loginr r/ r2 r3 r� r r r �&test_get_token_and_login_is_deprecatedR s zATestDeprecatedLoginMethods.test_get_token_and_login_is_deprecatedN)r r r r r� r� r r r r r� F s r� c @ sv e Zd Zdd� Zdd� Ze�ededd� ��dd � �Z e�eded d� ��d d � �Z e�eded d� ��dd� �Z dS )�!TestCredenitialSaveFailedCallbackc C s t � tj_t�� | _d S r �r ry rs �socketri r~ r r& r r r r' ` s z'TestCredenitialSaveFailedCallback.setUpc C s t tj_ t�| j� d S r �r� ry rs r� r� r r& r r r r{ g s z*TestCredenitialSaveFailedCallback.tearDown�_is_sudoc C s dS �NFr r r r r �<lambda>k � z*TestCredenitialSaveFailedCallback.<lambda>c s` g � � fdd�}t j�| jd�}d}tt� ��( tjd|||d� | �t � �d� W 5 Q R X d S )Nc s � � d � d S r )�appendr �Zcallback_calledr r �callbackq s zPTestCredenitialSaveFailedCallback.test_credentials_save_failed.<locals>.callbackry r rg )rO r� r� r+ ) ro r� r� r r r r rk r r/ )r r� r� rO r r� r � test_credentials_save_failedk s �z>TestCredenitialSaveFailedCallback.test_credentials_save_failedc C s dS r� r r r r r r� r� c C sD t j�| jd�}d}tt� �� | jttj d||d� W 5 Q R X d S �Nry r rg r� ) ro r� r� r r r rJ � RuntimeErrorr rk �r r� rO r r r �:test_default_credentials_save_failed_is_to_raise_exception s �z\TestCredenitialSaveFailedCallback.test_default_credentials_save_failed_is_to_raise_exceptionc C s dS )NTr r r r r r� � r� c C s> t j�| jd�}d}tt� �� tjd||d� W 5 Q R X d S r� )ro r� r� r r r r rk r� r r r �>test_credentials_save_fail_under_sudo_does_not_raise_exception� s �z`TestCredenitialSaveFailedCallback.test_credentials_save_fail_under_sudo_does_not_raise_exceptionN) r r r r' r{ r r� r � staticmethodr� r� r� r r r r r� \ s r� c @ sX e Zd Zdd� Zdd� Ze�ededd� ��dd � �Z e�eded d� ��d d � �Z d S )�TestMultipleSitesc C s t � tj_t�� | _d S r r� r& r r r r' � s zTestMultipleSites.setUpc C s t tj_ t�| j� d S r r� r& r r r r{ � s zTestMultipleSites.tearDownr� c C s dS r� r r r r r r� � r� zTestMultipleSites.<lambda>c C s� t j�| jd�}t� }d}d}t|�� tj|||d�}|jj j }W 5 Q R X t |j � � �d d }| �||� | �||� | �||d | � d S )Nry r zSuper App 3000r� r r+ �@)ro r� r� r r r r rk rn rl rm �list�datar9 r4 r )r r� rz rO r^ rs rc Zapplication_keyr r r �"test_components_of_application_key� s � z4TestMultipleSites.test_components_of_application_keyc C s dS r� r r r r r r� � r� c C s� t j�| jd�}t� }|jr$td��t|��& tj dd|d� tj dd|d� W 5 Q R X t |j� � �dksnt�t |j� � �d d }t |j� � �d d }| � ||� d S ) Nry z$oops, a fresh keyring has data in itr� zhttp://alpha.example.com/r� zhttp://beta.example.com/� r r+ )ro r� r� r r r� rT r r rk r/ r9 r� r� )r r� rz Zapplication_key_1Zapplication_key_2r r r � test_same_app_different_servers� s$ � �z1TestMultipleSites.test_same_app_different_serversN) r r r r' r{ r r� r r� r� r� r r r r r� � s r� c C s t �� �t�S r )�unittestZ TestLoaderZloadTestsFromNamer r r r r � test_suite� s r� )/r �typeZ __metaclass__� contextlibr ro r� r� r� ri r� Z unittest.mockr � ImportErrorZmockr, Zlazr.restfulclient.resourcer Zlaunchpadlib.credentialsr r ry r Zlaunchpadlib.launchpadr Z launchpadlib.testing.helpersr r r r r r r r r[ ZTestCaser r rN rY rf rx r| r� r� r� r� r r r r �<module> sD $ ^- 5<B
Save File
Cancel