← Back
Editing: test_credential_store.cpython-38.pyc
U H�9^�" � @ s� d Z ddlZddlZddlZddlmZ eekr6eZ neZ ddl m Z m Z ddl mZmZmZmZ G dd� dej�ZG dd � d ej�ZG d d � d e�ZG d d � d e�ZdS )z'Tests for the credential store classes.� N)� b64decode)� fake_keyring�InMemoryKeyring)� AccessToken� Credentials�KeyringCredentialStore� UnencryptedFileCredentialStorec @ s e Zd ZdZdd� Zdd� ZdS )�TestAccessTokenz Tests for the AccessToken class.c C s6 t �d�}| �d|j� | �d|j� | �|j� d S )Nz;oauth_token_secret=secret%3Dpassword&oauth_token=lock%26key�lock&key�secret=password)r � from_string� assertEqual�key�secret� assertIsNone�context��self� access_token� r �J/usr/lib/python3/dist-packages/launchpadlib/tests/test_credential_store.py�test_from_string/ s �z TestAccessToken.test_from_stringc C s8 t �d�}| �d|j� | �d|j� | �d|j� d S )NzNoauth_token_secret=secret%3Dpassword&oauth_token=lock%26key&lp.context=firefoxr r Zfirefox)r r r r r r r r r r � test_from_string_with_context6 s �z-TestAccessToken.test_from_string_with_contextN)�__name__� __module__� __qualname__�__doc__r r r r r r r , s r c @ s e Zd Zdd� ZdS )�CredentialStoreTestCasec C s t ddt|d�d�S )z(Helper method to make a fake credential.zapp namezconsumer_secret:42zaccess_secret:168)Zconsumer_secretr )r r )r Z consumer_keyr r r �make_credentialA s �z'CredentialStoreTestCase.make_credentialN)r r r r r r r r r ? s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �"TestUnencryptedFileCredentialStorez3Tests for the UnencryptedFileCredentialStore class.c C s t �� \}| _t| j�| _d S �N)�tempfileZmkstemp�filenamer �store)r �ignorer r r �setUpK s z(TestUnencryptedFileCredentialStore.setUpc C s t j�| j�rt �| j� d S r )�os�path�existsr" �remove�r r r r �tearDownO s z+TestUnencryptedFileCredentialStore.tearDownc C s< | � d�}| j�|d� | j�d�}| �|jj|jj� d S �N� consumer key� unique key�r r# �save�loadr �consumerr �r � credential� credential2r r r �test_save_and_loadS s z5TestUnencryptedFileCredentialStore.test_save_and_loadc C s< | � d�}| j�|d� | j�d�}| �|jj|jj� d S )Nr- zsome keyzsome other keyr/ r3 r r r � test_unique_id_doesnt_matterZ s z?TestUnencryptedFileCredentialStore.test_unique_id_doesnt_matterc C sT | � d�}| � d�}| j�|d� | j�|d� | j�d�}| �|jj|jj� d S )Nr- � consumer key2z unique key 1z unique key 2r/ �r � credential1r5 Zloadedr r r �&test_file_only_contains_one_credentialb s zITestUnencryptedFileCredentialStore.test_file_only_contains_one_credentialN) r r r r r% r+ r6 r7 r; r r r r r H s r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�TestKeyringCredentialStorez+Tests for the KeyringCredentialStore class.c C s t � | _t� | _d S r )r �keyringr r# r* r r r r% r s z TestKeyringCredentialStore.setUpc C sR t | j��> | �d�}| j�|d� | j�d�}| �|jj|jj� W 5 Q R X d S r, � r r= r r# r0 r1 r r2 r r3 r r r r6 v s �z-TestKeyringCredentialStore.test_save_and_loadc C s� t | j��� | �d�}| j�|d� | �d�}| j�|d� | j�d�}| �|� | �|jj |jj � | j�d�}| �|jj |jj � W 5 Q R X d S )N� consumer key1zkey 1r8 zkey 2) r r= r r# r0 r1 � assertTruer r2 r )r r: r5 Zloaded1Zloaded2r r r �test_lookup_by_unique_key s � �z4TestKeyringCredentialStore.test_lookup_by_unique_keyc C sj t | j��V | �d�}| j�|d� | �d�}| j�|d� | j�d�}| �|jj|jj� W 5 Q R X d S )Nr? z the only keyr8 r> r9 r r r �/test_reused_unique_id_overwrites_old_credential� s �zJTestKeyringCredentialStore.test_reused_unique_id_overwrites_old_credentialc C s, t | j�� | �| j�d�� W 5 Q R X d S )Nz no such key)r r= r r# r1 r* r r r � test_bad_unique_id_returns_none� s z:TestKeyringCredentialStore.test_bad_unique_id_returns_nonec s� G � fdd�dt �� � � | _t| j��f | �d�}| �|� | j�|d� | j�d�}| �|� | �|j j |j j � | �|j j |j j � W 5 Q R X d S )Nc s e Zd Z�� fdd�Z� ZS )zWTestKeyringCredentialStore.test_keyring_returns_unicode.<locals>.UnicodeInMemoryKeyringc s* t � | ��||�}t|t�r&|�d�}|S )Nzutf-8)�super� get_password� isinstance� unicode_type�encode)r �service�usernameZpassword)�UnicodeInMemoryKeyring� __class__r r rE � s � zdTestKeyringCredentialStore.test_keyring_returns_unicode.<locals>.UnicodeInMemoryKeyring.get_password�r r r rE � __classcell__r �rK �rL r rK � s rK r- r. ) r r= r r r@ r# r0 r1 r r2 r r r3 r rO r � test_keyring_returns_unicode� s � �z7TestKeyringCredentialStore.test_keyring_returns_unicodec s� G � fdd�dt �� � � | _t| j��R | �d�}| j�|d� | j�d�}| �|jj |jj � | �|jj |jj � W 5 Q R X d S )Nc s e Zd Z�� fdd�Z� ZS )zXTestKeyringCredentialStore.test_nonencoded_key_handled.<locals>.UnencodedInMemoryKeyringc s" t � | ��||�}t|dd � �S )N� )rD rE r )r rI rJ Zpw)�UnencodedInMemoryKeyringrL r r rE � s �zeTestKeyringCredentialStore.test_nonencoded_key_handled.<locals>.UnencodedInMemoryKeyring.get_passwordrM r �rS rP r rS � s rS r- r. ) r r= r r r# r0 r1 r r2 r r r3 r rT r �test_nonencoded_key_handled� s � �z6TestKeyringCredentialStore.test_nonencoded_key_handledc C s` G dd� dt �}|� | _t| j��4 | �d�}| j�|d� | j�d�}| �|� W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )zWTestKeyringCredentialStore.test_corrupted_key_handled.<locals>.CorruptedInMemoryKeyringc S s dS )NZbadr )r rI rJ r r r rE � s zdTestKeyringCredentialStore.test_corrupted_key_handled.<locals>.CorruptedInMemoryKeyring.get_passwordN)r r r rE r r r r �CorruptedInMemoryKeyring� s rV r- r. )r r= r r r# r0 r1 r )r rV r4 r5 r r r �test_corrupted_key_handled� s z5TestKeyringCredentialStore.test_corrupted_key_handledN) r r r r r% r6 rA rB rC rQ rU rW r r r r r< o s r< )r r& r! Zunittest�base64r �bytes�strZunicoderG Z launchpadlib.testing.helpersr r Zlaunchpadlib.credentialsr r r r ZTestCaser r r r< r r r r �<module> s '
Save File
Cancel