File: //usr/lib/python3/dist-packages/launchpadlib/tests/__pycache__/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 '