CasperSecurity
o
���a~ � @ sh d dl Z d dlZd dlZd dlZd dlZG dd� d�Ze jdd� �Zdd� Zej ej
Z e fdd �Z dS )
� Nc @ s: e Zd ZdZdd� Zd
dd�Zdd� Zd d
� Zd d � ZdS )� ImportKillerz@Context manager to make an import of a given name or names fail.c G s
|| _ d S �N��names)�selfr � r �6/usr/lib/python3/dist-packages/keyring/testing/util.py�__init__ s
zImportKiller.__init__Nc C s || j v r| S d S r r )r �fullname�pathr r r � find_module s
�zImportKiller.find_modulec C s || j v sJ �t|��r )r � ImportError)r r
r r r � load_module s zImportKiller.load_modulec C s8 i | _ | jD ] }tj�|d �| j |< qtj�d| � d S )Nr )�originalr �sys�modules�pop� meta_path�insert)r �namer r r � __enter__ s
zImportKiller.__enter__c G s6 t j�| � | j�� D ]
\}}|d ur|t j|< q d S r )r r �remover �itemsr )r �args�key�valuer r r �__exit__ s
��zImportKiller.__exit__r ) �__name__�
__module__� __qualname__�__doc__r r r r r r r r r r s
r c k s� � i }|� � D ]\}}| �|�||< |du r || v r | |= q|| |< qdV |� � D ]\}}|du r:|| v r9| |= q*|| |< q*dS )z�Helper context manager to make and unmake changes to a dict.
A None is not a valid value for the destination, and so means that the
associated name should be removed.N)r �get)� destination�changesr r r r r r �NoNoneDictMutator# s" ��
�
�r$ c K s t tjfi | ��S )z6A context manager to temporarily change the os.environ)r$ �os�environ)r# r r r �Environ: s r' c C s& d}t d| �D ] }|t�|�7 }q|S )z-Generate a random string with length <i>k</i>� r )�range�random�choice)�k�source�result�ir r r �
random_stringB s r0 )
�
contextlibr% r r* �stringr �contextmanagerr$ r' �
ascii_letters�digits�ALPHABETr0 r r r r �<module> s